package com.oying.utils;
|
|
import cn.hutool.core.map.MapUtil;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.dao.DuplicateKeyException;
|
|
import java.sql.SQLException;
|
import java.util.Map;
|
import java.util.Optional;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
@Slf4j
|
public class DuplicateKeyExceptionUtil {
|
|
public static final Map<String, String> INDEX_MAPPING = MapUtil.ofEntries(
|
MapUtil.entry("pc_store.uk_store_name", "店铺名称")
|
);
|
|
/**
|
* 获取错误消息
|
*/
|
public static String getDisplayMessage(DuplicateKeyException e) {
|
try {
|
String indexName = findMySQLDuplicateKey(e);
|
String fieldDisplayName = INDEX_MAPPING.getOrDefault(indexName, indexName);
|
return Optional.ofNullable(fieldDisplayName).map(o -> o + "已存在").orElse(e.getMessage());
|
} catch (Exception ex) {
|
log.error("解析'DuplicateKeyException'消息失败", ex);
|
return e.getMessage();
|
}
|
}
|
|
/**
|
* 提取索引名称
|
*/
|
public static String findMySQLDuplicateKey(DuplicateKeyException e) {
|
if (e.getCause() instanceof SQLException) {
|
SQLException sqlEx = (SQLException) e.getCause();
|
if (sqlEx.getErrorCode() == 1062) { // MySQL duplicate key error code
|
String errorMsg = sqlEx.getMessage();
|
// 使用正则提取索引名
|
Matcher matcher = Pattern.compile("for key '(.+?)'").matcher(errorMsg);
|
if (matcher.find()) {
|
return matcher.group(1);
|
}
|
}
|
}
|
return null;
|
}
|
}
|
