Springboot入门教程(9)-用validation做参数校验及全局异常处理

参数校验和异常处理也是后台代码中很重要的一部分,如果每次都自己写代码做校验就会很繁琐,所以spring框架中也提供了validation组件来直接做参数校验,本文就是讲述validation组件的一些常见的用法,以及顺便讲一下如何全局的处理异常。

  1. 首先依然是先要在build.gradle的dependencies中添加依赖包
implementation "org.springframework.boot:spring-boot-starter-validation"
  1. 接着只要直接在java bean中配置参数条件就可以了,例如我们给teacher的几个属性加一些条件
    @Size(max=4, min=2, message="老师姓名应为2-4字")
    private String name;
    @NotNull(message="老师性别不能为空")
    @Min(value = 0, message = "性别值只能为0或1,0:女,1:男")
    @Max(value = 1, message = "性别值只能为0或1,0:女,1:男")
    private Integer gender;
    @NotNull(message="老师年龄不能为空")
    @Min(value = 20, message = "老师年龄不能小于20")
    @Max(value = 70, message = "老师年龄不能大于70")

这里的注解约束还有很多其他的,具体可以参考SpringBoot使用Validation校验参数中的说明。也可以查看Hibernate Validator的官方文档,里面有更详细的说明,还有一些不是很常见的特殊的注解约束。

  1. 然后就可以在controller接口的参数前加上需要校验的注解,注解有两种,一个是@Valid,一个是@Validated,这两个大部分情况使用是一样的。例如,这样加上:
    @PostMapping(value = "/addTeacher", consumes = { "application/x-www-form-urlencoded" })
    @ResponseBody
    public ResponseData addTeacher(@Validated Teacher teacher)
    {
        if(teacher.getFile() != null){
            String fileName = FileUtil.upload(teacher.getFile(), path, teacher.getFile().getOriginalFilename());
            if ( fileName!= null){
                teacher.setImageUrl(fileName);
            }
        }
        teacherMapper.insertTeacher(teacher);
        ResponseData responseData = ResponseData.ok();
        return responseData;
    }

然后我们来运行试试


传入错误的参数

返回结果

可以看到返回了默认格式的错误信息的json字符串。
但是由于这个信息格式是默认的,和我们自己定义的不一样,前端可能就无法辨认,这时就有两个方法可以处理:
第一个方法是使用BindingResult,我们可以用BindingResult接收验证的结果,如果错误,再按我们自己定义的格式返回错误信息。

@PostMapping(value = "/addTeacher", consumes = { "application/x-www-form-urlencoded" })
    @ResponseBody
    public ResponseData addTeacher(@Validated Teacher teacher, BindingResult bindingResult)
    {
        if (bindingResult.hasErrors()) {
            ResponseData responseData = new ResponseData(400, bindingResult.getFieldError().getDefaultMessage());
            return responseData;
        }
        if(teacher.getFile() != null){
            String fileName = FileUtil.upload(teacher.getFile(), path, teacher.getFile().getOriginalFilename());
            if ( fileName!= null){
                teacher.setImageUrl(fileName);
            }
        }
        teacherMapper.insertTeacher(teacher);
        ResponseData responseData = ResponseData.ok();
        return responseData;
    }

结果就会变成这样


返回自定义的格式结果

第二个方法则就要引入本文的第二个课题了,就是全局的处理异常。因为如果每个校验的异常都要这样写的话,那也是非常麻烦了。所以Spring也提供了非常方便的全局异常的注解,就是@RestControllerAdvice和@ExceptionHandler。我们就可以构建如下的全局异常处理的类:

@RestControllerAdvice
public class GlobalExceptionHandler {
    /**
     * 处理Validated校验异常
     * 

* 注: 常见的ConstraintViolationException异常, 也属于ValidationException异常 * * @param e * 捕获到的异常 * @return 返回给前端的data */ @ResponseStatus(code = HttpStatus.BAD_REQUEST) @ExceptionHandler(value = {BindException.class, ValidationException.class, MethodArgumentNotValidException.class}) public ResponseData handleParameterVerificationException(Exception e) { String msg = null; /// BindException if (e instanceof BindException) { // getFieldError获取的是第一个不合法的参数(P.S.如果有多个参数不合法的话) FieldError fieldError = ((BindException) e).getFieldError(); if (fieldError != null) { msg = fieldError.getDefaultMessage(); } /// MethodArgumentNotValidException } else if (e instanceof MethodArgumentNotValidException) { BindingResult bindingResult = ((MethodArgumentNotValidException) e).getBindingResult(); // getFieldError获取的是第一个不合法的参数(P.S.如果有多个参数不合法的话) FieldError fieldError = bindingResult.getFieldError(); if (fieldError != null) { msg = fieldError.getDefaultMessage(); } /// ValidationException 的子类异常ConstraintViolationException } else if (e instanceof ConstraintViolationException) { /* * ConstraintViolationException的e.getMessage()形如 * {方法名}.{参数名}: {message} * 这里只需要取后面的message即可 */ msg = e.getMessage(); if (msg != null) { int lastIndex = msg.lastIndexOf(':'); if (lastIndex >= 0) { msg = msg.substring(lastIndex + 1).trim(); } } /// ValidationException 的其它子类异常 } else { msg = "处理参数时异常"; } ResponseData responseData = new ResponseData(400, msg); return responseData; } @ExceptionHandler @ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR) public ResponseData handleException(Exception ex){ if (ex instanceof DataIntegrityViolationException) { // 数据库操作异常 if(ex.toString().contains("a foreign key constraint fails")){ //外键关联问题,具体前端可以根据发送的请求判断 return new ResponseData(5001, "a foreign key constraint fails"); } } return new ResponseData(500, "Internal Server Error"); } }

这里我写了两个方法,一个是专门处理参数校验异常的,我把它定义为BadRequest一类的返回,我主要考虑了三种参数异常的捕获,例如前面的结果,就是BindException,在未加BindResult的处理之前,我们可以从控制台打印的日志看出。


控制台打出的BindException异常

其实这个就是当参数为Java bean且传参方式为RequestParam就是直接在url地址上传参的方式时校验会返回的异常。
而第二种MethodArgumentNotValidException则是同样参数为Java bean但是传参方式为@RequestBody且applicationType为application/json的时候校验会返回的异常,我们可以试一下


控制台打出的MethodArgumentNotValidException异常

而第三种ConstraintViolationException呢,则是参数为普通类型的直接在参数前加校验条件的异常返回类型。例如在findSubjects的参数上加上校验
@GetMapping(value = "/subjects")
    public ResponseDataNew> findSubjects(final String name, @Min(value = 0, message = "页码不能小于0")final Integer index, final Integer size){
        Page page = PageHelper.startPage(index + 1, size);
        List subjectList = subjectMapper.findSubjects(name);
        ResponseDataNew> response = new ResponseDataNew<>();
        response.ok();
        ListWithPageData data = new ListWithPageData<>();
        data.setPageCount(page.getPages());
        data.setTotal(page.getTotal());
        data.setList(subjectList);
        response.setData(data);
        return response;
    }

这里还涉及到@Validated的另一个用法,就是加在类上的注解,只有这样

@RestController
@RequestMapping(value = "/subject")
@Validated
public class SubjectController {
...
}

直接加在参数前的校验注解才会有用。传入index为-1时就会抛出这个异常


控制台打出的ConstraintViolationException异常

另外我还写了一个方法用来捕获其他类型的异常,比如这个外键关联的异常。@ExceptionHandler这个注解就是可以指定捕获的Exception的类型,如果没有指定,那么就会捕获任意类型。

此外,@Validated还支持分组,比如当我们新建一条数据时,id是必然为空的,而更新数据时,id又必须不为空,这时就可以用到这个。
(1)首先,我们在entity包中分别建两个接口Insert和Update

public interface Insert extends Default {
}
public interface Update extends Default {
}

(2)接着,以Subject为例,需要在id上加两组注解

    @Schema(example = "1")
    @Null(groups = {Insert.class})
    @NotNull(groups = {Update.class}, message="id不能为空")
    private Long id;

(3)分别在新增和更新的接口上加上对应的注解,如下

    @PostMapping(value = "/addSubject", consumes = { "application/x-www-form-urlencoded" })
    public ResponseData addSubject(@Validated(value = Insert.class) Subject subject){
        subjectMapper.insertSubject(subject.getName());
        ResponseData responseData = ResponseData.ok();
        return responseData;
    }

    @PostMapping(value = "/editSubject", consumes = { "application/x-www-form-urlencoded" })
    public ResponseData editSubject(@Validated(value = Update.class) Subject subject)
    {
        subjectMapper.updateSubject(subject.getId(), subject.getName());
        ResponseData responseData = ResponseData.ok();
        return responseData;
    }

但是这样加完会有个问题,就是在swagger上,我们会发现addSubject的接口id的参数仍然是required的,这似乎是一个bug。而用Postman测试,结果则是正常的


addSubject校验

editSubject校验

不过,我也试了一下,如果把传参方式改为@RequestBody就是application/json的话也可以解决这个问题。
需要注意的是@Valid是不支持这样分组的,这是这两个注解其中一个差异。

@Validated和@Valid还有一个差异在于@Valid支持嵌套校验、而@Validated不支持。这是什么意思呢?比如我需要做一个批量新增的功能,所以我传参的时候会传一个list,就像这样

@PostMapping(value = "/addSubjects")
    public ResponseData addSubjects(@Validated(value = Insert.class) @RequestBody List subjects){
        subjectMapper.insertSubjects(subjects);
        ResponseData responseData = ResponseData.ok();
        return responseData;
    }

但是这时候我们加的这个@Validated的注解会发现是不起作用的,就是因为它不支持嵌套,而要验证的对象包在List中。这时我们只能把它改为@Valid,分组也就没办法使用了。还要注意的是同样要在SubjectController类上加了@Validated注解才有用。
不过也还有一种方法可以同时解决这两个问题,就是自定义实现一个List ValidatedList,这个方法的话可以参考使用@Validated校验List接口参数的两种方式这篇博客。

最后再说一下的是,validation还支持自定义的校验,这个也可以参考SpringBoot使用Validation校验参数这篇博客,我这里也就不再详细说明了。
代码依旧可以参考我在github上面的代码https://github.com/ahuadoreen/studentmanager。

参考文档
SpringBoot使用Validation校验参数
Spring 参数校验的异常处理
使用@Validated校验List接口参数的两种方式

你可能感兴趣的:(Springboot入门教程(9)-用validation做参数校验及全局异常处理)