更简洁的参数校验,使用 SpringBoot Validation 对参数进行校验
创始人
2024-02-14 12:18:45
0

在开发接口时,如果要对参数进行校验,你会怎么写?编写 if-else 吗?虽然也能达到效果,但是不够优雅。

今天,推荐一种更简洁的写法,使用 SpringBoot Validation 对方法参数进行校验,特别是在编写 Controller 层的方法时,直接使用一个注解即可完成参数校验。

示例代码:spring-validation-demo: SpringBootValidation Demo (gitee.com)

🚀引入依赖

想要完成上述所说的参数校验,我们需要一个核心依赖:spring-boot-starter-validation,此外,为了方便演示,还需要其他依赖。

依赖如下:

 org.springframework.bootspring-boot-starter-weborg.springframework.bootspring-boot-starter-validationorg.projectlomboklomboktrue
复制代码

💡 以下部分不是核心内容:

你在编写下面的示例代码中,会发现主要使用到了javax.validation.constraints 包下的注解,而这个包主要来自于 jakarta.validation-api 这个依赖。

如果引入依赖的时候直接引入 jakarta.validation-api 是无法实现参数校验功能的,因为它只定义了规范,而没有具体实现。但是 hibernate-validator 实现了这个规范,直接引入 hibernate-validator 也是可以实现参数校验功能的。

 jakarta.validationjakarta.validation-apiorg.hibernate.validatorhibernate-validator
复制代码

🚀 相关注解说明

这里罗列出一些主要的注解,这些注解主要来自于包 javax.validation.constraints,有兴趣查看源码的可以去这个包下查看。

可以先跳过这部分内容,下面的代码如果遇到不清楚作用的注解再回来查阅。

✈ 空值检查

注解说明
@NotBlank用于字符串,字符串不能为null 也不能为空字符串
@NotEmpty字符串同上,对于集合(Map,List,Set)不能为空,必须有元素
@NotNull不能为 null
@Null必须为 null

✈ 数值检查

注解说明
@DecimalMax(value)被注释的元素必须为数字,其值必须小于等于指定的值
@DecimalMin(value)被注释的元素必须为数字,其值必须大于等于指定的值
@Digits(integer, fraction)被注释的元素必须为数字,其值的整数部分精度为 integer,小数部分精度为 fraction
@Positive被注释的元素必须为正数
@PositiveOrZero被注释的元素必须为正数或 0
@Max(value)被注释的元素必须小于等于指定的值
@Min(value)被注释的元素必须大于等于指定的值
@Negative被注释的元素必须为负数
@NegativeOrZero被注释的元素必须为负数或 0

✈ Boolean 检查

注解说明
@AssertFalse被注释的元素必须值为 false
@AssertTrue被注释的元素必须值为 true

✈ 长度检查

注解说明
@Size(min,max)被注释的元素长度必须在 minmax 之间,可以是 String、Collection、Map、数组

✈ 日期检查

注解说明
@Future被注释的元素必须是一个将来的日期
@FutureOrPresent被注释的元素必须是现在或者将来的日期
@Past被注释的元素必须是一个过去的日期
@PastOrPresent被注释的元素必须是现在或者过去的日期

✈ 其他检查

注解说明
@Email被注释的元素必须是电子邮箱地址
@Pattern(regexp)被注释的元素必须符合正则表达式

除此之外,org.hibernate.validator.constraints 包下还有其他校验注解,例如 @ISBN 检查一个字符串是否是一个有效地 ISBN 序列号。

🚀 参数校验

接下来开始体验 Spring Boot Validation。

首先,编写一个需要校验的实体类:

 @Datapublic class Student {@NotBlank(message = "主键不能为空")private String id;@NotBlank(message = "名字不能为空")@Size(min=2, max = 4, message = "名字字符长度必须为 2~4个")private String name;@Pattern(regexp = "^1(3\d|4[5-9]|5[0-35-9]|6[567]|7[0-8]|8\d|9[0-35-9])\d{8}$", message = "手机号格式错误")private String phone;@Email(message = "邮箱格式错误")private String email;@Past(message = "生日必须早于当前时间")private Date birth;@Min(value = 0, message = "年龄必须为 0~100")@Max(value = 100, message = "年龄必须为 0~100")private Integer age;@PositiveOrZeroprivate Double score;}
复制代码

