• 如何对SpringBoot接口参数进行校验?


    什么是不优雅的参数校验

    后端对前端传过来的参数也是需要进行校验的,如果在controller中直接校验需要用大量的if else做判断

    以添加用户的接口为例,需要对前端传过来的参数进行校验, 如下的校验就是不优雅的:

    1. @RestController
    2. @RequestMapping("/user")
    3. public class UserController {
    4. @PostMapping("add")
    5. public ResponseEntity add(User user) {
    6. if(user.getName()==null) {
    7. return ResponseResult.fail("user name should not be empty");
    8. } else if(user.getName().length()<5 || user.getName().length()>50){
    9. return ResponseResult.fail("user name length should between 5-50");
    10. }
    11. if(user.getAge()< 1 || user.getAge()> 150) {
    12. return ResponseResult.fail("invalid age");
    13. }
    14. // ...
    15. return ResponseEntity.ok("success");
    16. }
    17. }

    针对这个普遍的问题,Java开者在Java API规范 (JSR303) 定义了Bean校验的标准validation-api,但没有提供实现。

    hibernate validation是对这个规范的实现,并增加了校验注解如@Email、@Length等。

    Spring Validation是对hibernate validation的二次封装,用于支持spring mvc参数自动校验。

    接下来,我们以springboot项目为例,介绍Spring Validation的使用。

    实现案例

    本例子采用 spring validation 对参数绑定进行校验,主要给你提供参数校验的思路。针对接口统一的错误信息(比如绑定参数检查的错误)封装请看

    POM

    添加pom依赖

    1. org.springframework.boot
    2. spring-boot-starter-validation

    请求参数封装

    单一职责,所以将查询用户的参数封装到UserParam中, 而不是User(数据库实体)本身。

    对每个参数字段添加validation注解约束和message。

    1. /**
    2. * user.
    3. *
    4. * @author pdai
    5. */
    6. @Data
    7. @Builder
    8. @ApiModel(value = "User", subTypes = {AddressParam.class})
    9. public class UserParam implements Serializable {
    10. private static final long serialVersionUID = 1L;
    11. @NotEmpty(message = "could not be empty")
    12. private String userId;
    13. @NotEmpty(message = "could not be empty")
    14. @Email(message = "invalid email")
    15. private String email;
    16. @NotEmpty(message = "could not be empty")
    17. @Pattern(regexp = "^(\\d{6})(\\d{4})(\\d{2})(\\d{2})(\\d{3})([0-9]|X)$", message = "invalid ID")
    18. private String cardNo;
    19. @NotEmpty(message = "could not be empty")
    20. @Length(min = 1, max = 10, message = "nick name should be 1-10")
    21. private String nickName;
    22. @NotEmpty(message = "could not be empty")
    23. @Range(min = 0, max = 1, message = "sex should be 0-1")
    24. private int sex;
    25. @Max(value = 100, message = "Please input valid age")
    26. private int age;
    27. @Valid
    28. private AddressParam address;
    29. }

    Controller中获取参数绑定结果

    使用@Valid或者@Validate注解,参数校验的值放在BindingResult中

    1. /**
    2. * @author pdai
    3. */
    4. @Slf4j
    5. @Api(value = "User Interfaces", tags = "User Interfaces")
    6. @RestController
    7. @RequestMapping("/user")
    8. public class UserController {
    9. /**
    10. * http://localhost:8080/user/add .
    11. *
    12. * @param userParam user param
    13. * @return user
    14. */
    15. @ApiOperation("Add User")
    16. @ApiImplicitParam(name = "userParam", type = "body", dataTypeClass = UserParam.class, required = true)
    17. @PostMapping("add")
    18. public ResponseEntity add(@Valid @RequestBody UserParam userParam, BindingResult bindingResult) {
    19. if (bindingResult.hasErrors()) {
    20. List errors = bindingResult.getAllErrors();
    21. errors.forEach(p -> {
    22. FieldError fieldError = (FieldError) p;
    23. log.error("Invalid Parameter : object - {},field - {},errorMessage - {}", fieldError.getObjectName(), fieldError.getField(), fieldError.getDefaultMessage());
    24. });
    25. return ResponseEntity.badRequest().body("invalid parameter");
    26. }
    27. return ResponseEntity.ok("success");
    28. }
    29. }

    校验结果

    POST访问添加User的请求

    后台输出参数绑定错误信息:(包含哪个对象,哪个字段,什么样的错误描述)

    1. 2021-09-16 10:37:05.173 ERROR 21216 --- [nio-8080-exec-8] t.p.s.v.controller.UserController : Invalid Parameter : object - userParam,field - nickName,errorMessage - could not be empty
    2. 2021-09-16 10:37:05.176 ERROR 21216 --- [nio-8080-exec-8] t.p.s.v.controller.UserController : Invalid Parameter : object - userParam,field - email,errorMessage - could not be empty
    3. 2021-09-16 10:37:05.176 ERROR 21216 --- [nio-8080-exec-8] t.p.s.v.controller.UserController : Invalid Parameter : object - userParam,field - cardNo,errorMessage - could not be empty

    (本例只是springboot-validation的简单用例,针对接口统一的错误信息封装请看

    进一步理解

    我们再通过一些问题来帮助你更深入理解validation校验。@pdai

    Validation分组校验?

    上面的例子中,其实存在一个问题,UserParam既可以作为addUser的参数(id为空),又可以作为updateUser的参数(id不能为空),这时候怎么办呢?分组校验登场。

    1. @Data
    2. @Builder
    3. @ApiModel(value = "User", subTypes = {AddressParam.class})
    4. public class UserParam implements Serializable {
    5. private static final long serialVersionUID = 1L;
    6. @NotEmpty(message = "could not be empty") // 这里定为空,对于addUser时是不合适的
    7. private String userId;
    8. }

    这时候可以使用Validation分组

    • 先定义分组(无需实现接口)
    1. public interface AddValidationGroup {
    2. }
    3. public interface EditValidationGroup {
    4. }
    • 在UserParam的userId字段添加分组
    1. @Data
    2. @Builder
    3. @ApiModel(value = "User", subTypes = {AddressParam.class})
    4. public class UserParam implements Serializable {
    5. private static final long serialVersionUID = 1L;
    6. @NotEmpty(message = "{user.msg.userId.notEmpty}", groups = {EditValidationGroup.class}) // 这里
    7. private String userId;
    8. }
    • controller中的接口使用校验时使用分组

    PS: 需要使用@Validated注解

    1. @Slf4j
    2. @Api(value = "User Interfaces", tags = "User Interfaces")
    3. @RestController
    4. @RequestMapping("/user")
    5. public class UserController {
    6. /**
    7. * http://localhost:8080/user/add .
    8. *
    9. * @param userParam user param
    10. * @return user
    11. */
    12. @ApiOperation("Add User")
    13. @ApiImplicitParam(name = "userParam", type = "body", dataTypeClass = UserParam.class, required = true)
    14. @PostMapping("add")
    15. public ResponseEntity add(@Validated(AddValidationGroup.class) @RequestBody UserParam userParam) {
    16. return ResponseEntity.ok(userParam);
    17. }
    18. /**
    19. * http://localhost:8080/user/add .
    20. *
    21. * @param userParam user param
    22. * @return user
    23. */
    24. @ApiOperation("Edit User")
    25. @ApiImplicitParam(name = "userParam", type = "body", dataTypeClass = UserParam.class, required = true)
    26. @PostMapping("edit")
    27. public ResponseEntity edit(@Validated(EditValidationGroup.class) @RequestBody UserParam userParam) {
    28. return ResponseEntity.ok(userParam);
    29. }
    30. }
    • 测试

    @Validate和@Valid什么区别?

    细心的你会发现,上个例子中用的是@Validate, 而不是@Valid,那它们之间的区别是什么呢?

    在检验Controller的入参是否符合规范时,使用@Validated或者@Valid在基本验证功能上没有太多区别。但是在分组、注解地方、嵌套验证等功能上两个有所不同:

    • 分组

    @Validated:提供了一个分组功能,可以在入参验证时,根据不同的分组采用不同的验证机制,这个网上也有资料,不详述。@Valid:作为标准JSR-303规范,还没有吸收分组的功能。

    • 注解地方

    @Validated:可以用在类型、方法和方法参数上。但是不能用在成员属性(字段)上

    @Valid:可以用在方法、构造函数、方法参数和成员属性(字段)上

    • 嵌套类型

    比如本文例子中的address是user的一个嵌套属性, 只能用@Valid

    1. @Data
    2. @Builder
    3. @ApiModel(value = "User", subTypes = {AddressParam.class})
    4. public class UserParam implements Serializable {
    5. private static final long serialVersionUID = 1L;
    6. @Valid // 这里只能用@Valid
    7. private AddressParam address;
    8. }

    有哪些常用的校验?

    从以下三类理解。

    • JSR303/JSR-349: JSR303是一项标准,只提供规范不提供实现,规定一些校验规范即校验注解,如@Null,@NotNull,@Pattern,位于javax.validation.constraints包下。JSR-349是其的升级版本,添加了一些新特性
    1. @AssertFalse 被注释的元素只能为false
    2. @AssertTrue 被注释的元素只能为true
    3. @DecimalMax 被注释的元素必须小于或等于{value}
    4. @DecimalMin 被注释的元素必须大于或等于{value}
    5. @Digits 被注释的元素数字的值超出了允许范围(只允许在{integer}位整数和{fraction}位小数范围内)
    6. @Email 被注释的元素不是一个合法的电子邮件地址
    7. @Future 被注释的元素需要是一个将来的时间
    8. @FutureOrPresent 被注释的元素需要是一个将来或现在的时间
    9. @Max 被注释的元素最大不能超过{value}
    10. @Min 被注释的元素最小不能小于{value}
    11. @Negative 被注释的元素必须是负数
    12. @NegativeOrZero 被注释的元素必须是负数或零
    13. @NotBlank 被注释的元素不能为空
    14. @NotEmpty 被注释的元素不能为空
    15. @NotNull 被注释的元素不能为null
    16. @Null 被注释的元素必须为null
    17. @Past 被注释的元素需要是一个过去的时间
    18. @PastOrPresent 被注释的元素需要是一个过去或现在的时间
    19. @Pattern 被注释的元素需要匹配正则表达式"{regexp}"
    20. @Positive 被注释的元素必须是正数
    21. @PositiveOrZero 被注释的元素必须是正数或零
    22. @Size 被注释的元素个数必须在{min}和{max}之间
    • hibernate validation:hibernate validation是对这个规范的实现,并增加了一些其他校验注解,如@Email,@Length,@Range等等
    1. @CreditCardNumber 被注释的元素不合法的信用卡号码
    2. @Currency 被注释的元素不合法的货币 (必须是{value}其中之一)
    3. @EAN 被注释的元素不合法的{type}条形码
    4. @Email 被注释的元素不是一个合法的电子邮件地址 (已过期)
    5. @Length 被注释的元素长度需要在{min}和{max}之间
    6. @CodePointLength 被注释的元素长度需要在{min}和{max}之间
    7. @LuhnCheck 被注释的元素${validatedValue}的校验码不合法, Luhn模10校验和不匹配
    8. @Mod10Check 被注释的元素${validatedValue}的校验码不合法, 模10校验和不匹配
    9. @Mod11Check 被注释的元素${validatedValue}的校验码不合法, 模11校验和不匹配
    10. @ModCheck 被注释的元素${validatedValue}的校验码不合法, ${modType}校验和不匹配 (已过期)
    11. @NotBlank 被注释的元素不能为空 (已过期)
    12. @NotEmpty 被注释的元素不能为空 (已过期)
    13. @ParametersScriptAssert 被注释的元素执行脚本表达式"{script}"没有返回期望结果
    14. @Range 被注释的元素需要在{min}和{max}之间
    15. @SafeHtml 被注释的元素可能有不安全的HTML内容
    16. @ScriptAssert 被注释的元素执行脚本表达式"{script}"没有返回期望结果
    17. @URL 被注释的元素需要是一个合法的URL
    18. @DurationMax 被注释的元素必须小于${inclusive == true ? '或等于' : ''}${days == 0 ? '' : days += '天'}${hours == 0 ? '' : hours += '小时'}${minutes == 0 ? '' : minutes += '分钟'}${seconds == 0 ? '' : seconds += '秒'}${millis == 0 ? '' : millis += '毫秒'}${nanos == 0 ? '' : nanos += '纳秒'}
    19. @DurationMin 被注释的元素必须大于${inclusive == true ? '或等于' : ''}${days == 0 ? '' : days += '天'}${hours == 0 ? '' : hours += '小时'}${minutes == 0 ? '' : minutes += '分钟'}${seconds == 0 ? '' : seconds += '秒'}${millis == 0 ? '' : millis += '毫秒'}${nanos == 0 ? '' : nanos += '纳秒'}
    • spring validation:spring validation对hibernate validation进行了二次封装,在springmvc模块中添加了自动校验,并将校验信息封装进了特定的类中

    自定义validation?

    如果上面的注解不能满足我们检验参数的要求,我们能不能自定义校验规则呢? 可以。

    • 定义注解
    1. package tech.pdai.springboot.validation.group.validation.custom;
    2. import javax.validation.Constraint;
    3. import javax.validation.Payload;
    4. import java.lang.annotation.Documented;
    5. import java.lang.annotation.Retention;
    6. import java.lang.annotation.Target;
    7. import static java.lang.annotation.ElementType.*;
    8. import static java.lang.annotation.RetentionPolicy.RUNTIME;
    9. @Target({ METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE })
    10. @Retention(RUNTIME)
    11. @Documented
    12. @Constraint(validatedBy = {TelephoneNumberValidator.class}) // 指定校验器
    13. public @interface TelephoneNumber {
    14. String message() default "Invalid telephone number";
    15. Class[] groups() default { };
    16. Classextends Payload>[] payload() default { };
    17. }
    • 定义校验器
    1. public class TelephoneNumberValidator implements ConstraintValidator {
    2. private static final String REGEX_TEL = "0\\d{2,3}[-]?\\d{7,8}|0\\d{2,3}\\s?\\d{7,8}|13[0-9]\\d{8}|15[1089]\\d{8}";
    3. @Override
    4. public boolean isValid(String s, ConstraintValidatorContext constraintValidatorContext) {
    5. try {
    6. return Pattern.matches(REGEX_TEL, s);
    7. } catch (Exception e) {
    8. return false;
    9. }
    10. }
    11. }
    • 使用
    1. @Data
    2. @Builder
    3. @ApiModel(value = "User", subTypes = {AddressParam.class})
    4. public class UserParam implements Serializable {
    5. private static final long serialVersionUID = 1L;
    6. @NotEmpty(message = "{user.msg.userId.notEmpty}", groups = {EditValidationGroup.class})
    7. private String userId;
    8. @TelephoneNumber(message = "invalid telephone number") // 这里
    9. private String telephone;
    10. }

  • 相关阅读:
    Algorithm Review 2
    MySQL索引
    基于形状的匹配提纲
    Maxwell安装部署
    vmware安装ubuntu22.04无法和window主机拷贝文件处理
    Furion api npm web vue混合开发
    【100天精通Python】Day65:Python可视化_Matplotlib3D绘图mplot3d,绘制3D散点图、3D线图和3D条形图,示例+代码
    bat 之 特殊字符&转义
    c++ decltype()的两个特殊情况
    黑盒测试用例设计方法案例与练习题
  • 原文地址:https://blog.csdn.net/chengxuyuanlaow/article/details/127626679