本文共 5375 字,大约阅读时间需要 17 分钟。
Spring校验器,参数校验从此简单。
应用在执行业务逻辑之前,必须通过校验保证接受到的输入数据是合法正确的,但很多时候同样的校验出现了多次,在不同的层,不同的方法上,导致代码冗余,浪费时间,违反DRY原则。
可以考虑把校验的代码封装起来,来解决出现的这些问题。
JSR-303是Java为Bean数据合法性校验提供的标准框架,它定义了一套可标注在成员变量,属性方法上的校验注解。
Hibernate Validation提供了这套标准的实现,在我们引入Spring Boot web starter或者Spring boot starter validation的时候,默认会引入Hibernate Validation。说了这么多废话,上代码。
org.springframework.boot spring-boot-starter-web org.hibernate.validator hibernate-validator org.projectlombok lombok
@Datapublic class User { // 名字不允许为空,并且名字的长度在2位到30位之间 // 如果名字的长度校验不通过,那么提示错误信息 @NotNull @Size(min=2, max=30,message = "请检查名字的长度是否有问题") private String name; // 不允许为空,并且年龄的最小值为18 @NotNull @Min(18) private Integer age;}
@SpringBootApplication@RestControllerpublic 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"; } }
稍作演示下运行的结果,可以看出校验框架已经生效了。
@Null 被注释的元素必须为 null
@NotNull 被注释的元素必须不为 null @AssertTrue 被注释的元素必须为 true @AssertFalse 被注释的元素必须为 false @Min(value) 被注释的元素必须是一个数字,其值必须大于等于指定的最小值 @Max(value) 被注释的元素必须是一个数字,其值必须小于等于指定的最大值 @DecimalMin(value) 被注释的元素必须是一个数字,其值必须大于等于指定的最小值 @DecimalMax(value) 被注释的元素必须是一个数字,其值必须小于等于指定的最大值 @Size(max=, min=) 被注释的元素的大小必须在指定的范围内 @Digits (integer, fraction) 被注释的元素必须是一个数字,其值必须在可接受的范围内 @Past 被注释的元素必须是一个过去的日期 @Future 被注释的元素必须是一个将来的日期 @Pattern(regex=,flag=) 被注释的元素必须符合指定的正则表达式Hibernate Validator提供的校验注解:
@NotBlank(message =) 验证字符串非null,且长度必须大于0 @Email 被注释的元素必须是电子邮箱地址 @Length(min=,max=) 被注释的字符串的大小必须在指定的范围内 @NotEmpty 被注释的字符串的必须非空 @Range(min=,max=,message=) 被注释的元素必须在合适的范围内有时候,第三方库中并没有我们想要的校验类型,好在系统提供了很好的扩展能力,我们可以自定义检验。
比如,我们想校验用户的手机格式,写手机号码校验器// 我们可以直接拷贝系统内的注解如@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 [] 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); } } } }
@Datapublic class User { @NotNull @Size(min=2, max=30,message = "请检查名字的长度是否有问题") private String name; @NotNull @Min(18) private Integer age; //这里是新添加的注解奥 @IsMobile private String phone;}
测试
可以看出自定义的注解已经生效了。
我们还可以继续优化的地方,新建一个全局的异常,如果校验失败的话,抛出全局的业务异常,捕获业务异常,然后返回用户友好的提示信息。也可以通过方法的校验。
@Validated@RestController@SpringBootApplicationpublic 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){ HashSetmessageSet = new HashSet(); for (ConstraintViolation constraintViolation : cve.getConstraintViolations()) { messageSet.add(constraintViolation.getMessage()); } return messageSet; }}
通过使用校验器,所有的控制器,我们都不用再去做校验啦,代码再回看是不是清爽很多。我们写代码很简答,但是一定要想到如何把代码写的更简单,更清晰,更利于维护,写重复的代码是在浪费自己的时间奥。
以后再碰到参数校验的情况,首先想到的不是直接就去校验,可以查找自己是否写过某一类的验证器,可以直接拿来即用。
希望能帮助大家。
我在知识星球开通了一个交流群,免费的。希望能和大家一起交流学习,进群的可以获得500G Java学习资料奥,多年的收藏,除了Java之外,还收藏了IOS,Linux,Android相关资料。想要的也可以找我拿。
转载地址:http://hcvwo.baihongyu.com/