随后编写一个控制层代码,进行测试:

 @RestControllerpublic class TestController {​@GetMapping("/test")public Student test(@RequestBody @Validated Student student) {return student;}}
复制代码

使用 postman 进行测试,发送一个不带参数的请求,查看结果:

💡后端控制台日志打印是这样的(显示极度不友好),可以看到校验规则生效了:

 2022-11-23 22:10:13.249  WARN 19840 --- [nio-8080-exec-1] .w.s.m.s.DefaultHandlerExceptionResolver : Resolved [org.springframework.web.bind.MethodArgumentNotValidException: Validation failed for argument [0] in public com.example.springvalidationdemo.domain.Student com.example.springvalidationdemo.controller.TestController.test(com.example.springvalidationdemo.domain.Student) with 2 errors: [Field error in object 'student' on field 'name': rejected value [null]; codes [NotBlank.student.name,NotBlank.name,NotBlank.java.lang.String,NotBlank]; arguments [org.springframework.context.support.DefaultMessageSourceResolvable: codes [student.name,name]; arguments []; default message [name]]; default message [名字不能为空]] [Field error in object 'student' on field 'id': rejected value [null]; codes [NotBlank.student.id,NotBlank.id,NotBlank.java.lang.String,NotBlank]; arguments [org.springframework.context.support.DefaultMessageSourceResolvable: codes [student.id,id]; arguments []; default message [id]]; default message [主键不能为空]] ]
复制代码

🚀 全局异常处理

查看上面的日志打印,可以看到当参数校验不通过时,会抛出异常 MethodArgumentNotValidException,同时也会打印那些参数没有通过校验,以及该参数校验规则

