第18章 SpringBoot集成Validation校验

在日常的项目开发中,我们需要对用户的请求参数做合法性校验。通常采用的是在代码中手动校验数据。但是手动校验数据会带来代码冗余、错误处理不统一的缺点。Spring Boot 提供了强大的数据验证机制,使开发者能够轻松地执行验证操作。

接下来,我们创建一个 “SpringBootValidationDemo” 工程

然后我们修改编码格式以及Maven仓库地址,我们省略这个过程了。

接下来,我们修改 “pom.xml” 文件,添加Validation依赖,如下所示

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>com.demo</groupId>
    <artifactId>SpringBootValidationDemo</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
    </properties>

    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.6.13</version>
        <relativePath/>
    </parent>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-thymeleaf</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-validation</artifactId>
        </dependency>
        <dependency>
            <groupId>org.hibernate.validator</groupId>
            <artifactId>hibernate-validator</artifactId>
            <version>6.0.18.Final</version>
            <scope>compile</scope>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.16</version>
        </dependency>        
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
                <version>2.7.18</version>
            </plugin>
        </plugins>
    </build>

</project>

请注意,我们增加了 “spring-boot-starter-validation” 和 “hibernate-validator” 两依赖项。

spring-boot-starter-validation 参数校验功能就是 hibernate-validator 包所提供的。Hibernate Validator是一个基于Bean Validation API的参数验证框架,它提供了丰富的约束注解和自定义验证机制。Hibernate Validator的使用非常简单,只需在用户请求实体类的字段上使用标准的 Bean Validation 注解就能定义数据的验证规则。

接下来,我们继续完善我们的工程,创建 Appliaction 入口类文件

package com.demo;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class Application {

    public static void main(String[] args) {

        SpringApplication.run(Application.class, args);
    }
}

对于web服务来说,我们一般在Controller层一定要做参数校验的,用户请求参数分为如下两种形式:

POST或PUT请求,使用requestBody传递参数

GET请求,使用requestParam/PathVariable传递参数

接下来,我们先创建一个用户请求数据类 StudentRequest

package com.demo.data;

import lombok.Data;

import javax.validation.constraints.Max;
import javax.validation.constraints.Min;
import javax.validation.constraints.NotBlank;

@Data
public class StudentRequest {

    @NotBlank(message = "姓名不能为空")
    private String name;

    @Min(value = 18, message = "年龄最小18岁")
    @Max(value = 28, message = "年龄最大28岁")
    private Integer age;

}

在上面的数据类中,我们给“name”和“age”使用了注解“@NotBlank”,“@Min”和“@Max”。从注解名称来看,我们就能知道,@NotBlank是不能为空的验证规则,而@Min和@Max则是最小值和最大值的验证规则。如果用户请求的数据不符合验证规则的话,就会返回“message”中定义的信息了。

接下来,我们创建 StudentController 控制器

package com.demo.controller;

import com.demo.data.StudentRequest;
import org.springframework.stereotype.Controller;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.GetMapping;

import java.util.Map;

@Controller
public class StudentController {

    @GetMapping("/add")
    public String add(@Validated StudentRequest request, Map map){

        map.put("info", request);
        return "student";
    }

}

注意,这里我们使用 “@Validated” 注解来开启对 “StudentRequest” 的验证。我们还可以使用“@Valid”注解。这个@Valid是Hibernate validation使用的,而@Validated是Spring Validator使用的,可以理解 @Validation 是对@Valid 进行了二次封装。一般情况下,我们使用Spring的 “@Validated” 注解即可。

接下来,我们创建 “resources\static\index.html” 入口文件

<!doctype html>
<html lang="zh-CN">
<head>
    <meta charset="utf-8" />
    <meta http-equiv="X-UA-Compatible" content="IE=edge" />
    <meta name="viewport" content="width=device-width, initial-scale=1" />
    <title>index</title>
</head>
<body>

<a href="/add?name=张三&age=20">/add?name=张三&age=20</a>
<br />
<a href="/add?age=20">/add?age=20</a>
<br />
<a href="/add?name=&age=20">/add?name=&age=20</a>
<br />
<a href="/add?name=张三&age=50">/add?name=张三&age=50</a>
<br />

</body>
</html>

最后,我们需要创建 “resources\templates\student.html” 文件

<!doctype html>
<html lang="zh-CN" xmlns:th="http://www.thymeleaf.org">
<head>
    <meta charset="utf-8" />
    <meta http-equiv="X-UA-Compatible" content="IE=edge" />
    <meta name="viewport" content="width=device-width, initial-scale=1" />
    <title>student</title>
</head>
<body>

<div th:if ="${info}">
    <span th:text="${info.name}"></span>
    <span th:text="${info.age}"></span>
</div>

</body>
</html>

接下来,我们就来运行测试一下吧

我们先点击第一个正确的请求“/add?name=张三&age=20”

