前言
数据的校验的重要性就不用说了,即使在前端对数据进行校验的情况下,我们还是要对传入后端的数据再进行一遍校验,避免用户绕过浏览器直接通过一些 HTTP 工具直接向后端请求一些违法数据。
最普通的做法就像下面这样。我们通过
if/else
语句对请求的每一个参数一一校验。
1
@RestController
@RequestMapping("/api/person")
public class PersonController {
@PostMapping
public ResponseEntity<PersonRequest> save(@RequestBody PersonRequest personRequest) {
if (personRequest.getClassId() == null
|| personRequest.getName() == null
|| !Pattern.matches("(^Man$|^Woman$|^UGM$)", personRequest.getSex())) {
return ResponseEntity.ok().body(personRequest);
这样的代码,小伙伴们在日常开发中一定不少见,很多开源项目都是这样对请求入参做校验的。
但是,不太建议这样来写,这样的代码明显违背了 单一职责原则 。大量的非业务代码混杂在业务代码中,非常难以维护,还会导致业务层代码冗杂!
实际上,我们是可以通过一些简单的手段对上面的代码进行改进的!这也是本文主要要介绍的内容!
注意:本文主要讲解基于Sping Boot的注解校验,Spring Boot版本依赖如下:
1
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.7.1</version>
<relativePath/> <!-- lookup parent from repository -->
</parent>
添加相关依赖
基于 Spring Boot 的话,只需要给项目添加上下面这些依赖:
1
<dependencies>
<!--参数校验依赖-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-validation</artifactId>
</dependency>
<!--测试依赖-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<version>2.2.4.RELEASE</version>
<scope>test</scope>
</dependency>
<!--Spring Web依赖-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<!--lombok工具-->
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<optional>true</optional>
</dependency>
</dependencies>
在Spring Boot 2.3 1 之前,
spring-boot-starter-validation
包括在了
spring-boot-starter-web
中,但如果你使用的Spring Boot版本大于2.3.1,比如我当前使用的是2.7.1,那么就必须手动添加依赖
spring-boot-starter-validation
验证 Controller 的输入
验证请求体
验证请求体即是验证被
@RequestBody
注解标记的方法参数。
我们在需要验证的参数上加上了
@Valid
注解,如果验证失败,它将抛出
MethodArgumentNotValidException
。默认情况下,Spring 会将此异常转换为 HTTP Status 400(错误请求)。
PersonController
1
@RestController
@RequestMapping("/api/person")
public class PersonController {
@PostMapping
public ResponseEntity<PersonRequest> save(@RequestBody @Valid PersonRequest personRequest) {
return ResponseEntity.ok().body(personRequest);
注意:这里开启Spring数据校验使用@Validated也可以
PersonRequest
我们使用校验注解对请求的参数进行校验!
1
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class PersonRequest {
@NotNull(message = "classId 不能为空")
private String classId;
@Size(max = 33)
@NotNull(message = "name 不能为空")
private String name;
@Pattern(regexp = "(^Man$|^Woman$|^UGM$)", message = "sex 值不在可选范围")
@NotNull(message = "sex 不能为空")
private String sex;
正则表达式说明:
-
^string
: 匹配以 string 开头的字符串 -
string$
:匹配以 string 结尾的字符串 -
^string$
:精确匹配 string 字符串 -
(^Man$|^Woman$|^UGM$)
: 值只能在 Man,Woman,UGM 这三个值中选择
自定义全局异常处理器捕获数据校验异常
自定义异常处理器可以帮助我们捕获异常,并进行一些简单的处理。
GlobalExceptionHandler
1
@Slf4j
@RestControllerAdvice
public class GlobalExceptionHandler {
* 处理参数校验失败异常
* @param exception 异常类
* @return 响应
@ResponseStatus(HttpStatus.BAD_REQUEST)
@ExceptionHandler(MethodArgumentNotValidException.class)
public ResultBean exceptionHandler(MethodArgumentNotValidException exception){
//我们主要获取这个接口BindingResult的数据,它就包含了我们使用@RequestBody绑定的参数的所有信息,无论是校验异常错误信息还是JavaBean参数的属性信息
BindingResult bindingResult = exception.getBindingResult();
Map<String, String> errorMap = new HashMap<>();
StringBuffer buffer = new StringBuffer();
if(bindingResult.getFieldErrors() != null){
for (FieldError fieldError : bindingResult.getFieldErrors()) {
String field = fieldError.getField();
Object rejectedValue = fieldError.getRejectedValue();
String defaultMessage = fieldError.getDefaultMessage();
errorMap.put(field, defaultMessage);
String msg = String.format("错误字段:%s, 错误值:%s, 原因:%s", field, rejectedValue, defaultMessage);
buffer.append(msg);
log.warn("错误字段:[{}], 错误值:[{}], 原因:[{}]", field, rejectedValue, defaultMessage);
return ResultBean.error(buffer.toString(), errorMap, 400);
通过Postman测试验证
验证成功的情况
验证失败的情况
顺便一提,如何在PostMan发送请求体json数据,也就是说后端用@RequestBody接收的参数:
-
设置请求头
Content-Type:application/json
,content-type首字母小写也是可行的
- 传递json参数
验证请求参数
这些参数通常被
@PathVariable
以及
@RequestParam
标记,并且相对于JavaBean的参数,我们往往将其称为平铺参数
注意:这里适用@Valid注解是不行的,因为它要求待校验的入参是JavaBean,所以如果需要校验平铺参数,请使用@Validated开启Spring自动参数校验
PersonController
1
@RestController
@RequestMapping("/api/person")
@Validated
public class PersonController {
@GetMapping("/{id}")
public ResponseEntity<Integer> getPersonByID(@PathVariable("id") @Max(value = 5, message = "超过 id 的范围了") Integer id) {
return ResponseEntity.ok().body(id);
@PutMapping("/{name}")
public ResponseEntity<String> getPersonByName(@RequestParam("name") @Size(max = 6, message = "超过 name 的范围了") String name) {
return ResponseEntity.ok().body(name);
ExceptionHandler
1
* 处理平铺参数校验失败
@ExceptionHandler(ConstraintViolationException.class)
public ResultBean exceptionHandler(ConstraintViolationException exception){
log.warn(exception.getMessage());
return ResultBean.error(exception.getMessage(), 400);
通过Postman测试验证
验证成功的情况
验证失败的情况
验证 Service 中的方法
我们不仅可以使用@Validated和@Valid验证Controller组件,也可以验证其他Spring管理的组件,比如Service, 不过Controller一般不提供接口,而Service一般是面向接口编程 ,而这个地方有坑,需要注意下面几点:
-
在实现类中重定义接口方法的参数校验配置会失败且会报错:
javax.validation.ConstraintDeclarationException: HV000151: A method overriding another method must not redefine the parameter constraint configuration
,这个异常信息也告诉我们:参数的校验配置应该写在接口方法中,并且实现类不能修改配置,要么保持一样,要么可以不用写参数校验配置 -
在非Controller组件中,像Service,必须组合使用@Validated和@Valid,其中@Validated作为类注解、@Valid作为方法参数注解javaBean,这样参数校验才会生效,并且它产生的异常是
ConstraintViolationException
,这个跟之前Controller中的平铺参数校验产生的异常是相同的,这个异常没有继承BindException
接口,相对而言它的错误不好像BindException
和MethodArgumentNotValidException
那样处理 - 如果方法参数是平铺参数,那么只要加@Validated就行了 ```java @Service @Validated public class PersonServiceImpl implements PersonService {
@Override public PersonRequest insertPerson(@NotNull @Min(10) Integer id, @NotNull String name) { return null; } }
1
4. @Validated可以放在接口中,也可以放在实现类中,不过我一般放在实现类中
**PersonService**
```java
public interface PersonService {
PersonRequest insertPerson(@Valid PersonRequest person);
PersonServiceImpl
1
@Service
@Validated
public class PersonServiceImpl implements PersonService {
@Override
public PersonRequest insertPerson(PersonRequest person) {
return person;
ExceptionHandler
1
@ExceptionHandler(ConstraintViolationException.class)
public ResultBean exceptionHandler(ConstraintViolationException exception){
log.warn(exception.getMessage());
return ResultBean.error(exception.getMessage(), 400);
通过Postman测试验证
参数校验失败
级联校验
级联校验关键点在于@Valid,级联校验的意思是JavaBean内部有其他JavaBean需要验证,那么这个JavaBean就需要加@Valid注解,并且只能用@Valid,因为它可以标记字段,@Validatd不行
PersonRequest
1
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class PersonRequest {
@NotNull(message = "classId 不能为空")
private String classId;
@Pattern(regexp = "(^Man$|^Woman$|^UGM$)", message = "sex 值不在可选范围")
@NotNull(message = "sex 不能为空")
private String sex;
@Valid //让InnerChild的属性也参与校验
@NotNull
private InnerChild child; //内部的JavaBean
@Getter
@Setter
@ToString
public static class InnerChild {
@Size(max = 33)
@NotNull(message = "name 不能为空")
private String name;
@NotNull(message = "年龄不能为空")
@Positive(message = "年龄只能为正数")
private Integer age;
Validator 编程方式手动进行参数验证
某些场景下可能会需要我们手动校验并获得校验结果。
我们通过
Validator
工厂类获得的
Validator
示例。另外,如果是在 Spring Bean 中的话,还可以通过
@Autowired
直接注入的方式。
1
@Autowired
Validator validate
具体使用情况如下:
1
* 手动校验对象
@Test
public void check_person_manually() {
ValidatorFactory factory = Validation.buildDefaultValidatorFactory();
Validator validator = factory.getValidator();
PersonRequest personRequest = PersonRequest.builder().sex("Man22")
.classId("82938390").build();
Set<ConstraintViolation<PersonRequest>> violations = validator.validate(personRequest);
violations.forEach(constraintViolation -> System.out.println(constraintViolation.getMessage()));
输出结果如下:
1
sex 值不在可选范围
name 不能为空
自定义 Validator(实用)
如果自带的校验注解无法满足你的需求的话,你还可以自定义实现注解。
案例一:校验特定字段的值是否在可选范围
比如我们现在多了这样一个需求:
PersonRequest
类多了一个
Region
字段,
Region
字段只能是
China
、
China-Taiwan
、
China-HongKong
这三个中的一个。
第一步,你需要创建一个注解
Region
。
1
@Target({FIELD})
@Retention(RUNTIME)
@Constraint(validatedBy = RegionValidator.class)
@Documented
public @interface Region {
String message() default "Region 值不在可选范围内";
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
第二步,你需要实现
ConstraintValidator
接口,并重写
isValid
方法。
1
public class RegionValidator implements ConstraintValidator<Region, String> {
@Override
public boolean isValid(String value, ConstraintValidatorContext context) {
HashSet<Object> regions = new HashSet<>();
regions.add("China");
regions.add("China-Taiwan");
regions.add("China-HongKong");
return regions.contains(value);
现在你就可以使用这个注解:
1
@Region
private String region;
通过测试验证
1
PersonRequest personRequest = PersonRequest.builder()
.region("Shanghai").build();
mockMvc.perform(post("/api/person")
.contentType(MediaType.APPLICATION_JSON)
.content(objectMapper.writeValueAsString(personRequest)))
.andExpect(MockMvcResultMatchers.jsonPath("region").value("Region 值不在可选范围内"));
使用
Postman
验证
案例二:校验电话号码
校验我们的电话号码是否合法,这个可以通过正则表达式来做,相关的正则表达式都可以在网上搜到,你甚至可以搜索到针对特定运营商电话号码段的正则表达式。
PhoneNumber.java
1
@Documented
@Constraint(validatedBy = PhoneNumberValidator.class)
@Target({FIELD, PARAMETER})
@Retention(RUNTIME)
public @interface PhoneNumber {
String message() default "Invalid phone number";
Class[] groups() default {};
Class[] payload() default {};
PhoneNumberValidator.java
1
public class PhoneNumberValidator implements ConstraintValidator<PhoneNumber, String> {
@Override
public boolean isValid(String phoneField, ConstraintValidatorContext context) {
if (phoneField == null) {
// can be null
return true;
// 大陆手机号码11位数,匹配格式:前三位固定格式+后8位任意数
// ^ 匹配输入字符串开始的位置
// \d 匹配一个或多个数字,其中 \ 要转义,所以是 \\d
// $ 匹配输入字符串结尾的位置
String regExp = "^[1]((3[0-9])|(4[5-9])|(5[0-3,5-9])|([6][5,6])|(7[0-9])|(8[0-9])|(9[1,8,9]))\\d{8}$";
return phoneField.matches(regExp);
搞定,我们现在就可以使用这个注解了。
1
@PhoneNumber(message = "phoneNumber 格式不正确")
@NotNull(message = "phoneNumber 不能为空")
private String phoneNumber;
通过测试验证
1
PersonRequest personRequest = PersonRequest.builder()
.phoneNumber("1816313815").build();
mockMvc.perform(post("/api/person")
.contentType(MediaType.APPLICATION_JSON)
.content(objectMapper.writeValueAsString(personRequest)))
.andExpect(MockMvcResultMatchers.jsonPath("phoneNumber").value("phoneNumber 格式不正确"));
使用验证组
验证组我们基本是不会用到的,也不太建议在项目中使用,理解起来比较麻烦,写起来也比较麻烦。简单了解即可!
当我们对对象操作的不同方法有不同的验证规则的时候才会用到验证组。
我写一个简单的例子,你们就能看明白了!
1.先创建两个接口,代表不同的验证组
1
public interface AddPersonGroup {
public interface DeletePersonGroup {
2.使用验证组
1
@Data
public class Person {
// 当验证组为 DeletePersonGroup 的时候 group 字段不能为空
@NotNull(groups = DeletePersonGroup.class)
// 当验证组为 AddPersonGroup 的时候 group 字段需要为空
@Null(groups = AddPersonGroup.class)
private String group;
@Service
@Validated
public class PersonService {
@Validated(AddPersonGroup.class)
public void validatePersonGroupForAdd(@Valid Person person) {
// do something
@Validated(DeletePersonGroup.class)
public void validatePersonGroupForDelete(@Valid Person person) {
// do something
通过测试验证:
1
@Test(expected = ConstraintViolationException.class)
public void should_check_person_with_groups() {
Person person = new Person();
person.setGroup("group1");
service.validatePersonGroupForAdd(person);