为了方便查看,我们可以编写一个全局异常处理,处理这个参数校验异常,并使用统一返回实体返回给前端。

 @ControllerAdvice@Slf4jpublic class GlobalExceptionHandler {​@ExceptionHandler(MethodArgumentNotValidException.class)@ResponseBodypublic ResponseEntity exception(MethodArgumentNotValidException e, HttpServletRequest request) {Map result = new HashMap<>();BindingResult bindingResult = e.getBindingResult();log.error("请求[ {} ] {} 的参数校验发生错误", request.getMethod(), request.getRequestURL());for (ObjectError objectError : bindingResult.getAllErrors()) {FieldError fieldError = (FieldError) objectError;log.error("参数 {} = {} 校验错误:{}", fieldError.getField(), fieldError.getRejectedValue(), fieldError.getDefaultMessage());result.put(fieldError.getField(), fieldError.getDefaultMessage());}// 一般项目都会有自己定义的公共返回实体类,这里直接使用现成的 ResponseEntity 进行返回,同时设置 Http 状态码为 400return ResponseEntity.badRequest().body(result);​}​}
复制代码 

再次使用 postman 发起测试:

控制台打印出自定义的日志信息:

 2022-11-23 22:16:37.800 ERROR 19880 --- [nio-8080-exec-2] c.e.s.handler.GlobalExceptionHandler     : 请求[ GET ] http://localhost:8080/test 的参数校验发生错误2022-11-23 22:16:37.800 ERROR 19880 --- [nio-8080-exec-2] c.e.s.handler.GlobalExceptionHandler     : 参数 name = null 校验错误:名字不能为空2022-11-23 22:16:37.800 ERROR 19880 --- [nio-8080-exec-2] c.e.s.handler.GlobalExceptionHandler     : 参数 id = null 校验错误:主键不能为空2022-11-23 22:19:36.594 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 请求[ GET ] http://localhost:8080/test 的参数校验发生错误2022-11-23 22:19:36.594 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 参数 email = abc.com 校验错误:邮箱格式错误2022-11-23 22:19:36.594 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 参数 score = -20 校验错误:必须是正数或零2022-11-23 22:19:36.595 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 参数 birth = Thu Jan 01 08:00:00 CST 2099 校验错误:生日必须早于当前时间2022-11-23 22:19:36.595 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 参数 phone = 12233 校验错误:手机号格式错误2022-11-23 22:19:36.595 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 参数 age = -40 校验错误:年龄必须为 0~1002022-11-23 22:19:36.595 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 参数 name = 我是很长的名字 校验错误:名字字符长度必须为 2~4个2022-11-23 22:19:36.595 ERROR 19880 --- [nio-8080-exec-6] c.e.s.handler.GlobalExceptionHandler     : 参数 score = -20 校验错误:需要在0和9223372036854775807之间
复制代码

🚀 传递校验

我们也可以使用传递校验,即一个参数类中包含了另一个参数类,被包含的参数类也可以被校验。

在声明一个新的参数类,同时修改 Student 类。

 @Datapublic class ClassInfo {@NotBlank(message = "班主任姓名不能为空")private String teacher;@NotNull(message = "教师不能为空")private Integer classroom;@NotNull(message = "年级不能为空")@Min(value = 1, message = "年级只能是 1-6")@Max(value = 6, message = "年级只能是 1-6")private Integer grade;}​@Datapublic class Student {//.............// 新加的字段,被包含的参数类,使用 @Valid 就能传递校验,如果不使用 @Valid 注解,则无法传递校验。@Validprivate ClassInfo classInfo;}
复制代码

再使用 postman 测试一次

🚀 分组校验

此外还可以使用分组校验,令一组方法对某些字段校验,而令一组方法对其他字段校验,例如:一般情况下,新增实体的接口方法 [POST] 不需要主键 ID,修改实体的接口方法 [PUT] 就需要主键 ID 以便进行修改。

为注解 @Validated 赋值属性 value,以及为那些校验注解赋值属性 group, 即可达到分组的效果。

接下来看看如何实现分组校验。

Student 类中添加两个内部接口 Inteface,同时修改 id 字段的注解,以进行分组

 @Datapublic class Student {// id 字段属于 Create 组@NotBlank(message = "主键不能为空", groups = {Student.Create.class})private String id;// .............// 更新分组public interface Update {}​// 创建分组public interface Create {}}
复制代码

在控制层新增两个接口

 @RestControllerpublic class TestController {​// @Validated 注解可以赋值 value 属性进行分组,value 是可以以数组的形式赋值,既可以分配多个组@PostMapping("/students")public Student create(@RequestBody @Validated(Student.Create.class) Student student) {return student;}​@PutMapping("/students")public Student update(@RequestBody @Validated(Student.Update.class) Student student) {return student;}}
复制代码

在 postman 上进行测试:

可以看到分组校验也生效了。

🚀 总结

在实际开发中,我们可以使用 Spring Boot Validation 提供的注解进行参数校验,提高代码的可读性,避免编写大量的 if-else 代码块和重复的校验语句。

相关内容

热门资讯

喜欢穿一身黑的男生性格(喜欢穿... 今天百科达人给各位分享喜欢穿一身黑的男生性格的知识,其中也会对喜欢穿一身黑衣服的男人人好相处吗进行解...
发春是什么意思(思春和发春是什... 本篇文章极速百科给大家谈谈发春是什么意思,以及思春和发春是什么意思对应的知识点,希望对各位有所帮助,...
网络用语zl是什么意思(zl是... 今天给各位分享网络用语zl是什么意思的知识,其中也会对zl是啥意思是什么网络用语进行解释,如果能碰巧...
为什么酷狗音乐自己唱的歌不能下... 本篇文章极速百科小编给大家谈谈为什么酷狗音乐自己唱的歌不能下载到本地?,以及为什么酷狗下载的歌曲不是...
家里可以做假山养金鱼吗(假山能... 今天百科达人给各位分享家里可以做假山养金鱼吗的知识,其中也会对假山能放鱼缸里吗进行解释,如果能碰巧解...
华为下载未安装的文件去哪找(华... 今天百科达人给各位分享华为下载未安装的文件去哪找的知识,其中也会对华为下载未安装的文件去哪找到进行解...
四分五裂是什么生肖什么动物(四... 本篇文章极速百科小编给大家谈谈四分五裂是什么生肖什么动物,以及四分五裂打一生肖是什么对应的知识点,希...
怎么往应用助手里添加应用(应用... 今天百科达人给各位分享怎么往应用助手里添加应用的知识,其中也会对应用助手怎么添加微信进行解释,如果能...
客厅放八骏马摆件可以吗(家里摆... 今天给各位分享客厅放八骏马摆件可以吗的知识,其中也会对家里摆八骏马摆件好吗进行解释,如果能碰巧解决你...
苏州离哪个飞机场近(苏州离哪个... 本篇文章极速百科小编给大家谈谈苏州离哪个飞机场近,以及苏州离哪个飞机场近点对应的知识点,希望对各位有...