显示正确了,接下来点击第二个错误的请求“/add?age=20”(我们没有传递姓名字段)

页面出错了,我们查看控制台输出

显示抛出了 “org.springframework.validation.BindException” 异常。很明显,spring-boot-starter-validation参数校验不通过的结果就是抛出异常。那么我们就做一个全局异常处理器。

关于spring如何处理全局异常,请参考上一个章节:第17章 SpringBoot异常处理-优快云博客

package com.demo.exception;

import org.springframework.validation.BindException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.servlet.ModelAndView;

@ControllerAdvice
public class GlobalExceptionHandler {

    @ExceptionHandler(BindException.class)
    public ModelAndView handler(BindException e) {

        e.printStackTrace();

        ModelAndView mv = new ModelAndView();
        mv.setViewName("exception");
        String message = e.getMessage();
        if (e.getBindingResult() != null &&
                e.getBindingResult().getAllErrors() != null &&
                e.getBindingResult().getAllErrors().size() > 0) {
            // 获取注解中定义的 message 信息
            message = e.getBindingResult().getAllErrors().get(0).getDefaultMessage();
        }
        mv.addObject("message", message);
        return mv;
    }

}

需要注意的是,注解校验不通过时,可能抛出的异常有BindException异常、ValidationException异常(或其子类异常)、MethodArgumentNotValidException异常,我们都要对这些异常做处理。

接下来,我们需要创建 “resources\templates\exception.html” 文件

<!doctype html>
<html lang="zh-CN" xmlns:th="http://www.thymeleaf.org">
<head>
    <meta charset="utf-8" />
    <meta http-equiv="X-UA-Compatible" content="IE=edge" />
    <meta name="viewport" content="width=device-width, initial-scale=1" />
    <title>exception</title>
</head>
<body>

<div th:text="${message}"></div>

</body>
</html>

最后,我们再来测试一下吧

其实,我们的 “@NotBlank” ,“@Min” 或 “@Max” 注解也可以直接应用到方法的参数上面

package com.demo.controller;

import com.demo.data.StudentRequest;
import org.springframework.stereotype.Controller;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.GetMapping;

import javax.validation.constraints.Max;
import javax.validation.constraints.Min;
import javax.validation.constraints.NotBlank;
import java.util.Map;

@Validated
@Controller
public class StudentController {

    @GetMapping("/add")
    public String add(@Validated StudentRequest request, Map map){

        map.put("info", request);
        return "student";
    }

    @GetMapping("/test")
    public String test(
            @NotBlank(message = "姓名不能为空") String name,
            @Min(value = 18, message = "年龄最小18岁")
            @Max(value = 28, message = "年龄最大28岁") Integer age, Map map){

        StudentRequest request = new StudentRequest();
        request.setName(name);
        request.setAge(age);
        map.put("info", request);
        return "student";
    }

}

注意,我们需要使用“@Validated”注解到控制器 “StudentController” 类上。

接下来,我们修改 “resources\static\index.html” 入口文件

<a href="/add?name=张三&age=20">/add?name=张三&age=20</a>
<br />
<a href="/add?age=20">/add?age=20</a>
<br />
<a href="/add?name=&age=20">/add?name=&age=20</a>
<br />
<a href="/test?name=张三&age=50">/test?name=张三&age=50</a>
<br />

我们测试一下 “/test?name=张三&age=50” 请求接口

页面出错了,我们查看一下控制台日志

我们可以看到抛出了“ConstraintViolationException”异常。 我们需要在全局异常处理器GlobalExceptionHandler中来处理这个“ConstraintViolationException”异常。

package com.demo.exception;

import org.springframework.validation.BindException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.servlet.ModelAndView;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
import java.util.stream.Collectors;

@ControllerAdvice
public class GlobalExceptionHandler {

    @ExceptionHandler(BindException.class)
    public ModelAndView handler(BindException e) {

        e.printStackTrace();

        ModelAndView mv = new ModelAndView();
        mv.setViewName("exception");
        String message = e.getMessage();
        if (e.getBindingResult() != null &&
                e.getBindingResult().getAllErrors() != null &&
                e.getBindingResult().getAllErrors().size() > 0) {
            // 获取注解中定义的 message 信息
            message = e.getBindingResult().getAllErrors().get(0).getDefaultMessage();
        }
        mv.addObject("message", message);
        return mv;
    }

    @ExceptionHandler(ConstraintViolationException.class)
    public ModelAndView handler(ConstraintViolationException e) {

        e.printStackTrace();

        ModelAndView mv = new ModelAndView();
        mv.setViewName("exception");
        String message = e.getConstraintViolations().stream().map(ConstraintViolation::getMessage).collect(Collectors.joining());
        mv.addObject("message", message);
        return mv;
    }
    
}

接下来,我们重新运行测试下

另外,对于 POST或PUT 请求,我们的控制器会使用 @requestBody 接收json类型参数。如果校验失败,会抛出 MethodArgumentNotValidException 异常。我们也一起处理吧。

