每天早上七点三十,准时推送干货
最近很多读者给阿粉留言,说怎么好久没看到我的文章了,这里说一下。
由于公众号不再按时间线排序,所以你会发现有时候能看到几天前的文章,这不是出BUG,是公众号的一次改变。
至于排序的具体标准是啥,阿粉也不太清楚,大概和你打开某个公众号的频率有关。
所以如果你想第一时间收到阿粉的文章,可以点击Java极客技术的的头像,再点右上角三个点,进去设置一下【星标】。
一、介绍
关于参数合法性验证的重要性就不多说了,即使前端对参数做了基本验证以外,后端依然还需要进行验证,以防不合规的数据直接进入后端,严重的甚至会造成系统直接崩溃!
本文结合自己在项目中的实际使用经验,主要以实用为主,对数据合法性验证做一次总结,不了解的朋友可以学习一下,同时可以立马实践到项目上去。
下面我们通过几个示例来演示如何判断参数是否合法,不多说直接开撸!
二、断言验证
对于参数的合法性验证,最初的做法比较简单,自定义一个异常类。
public class CommonException extends RuntimeException {
/**错误码*/
private Integer code;
/**错误信息*/
private String msg;
//...set/get
public CommonException(String msg) {
super(msg);
this.msg = msg;
}
public CommonException(String msg, Throwable cause) {
super(msg, cause);
this.msg = msg;
}
}
当判断某个参数不合法的时候,直接抛异常!
@RestController
public class HelloController {
@RequestMapping("/upload")
public void upload(MultipartFile file) {
if (file == null) {
throw new CommonException("请选择上传文件!");
}
//.....
}
}
然后写一个统一异常拦截器,对抛异常的程序进行处理。
这种做法比较直观,如果当前参数既要判断是否为空,又要判断长度是否超过最大长度的时候,代码就显得有点多了!
于是,程序界的大佬想到了一个更加优雅又能节省代码的方式,创建一个断言类工具类,专门用来判断参数的是否合法,如果不合法,就抛异常!
/**
* 断言工具类
*/
public abstract class LocalAssert {
public static void isTrue(boolean expression, String message) throws CommonException {
if (!expression) {
throw new CommonException(message);
}
}
public static void isStringEmpty(String param, String message) throws CommonException{
if(StringUtils.isEmpty(param)) {
throw new CommonException(message);
}
}
public static void isObjectEmpty(Object object, String message) throws CommonException {
if (object == null) {
throw new CommonException(message);
}
}
public static void isCollectionEmpty(Collection coll, String message) throws CommonException {
if (coll == null || (coll.size() == 0)) {
throw new CommonException(message);
}
}
}
当我们需要对参数进行验证的时候,直接通过这个类就可以完成基本操作,方式如下:
@RestController
public class HelloController {
@RequestMapping("/save")
public void save(String name, String email) {
LocalAssert.isStringEmpty(name, "用户名不能为空!");
LocalAssert.isStringEmpty(email, "邮箱不能为空!");
//.....
}
}
相比上个步骤,当要判断的参数比较多时,代码明显简洁多了!
类似这样的工具类,spring
也提供了一个名为Assert
的断言工具类,在开发的时候,可以直接使用!
![](https://i-blog.csdnimg.cn/blog_migrate/5cfb44dbb8f60acc7e92ae0a7f9fda4e.png)
三、注解验证
使用注解对数据进行合法性验证,可以说是 java 界一项非常伟大的创新,使用这种方式不仅使的代码变得很简洁,而且阅读起来非常令人赏心悦目!
3.1、依赖包引入
下面我们一起来看看具体的实践方式,以Spring Boot
工程为例,如果需要使用注解校验,直接引入spring-boot-starter-web
依赖包即可,会自动将注解验证相关的依赖包打入工程!
<!-- spring boot web -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
下面在创建实体类的时候,还会用到lombok
插件,因此还需要引入lombok
依赖包!
<!-- lombok -->
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.4</version>
<scope>provided</scope>
</dependency>
如果是普通的Java
工程,引入以下几个依赖包即可!
<dependency>
<groupId>org.hibernate.validator</groupId>
<artifactId>hibernate-validator</artifactId>
<version>6.0.9.Final</version>
</dependency>
<dependency>
<groupId>javax.el</groupId>
<artifactId>javax.el-api</artifactId>
<version>3.0.0</version>
</dependency>
<dependency>
<groupId>org.glassfish.web</groupId>
<artifactId>javax.el</artifactId>
<version>2.2.6</version>
</dependency>
3.2、注解校验请求对象
紧接着我们来创建一个实体User
,用于模拟用户注册时的请求实体对象!
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
public class User {
@NotBlank(message = "用户名不能为空!")
private String userName;
@Email(message = "邮箱格式不正确")
@NotBlank(message = "邮箱不能为空!")
private String email;
@NotBlank(message = "密码不能为空!")
@Size(min = 8, max = 16,message = "请输入长度在8~16位的密码")
private String userPwd;
@NotBlank(message = "确认密码不能为空!")
private String confirmPwd;
}
在web
层创建一个register()
注册接口方法,同时在请求参数上添加@Valid
,如下:
@RestController
public class UserController {
@RequestMapping("/register")
public boolean register(@RequestBody @Valid User user){
if(!user.getUserPwd().equals(user.getConfirmPwd())){
throw new CommonException("确认密码与密码不相同,请确认!");
}
//业务处理...
return true;
}
}
最后自定义一个异常全局处理器,用于处理异常消息,如下:
@Slf4j
@Configuration
public class GlobalWebMvcConfig implements WebMvcConfigurer {
/**
* 统一异常处理
* @param resolvers
*/
@Override
public void configureHandlerExceptionResolvers(List<HandlerExceptionResolver> resolvers) {
resolvers.add(new HandlerExceptionResolver() {
@Override
public ModelAndView resolveException(HttpServletRequest request, HttpServletResponse response, Object o, Exception e) {
log.error("【统一异常拦截】请求出现异常,内容如下:",e);
ModelAndView mv = new ModelAndView(new MappingJackson2JsonView());
String uri = request.getRequestURI();
if(e instanceof CommonException){
//CommonExecption为自定义异常类抛出的异常
printWrite(((CommonException) e).getMsg(),((CommonException) e).getData(), uri, mv);
} else if(e instanceof MethodArgumentNotValidException){
//MethodArgumentNotValidException为注解校验异常类
//获取注解校验异常信息
String error = ((MethodArgumentNotValidException) e).getBindingResult().getFieldError().getDefaultMessage();
printWrite(error,null, uri, mv);
} else {
printWrite(e.getMessage(),null, uri, mv);
}
return mv;
}
});
}
/**
* 异常封装相应结果
* @param object
*/
private void printWrite(String msg, Object object, String uri, ModelAndView mv){
ResResult resResult = new ResResult(uri, object);
if(msg != null){resResult.setMsg(msg);}
if(log.isDebugEnabled()){
log.debug("【response】异常输出结果:" + JSONObject.toJSONString(resResult, SerializerFeature.WriteMapNullValue));
}
Map resultMap = BeanToMapUtil.beanToMap(resResult);
mv.addAllObjects(resultMap);
}
}
下面我们启动项目,使用postman
来测试一把,看看效果如何?
测试字段是否为空
![](https://i-blog.csdnimg.cn/blog_migrate/24ed924331716dd0d3abe3002bdc3a96.png)
测试邮箱是否合法
![](https://i-blog.csdnimg.cn/blog_migrate/1e9ee2c18e315bbcefbd9be7d139d759.png)
测试密码长度是否符合要求
![](https://i-blog.csdnimg.cn/blog_migrate/71a83a813768b9c5991a538256efd719.png)
测试密码与确认密码是否相同
![](https://i-blog.csdnimg.cn/blog_migrate/b3d738c0b902f8dc7178750c9089e0b9.png)
3.3、注解校验请求参数
上面我们介绍了请求对象的验证方式,那如果直接在方法上对请求参数进行验证是否同样有效呢?
为了眼见为实,下面我们就来模拟在方法上对请求参数进行验证,看看结果如何。
新建一个查询接口query
,如下
@RestController
public class UserController {
@PostMapping("/query")
public boolean query(@RequestParam("userId") @Valid @NotBlank(message = "用户ID不能为空") String userId ){
return true;
}
}
使用postman
请求试一试,默认给userId
参数为null
,结果如下:
![](https://i-blog.csdnimg.cn/blog_migrate/8e9e4d14649178fb48a6cd560eaaa4af.png)
很清晰的看到,query()
方法中的参数注解验证无效!
当我们在UserController
类上加上@Validated
注解!
@RestController
@Validated
public class UserController {
@PostMapping("/query")
public boolean query(@RequestParam("userId") @Valid @NotBlank(message = "用户ID不能为空") String userId ){
return true;
}
}
使用postman
请求再试一试,结果如下!
![](https://i-blog.csdnimg.cn/blog_migrate/302f55c62a32dc14218ddf913279f9bc.png)
很清晰的看到,注解进行了验证,同时还抛出异常ConstraintViolationException
!
![](https://i-blog.csdnimg.cn/blog_migrate/b71a33f8a75b675c7a1b2c12bc962f0d.png)
@Validated
参数作用于类上时,表示告诉Spring
可以对方法中请求参数进行校验!
所有在实际开发的时候,我们可以使用@Validated
和@Valid
注解的组合来对方法中的请求参数和请求对象进行校验!
同时,@Validated
和@Valid
注解不仅仅只是验证控制器级别,可以验证任何Spring
组件,例如Service
层方法入参的验证!
@Service
@Validated
public class UserService {
public void saveUser(@Valid User user){
//dao插入
}
}
3.4、自定义注解验证
默认的情况下,依赖包已经给我们提供了非常多的校验注解,如下!
JSR提供的校验注解!
![](https://i-blog.csdnimg.cn/blog_migrate/bb2712ff47828ee41d13a331b411a8b4.png)
Hibernate Validator提供的校验注解
![](https://i-blog.csdnimg.cn/blog_migrate/2e264640886ae26c5609f182238ac570.png)
但是某些情况,例如性别这个参数可能需要我们自己去验证,同时我们也可以自定义一个注解来完成参数的校验,实现方式如下!
新创建一个
Sex
注解,其中SexValidator
类指的是具体的参数验证类
@Target({FIELD})
@Retention(RUNTIME)
@Constraint(validatedBy = SexValidator.class)
@Documented
public @interface Sex {
String message() default "性别值不在可选范围内";
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
}
SexValidator
类,实现自ConstraintValidator
接口
public class SexValidator implements ConstraintValidator<Sex, String> {
@Override
public boolean isValid(String value, ConstraintValidatorContext context) {
Set<String> sexSet = new HashSet<String>();
sexSet.add("男");
sexSet.add("女");
return sexSet.contains(value);
}
}
最后在User
实体类上加入一个性别参数,使用自定义注解进行校验!
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
public class User {
@NotBlank(message = "用户名不能为空!")
private String userName;
@Email(message = "邮箱格式不正确")
@NotBlank(message = "邮箱不能为空!")
private String email;
@NotBlank(message = "密码不能为空!")
@Size(min = 8, max = 16,message = "请输入长度在8~16位的密码")
private String userPwd;
@NotBlank(message = "确认密码不能为空!")
private String confirmPwd;
/**
* 自定义注解校验
*/
@Sex(message = "性别输入有误!")
private String sex;
}
使用postman
来请求试一试,结果如下!
不传
sex
参数
![](https://i-blog.csdnimg.cn/blog_migrate/0db7b47f9970b7bccbc5ca9e065927f9.png)
很清晰的看到,已经生效!
3.5、手动进行注解校验
某些时候呢,假如有100个类需要用到校验注解,此时我们可能在每个类会加上注解@Validated
或者@Valid
,再增加100个这样的类,就会造成很多大量的重复工作。
而此时,我们的诉求是想对有校验注解的实体类进行全局参数验证!
解决办法就会用到Validator
提供的手动注解校验证工具类,实现方法如下!
新建一个注解验证工具类
/**
* 注解校验工具类
*/
public class ValidatorUtils {
/**
* 获取对象中所有注解校验证异常信息
* @param object
* @return
*/
public static String validated(Object object){
List<String> errorMessageList = new ArrayList<>();
//获取注解校验工厂
ValidatorFactory factory = Validation.buildDefaultValidatorFactory();
Validator validator = factory.getValidator();
Set<ConstraintViolation<Object>> violations = validator.validate(object);
for (ConstraintViolation<Object> constraintViolation : violations) {
errorMessageList.add(constraintViolation.getMessage());
}
return errorMessageList.toString();
}
}
使用
ValidatorUtils
工具类,对参数进行验证
@Test
public void testUser(){
User user = new User();
System.out.println(ValidatorUtils.validated(user));
}
执行之后,结果如下!
[邮箱不能为空!, 用户名不能为空!, 密码不能为空!, 确认密码不能为空!, 性别输入有误!]
当然你还可以对ValidatorUtils
类进行改造,当有异常信息的时候,直接抛异常!
同时,你还可以通过@Autowired
直接注入的方式来获取Validator
对象!
@Autowired
Validator validator
3.6、spring 注解校验原理
如果你对springmvc
的方法参数解析器(HandlerMethodArgumentResolver
)了解的话,就可能会想到参数校验这块肯定是在对应的方法参数解析器里执行的。
直接定位到resolveArgument
这个方法,先通过WebDataBinder
进行入参属性绑定,然后再进行校验!
![](https://i-blog.csdnimg.cn/blog_migrate/bc2456401b8076001899142b21f5d1b7.png)
validateIfApplicable
方法逻辑,会遍历当前参数methodParam
所有的注解,如果注解是@Validated
或者注解的名字以Valid
开头,则使用WebDataBinder
对象执行校验逻辑。
![](https://i-blog.csdnimg.cn/blog_migrate/77c80e84268044b02f213b78996998ad.png)
方法参数解析器只针对接口请求时入参进行验证,如果想对任何组件中方法进行注解校验,似乎还缺了点什么!
而当需要对一个类中的方法参数使用注解校验时,在类上加上@Validated
就是为了告诉Spring
去校验方法参数!
底层核心是通过切面代理类并配合MethodValidationPostProcessor
这个后置处理器进行处理!
![](https://i-blog.csdnimg.cn/blog_migrate/728f2518dcec59c0d30a4ddb611e5cb6.png)
四、总结
参数验证,在开发中使用非常频繁,如何优雅的进行验证,让代码变得更加可读,是业界大佬一直在追求的目标!
本文主要是对自己在项目中的实际使用到参数验证方式加一整理,希望能帮助到各位网友!
五、参考
1、SpringMVC源码
2、JavaGuide - 如何在 Spring/Spring Boot 中做参数校验?[1]
3、胡峻峥 - SpringMvc @Validated注解执行原理[2]
参考资料
[1]
JavaGuide - 如何在 Spring/Spring Boot 中做参数校验?: https://juejin.im/post/5dc8bc745188254e7a155ba0#heading-14
[2]胡峻峥 - SpringMvc @Validated注解执行原理: https://www.cnblogs.com/hujunzheng/p/12570921.html
< END >
如果大家喜欢我们的文章,欢迎大家转发,点击在看让更多的人看到。也欢迎大家热爱技术和学习的朋友加入的我们的知识星球当中,我们共同成长,进步。
往期精彩回顾