参数校验
项目使用 Hibernate Validator 框架,对 RESTful API 接口进行参数的校验,以保证最终数据入库的正确性。例如说,用户注册时,会校验手机格式的正确性,密码非弱密码。
如果参数校验不通过,会抛出 ConstraintViolationException 异常,被全局的异常处理捕获,返回“请求参数不正确”的响应。示例如下:
{
  "code": 400,
  "data": null,
  "msg": "请求参数不正确:密码不能为空"}#1. 参数校验注解
Validator 内置了 20+ 个参数校验注解,整理成常用与不常用的注解。
#1.1 常用注解
| 注解 | 功能 | 
|---|---|
| @NotBlank | 只能用于字符串不为 null,并且字符串#trim()以后 length 要大于 0 | 
| @NotEmpty | 集合对象的元素不为 0 ,即集合不为空,也可以用于字符串不为 null | 
| @NotNull | 不能为 null | 
| @Pattern(value) | 被注释的元素必须符合指定的正则表达式 | 
| @Max(value) | 该字段的值只能小于或等于该值 | 
| @Min(value) | 该字段的值只能大于或等于该值 | 
| @Range(min=, max=) | 检被注释的元素必须在合适的范围内 | 
| @Size(max, min) | 检查该字段的 size是否在min和max之间,可以是字符串、数组、集合、Map 等 | 
| @Length(max, min) | 被注释的字符串的大小必须在指定的范围内。 | 
| @AssertFalse | 被注释的元素必须为 true | 
| @AssertTrue | 被注释的元素必须为 false | 
| @Email | 被注释的元素必须是电子邮箱地址 | 
| @URL(protocol=,host=,port=,regexp=,flags=) | 被注释的字符串必须是一个有效的 URL | 
#1.2 不常用注解
| 注解 | 功能 | 
|---|---|
| @Null | 必须为 null | 
| @DecimalMax(value) | 被注释的元素必须是一个数字,其值必须小于等于指定的最大值 | 
| @DecimalMin(value) | 被注释的元素必须是一个数字,其值必须大于等于指定的最小值 | 
| @Digits(integer, fraction) | 被注释的元素必须是一个数字,其值必须在可接受的范围内 | 
| @Positive | 判断正数 | 
| @PositiveOrZero | 判断正数或 0 | 
| @Negative | 判断负数 | 
| @NegativeOrZero | 判断负数或 0 | 
| @Future | 被注释的元素必须是一个将来的日期 | 
| @FutureOrPresent | 判断日期是否是将来或现在日期 | 
| @Past | 检查该字段的日期是在过去 | 
| @PastOrPresent | 判断日期是否是过去或现在日期 | 
| @SafeHtml | 判断提交的 HTML 是否安全。例如说,不能包含 JavaScript 脚本等等 | 
#2. 参数校验使用
只需要三步,即可开启参数校验的功能。
〇 第零步,引入参数校验的 spring-boot-starter-validation 依赖。一般不需要做,项目默认已经引入。
① 第一步,在需要参数校验的类上,添加 @Validated 注解,例如说 Controller、Service 类。代码如下:
// Controller 示例@Validatedpublic class AuthController {}// Service 示例,一般放在实现类上@Service@Validatedpublic class AdminAuthServiceImpl implements AdminAuthService {}② 第二步(情况一)如果方法的参数是 Bean 类型,则在方法参数上添加 @Valid 注解,并在 Bean 类上添加参数校验的注解。代码如下:
// Controller 示例@Validatedpublic class AuthController {
    @PostMapping("/login")
    public CommonResult<AuthLoginRespVO> login(@RequestBody @Valid AuthLoginReqVO reqVO) {}
    }// Service 示例,一般放在接口上public interface AdminAuthService {
    
    String login(@Valid AuthLoginReqVO reqVO, String userIp, String userAgent);}// Bean 类的示例。一般建议添加参数注解到属性上。原因:采用 Lombok 后,很少使用 getter 方法public class AuthLoginReqVO {
    @NotEmpty(message = "登录账号不能为空")
    @Length(min = 4, max = 16, message = "账号长度为 4-16 位")
    @Pattern(regexp = "^[A-Za-z0-9]+$", message = "账号格式为数字以及字母")
    private String username;
    @NotEmpty(message = "密码不能为空")
    @Length(min = 4, max = 16, message = "密码长度为 4-16 位")
    private String password;
    }② 第二步(情况二)如果方法的参数是普通类型,则在方法参数上直接添加参数校验的注解。代码如下:
// Controller 示例@Validatedpublic class DictDataController {
    @GetMapping(value = "/get")
    public CommonResult<DictDataRespVO> getDictData(@RequestParam("id") @NotNull(message = "编号不能为空") Long id) {}
    }// Service 示例,一般放在接口上public interface DictDataService {
    DictDataDO getDictData(@NotNull(message = "编号不能为空") Long id);
    }③ 启动项目,模拟调用 RESTful API 接口,少填写几个参数,看看参数校验是否生效。
疑问:Controller 做了参数校验后,Service 是否需要做参数校验?
是需要的。Service 可能会被别的 Service 进行调用,也会存在参数不正确的情况,所以必须进行参数校验。
#3. 自定义注解
如果 Validator 内置的参数校验注解不满足需求时,我们也可以自定义参数校验的注解。
在项目的 yudao-common 的 validation 包下,就自定义了多个参数校验的注解,以 @Mobile 注解来举例,它提供了手机格式的校验。
① 第一步,新建 @Mobile 注解,并设置自定义校验器为 MobileValidator 类。代码如下:
@Target({
        ElementType.METHOD,
        ElementType.FIELD,
        ElementType.ANNOTATION_TYPE,
        ElementType.CONSTRUCTOR,
        ElementType.PARAMETER,
        ElementType.TYPE_USE})@Retention(RetentionPolicy.RUNTIME)@Documented@Constraint(
        validatedBy = MobileValidator.class // 设置校验器)public @interface Mobile {
    String message() default "手机号格式不正确";
    Class<?>[] groups() default {};
    Class<? extends Payload>[] payload() default {};}② 第二步,新建 MobileValidator 校验器。代码如下:
public class MobileValidator implements ConstraintValidator<Mobile, String> {
    @Override
    public void initialize(Mobile annotation) {
    }
    @Override
    public boolean isValid(String value, ConstraintValidatorContext context) {
        // 如果手机号为空,默认不校验,即校验通过
        if (StrUtil.isEmpty(value)) {
            return true;
        }
        // 校验手机
        return ValidationUtils.isMobile(value);
    }}③ 第三步,在需要手机格式校验的参数上添加 @Mobile 注解。示例代码如下:
public class AppAuthLoginReqVO {
    @NotEmpty(message = "手机号不能为空")
    @Mobile // <=== here
    private String mobile;}#4. 更多使用文档
更多关于 Validator 的使用,可以系统阅读 《芋道 Spring Boot 参数校验 Validation 入门 》 文章。
例如说,手动参数校验、分组校验、国际化 i18n 等等。
版权声明
本文仅代表作者观点,不代表xx立场。
本文系作者授权xxx发表,未经许可,不得转载。


 
		
发表评论