package com.demo.exception;

import org.springframework.validation.BindException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.servlet.ModelAndView;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
import java.util.stream.Collectors;

@ControllerAdvice
public class GlobalExceptionHandler {

    @ExceptionHandler(BindException.class)
    public ModelAndView handler(BindException e) {

        e.printStackTrace();

        ModelAndView mv = new ModelAndView();
        mv.setViewName("exception");
        String message = e.getMessage();
        if (e.getBindingResult() != null &&
                e.getBindingResult().getAllErrors() != null &&
                e.getBindingResult().getAllErrors().size() > 0) {
            // 获取注解中定义的 message 信息
            message = e.getBindingResult().getAllErrors().get(0).getDefaultMessage();
        }
        mv.addObject("message", message);
        return mv;
    }

    @ExceptionHandler(ConstraintViolationException.class)
    public ModelAndView handler(ConstraintViolationException e) {

        e.printStackTrace();

        ModelAndView mv = new ModelAndView();
        mv.setViewName("exception");
        String message = e.getConstraintViolations().stream().map(ConstraintViolation::getMessage).collect(Collectors.joining());
        mv.addObject("message", message);
        return mv;
    }

    @ExceptionHandler(MethodArgumentNotValidException.class)
    public ModelAndView handler(MethodArgumentNotValidException e) {

        e.printStackTrace();

        ModelAndView mv = new ModelAndView();
        mv.setViewName("exception");
        String message = e.getBindingResult().getFieldError().getDefaultMessage();
        mv.addObject("message", message);
        return mv;
    }

}

还有类似于“HttpRequestMethodNotSupportedException”,“MissingPathVariableException”, “MethodArgumentTypeMismatchException” 等异常,大家也可以处理一起。

接下来,我们介绍一下 spring-boot-starter-validation 提供的常用注解

  1. @NotNull: 校验字段值不能为 null。

@NotNull(message = "姓名不存在")
private String name;
  1. @NotBlank: 校验字符串值不能为 null 或空字符串。

@NotBlank(message = "姓名不能为空")
private String name;
  1. NotEmpty: 校验集合元素或数组元素或者字符串是否非空。

@NotEmpty(message = "列表元素不能为空")
private List<String> items;
  1. @Length: 校验字符串元素的长度。

@Length(min = 6, max = 12, message = "密码长度为6-12位")
private String password;
  1. @Size: 校验集合元素个数或字符串的长度在指定范围内。

@Size(min = 1, max = 10, message = "集合元素数量必须是1-10位")
private List<String> items;
  1. @Min: 校验数字元素的最小值。

@Min(value = 18, message = "年龄最小18岁")
private int age;
  1. @Max: 校验数字元素的最大值。

@Max(value = 60, message = "年龄最大60岁")
private int age;
  1. @Pattern: 根据正则表达式校验字符串元素的格式。

@Pattern(regexp = "[a-zA-Z0-9]+", message = "用户名必须是数字字母组合")
private String username;

如果官方提供可用的校验注解不能满足我们的需要,我们自己可以定义一个自定义校验器。

package com.demo.validation;

import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.Documented;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;

import static java.lang.annotation.ElementType.*;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER})
@Retention(RUNTIME)
@Documented
@Constraint(validatedBy = {MyCustomerValidationImpl.class})
public @interface MyCustomerValidation {
    
    String message() default "";
    Class<?>[] groups() default { };
    Class<? extends Payload>[] payload() default { };

}

我们定义了一个 “MyCustomerValidation” 注解

package com.demo.validation;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;

public class MyCustomerValidationImpl implements ConstraintValidator<MyCustomerValidation, String> {

    @Override
    public boolean isValid(String param, ConstraintValidatorContext context) {
        
        if(param == null || param.length() <= 0) return false;
        return true;
    }
}

我们定义了校验规则,它的参数是一个字符串类型,如果为空的话,则校验不通过。

当然,我们可以根据自己的修在这里完成复杂的校验规则。

接下来,我们就使用一下这个 “@MyCustomerValidation” 注解。

    @GetMapping("/mytest")
    public String mytest(@MyCustomerValidation(message = "姓名不能为空") String name, Map map){

        StudentRequest request = new StudentRequest();
        request.setName(name);
        request.setAge(20);
        map.put("info", request);
        return "student";
    }

接下来,我们修改 “resources\static\index.html” 入口文件

<a href="/add?name=张三&age=20">/add?name=张三&age=20</a>
<br />
<a href="/add?age=20">/add?age=20</a>
<br />
<a href="/add?name=&age=20">/add?name=&age=20</a>
<br />
<a href="/test?name=张三&age=50">/test?name=张三&age=50</a>
<br />
<a href="/mytest?name=">/mytest?name=</a>
<br />

我们重新运行测试一下

本工程完整代码下载:https://download.youkuaiyun.com/download/richieandndsc/89953250

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值