Spring中校验器(Validator)的深入讲解

 
  org.springframework.boot
  spring-boot-starter-web
 
 
  org.hibernate.validator
  hibernate-validator
 
  
 
  org.projectlombok
  lombok
  

@Data
public class User {
 // 名字不允许为空，并且名字的长度在2位到30位之间
 // 如果名字的长度校验不通过，那么提示错误信息
 @NotNull
 @Size(min=2, max=30,message = "请检查名字的长度是否有问题")
 private String name;

 // 不允许为空，并且年龄的最小值为18
 @NotNull
 @Min(18)
 private Integer age;
}

@SpringBootApplication
@RestController
public class UserApplication{
 public static void main(String[] args) {
 SpringApplication.run(UserApplication.class,args);
 }
 
 // 1. 要校验的参数前，加上@Valid注解
 // 2. 紧随其后的，跟上一个BindingResult来存储校验信息
 @RequestMapping("/test1")
 public Object test1(
  @Valid User user,
  BindingResult bindingResult
 ) {
 //如果检验出了问题，就返回错误信息
 // 这里我们返回的是全部的错误信息，实际中可根据bindingResult的方法根据需要返回自定义的信息。
 // 通常的解决方案为：JSR-303 + 全局ExceptionHandler
 if (bindingResult.hasErrors()){
  return bindingResult.getAllErrors();
 }
 return "OK";
 } 
}

// 我们可以直接拷贝系统内的注解如@Min，复制到我们新的注解中，然后根据需要修改。
@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER})
@Retention(RUNTIME)
@Documented
//注解的实现类。
@Constraint(validatedBy = {IsMobileValidator.class})
public @interface IsMobile {
 //校验错误的默认信息
 String message() default "手机号码格式有问题";
 //是否强制校验
 boolean isRequired() default false;
 Class<?>[] groups() default {};
 Class<? extends Payload>[] payload() default {};
}

// 自定义注解一定要实现ConstraintValidator接口奥，里面的两个参数
// 第一个为 具体要校验的注解
// 第二个为 校验的参数类型
public class IsMobileValidator implements ConstraintValidator {

 private boolean required = false;

 private static final Pattern mobile_pattern = Pattern.compile("1\\d{10}");
 //工具方法，判断是否是手机号
 public static boolean isMobile(String src) {
  if (StringUtils.isEmpty(src)) {
   return false;
  }
  Matcher m = mobile_pattern.matcher(src);
  return m.matches();
 }

 @Override
 public void initialize(IsMobile constraintAnnotation) {
  required = constraintAnnotation.isRequired();
 }

 @Override
 public boolean isValid(String phone, ConstraintValidatorContext constraintValidatorContext) {
  //是否为手机号的实现
  if (required) {
   return isMobile(phone);
  } else {
   if (StringUtils.isEmpty(phone)) {
    return true;
   } else {
    return isMobile(phone);
   }
  }
 }
}

@Data
public class User {
 @NotNull
 @Size(min=2, max=30,message = "请检查名字的长度是否有问题")
 private String name;
 @NotNull
 @Min(18)
 private Integer age;
 //这里是新添加的注解奥
 @IsMobile
 private String phone;
}

@Validated
@RestController
@SpringBootApplication
public class UserApplication{
 public static void main(String[] args) {
  SpringApplication.run(UserApplication.class,args);
 }

 @RequestMapping("/test2")
 public String test2(
   @IsMobile String phone

 ){
  return phone + "ok";
 }

 @ExceptionHandler(ConstraintViolationException.class)
 @ResponseBody
 public Object handleConstraintViolationException(ConstraintViolationException cve){

  HashSet messageSet = new HashSet();
  for (ConstraintViolation constraintViolation : cve.getConstraintViolations()) {
   messageSet.add(constraintViolation.getMessage());
  }
  return messageSet;
 }
}