Springboot 优雅的处理数据库报错信息

最新推荐文章于 2024-08-06 11:26:03 发布

言小溪

最新推荐文章于 2024-08-06 11:26:03 发布

阅读量4.6k

点赞数 6

分类专栏： Java框架【SpringBoot】文章标签：数据库 java mybatis sql 反射

原创声明：本文为博主言小溪的博文，未经博主允许不得转载！

本文链接：https://blog.csdn.net/qq_31254489/article/details/119815991

版权

Java框架【SpringBoot】专栏收录该内容

11 篇文章 0 订阅

订阅专栏

本文介绍如何通过Spring Boot的@RestControllerAdvice捕获并解析SQLIntegrityConstraintViolationException，利用反射找到被占用字段及其注解，解决数据库完整性约束冲突问题。实例包括邮箱和账号占用，原理涉及正则表达式和Java反射。

摘要由CSDN通过智能技术生成

@RestController
@ResponseHandler
@RestControllerAdvice
@Api(tags = "統一错误请求控制")
public class MyErrorController implements ErrorController {
	// 统一sql异常处理类
    @Autowired
    SqlExceptionHandler sqlExceptionHandler;

    // sql 报错处理
    @ExceptionHandler(value = SQLException.class)
    @ResponseStatus
    public String sqlError(SQLException e) {
    	// 我们交给异常处理类去处理
        return sqlExceptionHandler.handle(e);
    }
}

具体流程如下
在这里插入图片描述

完整性约束报错处理

一般就是插入的时候和已经有的数据重复了

接下来复制粘贴下面2个类，并要遵守以下规则

数据库的约束名称必须遵守 表名.表名_键1_键2_键3_xxx_uindex 例如 user.user_name_unidex 意思就是 user 表下的 name 的唯一约束。
实体类的名称，还有属性，必须和数据库中的一致，mysql 必须遵守下划线命名规则， java 则可以使用驼峰或者下划线。
使用 @HandleSqlException 注解去方法上面标注需要处理的具体异常。例如下面的 integrityConstraint 方法。

/**
 * sql 错误异常处理
 *
 * @author enncy
 */
@Component
public class SqlExceptionHandler {
    /**
     *  分发异常处理
     * @return: void
     */
    public String handle(SQLException e) {
    	// 获取当前类的方法
        Optional<Method> first = Arrays.stream(SqlExceptionHandler.class.getDeclaredMethods())
        		// 寻找有 HandleSqlException 注解的方法
                .filter(m -> m.isAnnotationPresent(HandleSqlException.class))
                // 寻找和参数 e 的类相等的 HandleSqlException 
                .filter(m -> m.getAnnotation(HandleSqlException.class).value() == e.getClass())
                .findFirst();

		// 如果存在，则分发给指定的方法
        if( first.isPresent()){
            try {
                return (String) first.get().invoke(this, e);
            } catch (IllegalAccessException | InvocationTargetException exception) {
                return "服务器内部出现异常";
            }
        }else{
            return "服务器内部出现异常";
        }
    }

    /**
     *  完整性约束被破坏处理
     * @return: java.lang.String
     */
    @HandleSqlException(SQLIntegrityConstraintViolationException.class)
    public String integrityConstraint(SQLException e) {
        // 匹配数据库报错信息
        String regex = "Duplicate entry '(.*?)' for key '(.*?)\\..*?_(.*?)_uindex'";
        Matcher matcher = Pattern.compile(regex).matcher(e.getMessage());
        // 如果存在信息
        if (matcher.find()) {
            // 获取被占用的值
            String value = matcher.group(1);
            // 获取数据库表
            String table = matcher.group(2);
            // 获取发生冲突的约束键
            String keys = matcher.group(3);
            // 扫描实体类
            List<Class<?>> scan = ClassScanner.scan("cn.enncy.funny.entity");
            // 寻找冲突的字段的注解描述
            String description = scan.stream()
                    .filter(BaseEntity.class::isAssignableFrom)
                    .filter(c -> c.getName().toLowerCase().contains(table))
                    .map(c -> {
                        // 获取实体类的属性值
                        Field[] declaredFields = c.getDeclaredFields();
                        for (Field declaredField : declaredFields) {
                            // 驼峰转下划线，这里的 humpToUnderline 方法参考文章 : https://blog.csdn.net/qq_31254489/article/details/115842821
                            String name = StringUtils.humpToUnderline(declaredField.getName());
                            // 如果属性包含在 keys 的值里面
                            if (keys.contains(name)) {
                                // 返回属性上面的 注解信息
                                return declaredField.getAnnotation(ApiModelProperty.class).value();
                            }
                        }
                        return "";
                    }).findFirst().orElse("");

            return description + " " + value + " 已经被占用";
        }
        return "信息已经被占用";
    }
}

/**
 * @author enncy
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface HandleSqlException {
    Class<? extends SQLException> value();
}