SpringBoot中处理校验逻辑的两种方式,真的很机智!


大家好,我是二哥呀。最近正在开发一个知识库学习网站编程猫,需要对请求参数进行校验,比如说非空啊、长度限制啊等等,可选的解决方案有两种:
一种是用 Hibernate Validator 来处理
一种是用全局异常来处理
两种方式,我们一一来实践体验一下。
一、Hibernate Validator
Spring Boot 已经内置了 Hibernate Validator 校验框架,这个可以通过 Spring Boot 官网查看和确认。
第一步,进入 Spring Boot 官网,点击 learn 这个面板,点击参考文档。

第二步,在参考文档页点击「依赖的版本」。

第三步,在依赖版本页就可以查看到所有的依赖了,包括版本号。

PS:如果发现没有起效,可能是依赖版本冲突了,手动把 Hibernate Validator 依赖添加到 pom.xml 文件就可以了。
<dependency>    <groupId>org.hibernate.validator</groupId>    <artifactId>hibernate-validator</artifactId>    <version>6.0.17.Final</version></dependency><dependency>    <groupId>javax.validation</groupId>    <artifactId>validation-api</artifactId>    <version>2.0.1.Final</version></dependency>
通过 Hibernate Validator 校验框架,我们可以直接在请求参数的字段上加入注解来完成校验。
具体该怎么做呢?
第一步,在需要验证的字段上加上 Hibernate Validator 提供的校验注解。
比如说我现在有一个用户名和密码登录的请求参数 UsersLoginParam 类:
@Data@ApiModel(value="用户登录", description="用户表")public class UsersLoginParam implements Serializable {    private static final long serialVersionUID = 1L;    @ApiModelProperty(value = "登录名")    @NotBlank(message="登录名不能为空")    private String userLogin;    @ApiModelProperty(value = "密码")    @NotBlank(message="密码不能为空")    private String userPass;}
就可以通过 @NotBlank 注解来对用户名和密码进行判空校验。除了 @NotBlank 注解,Hibernate Validator 还提供了以下常用注解:
@NotNull:被注解的字段不能为 null;
@NotEmpty:被注解的字段不能为空;
@Min:被注解的字段必须大于等于其value值;
@Max:被注解的字段必须小于等于其value值;
@Size:被注解的字段必须在其min和max值之间;
@Pattern:被注解的字段必须符合所定义的正则表达式;
@Email:被注解的字段必须符合邮箱格式。
第二步,在对应的请求接口(UsersController.login())中添加 @Validated 注解,并注入一个 BindingResult 参数。
@Controller@Api(tags="用户")@RequestMapping("/users")public class UsersController {    @Autowired    private IUsersService usersService;    @ApiOperation(value = "登录以后返回token")    @RequestMapping(value = "/login", method = RequestMethod.POST)    @ResponseBody    public ResultObject login(@Validated UsersLoginParam users, BindingResult result) {        String token = usersService.login(users.getUserLogin(), users.getUserPass());        if (token == null) {            return ResultObject.validateFailed("用户名或密码错误");        }        Map<String, String> tokenMap = new HashMap<>();        tokenMap.put("token", token);        tokenMap.put("tokenHead", tokenHead);        return ResultObject.success(tokenMap);    }}
第三步,为控制层(UsersController)创建一个切面,将通知注入到 BindingResult 对象中,然后再判断是否有校验错误,有错误的话返回校验提示信息,否则放行。
@Aspect@Component@Order(2)public class BindingResultAspect {    @Pointcut("execution(public * com.codingmore.controller.*.*(..))")    public void BindingResult() {    }    @Around("BindingResult()")    public Object doAround(ProceedingJoinPoint joinPoint) throws Throwable {        Object[] args = joinPoint.getArgs();        for (Object arg : args) {            if (arg instanceof BindingResult) {                BindingResult result = (BindingResult) arg;                if (result.hasErrors()) {                    FieldError fieldError = result.getFieldError();                    if(fieldError!=null){                        return ResultObject.validateFailed(fieldError.getDefaultMessage());                    }else{                        return ResultObject.validateFailed();                    }                }            }        }        return joinPoint.proceed();    }}
这里涉及到了 SpringBoot AOP 的知识,我在前面的文章里讲解过了,戳这个链接可以直达:SpringBoot AOP 扫盲
第四步,访问登录接口,用户名和密码都不传入的情况下,就会返回“用户名不能为空”的提示信息。

通过 debug 的形式,体验一下整个工作流程。

可以看得出,Hibernate Validator 带来的优势有这些:
验证逻辑与业务逻辑进行了分离,降低了程序耦合度;
统一且规范的验证方式,无需再次编写重复的验证代码。
不过,也带来一些弊端,比如说:
需要在请求接口的方法中注入 BindingResult 对象
只能校验一些非常简单的逻辑,涉及到数据查询就无能为力了。
二、全局异常处理
使用全局异常处理的优点就是比较灵活,可以处理比较复杂的逻辑校验,在校验失败的时候直接抛出异常,然后进行捕获处理就可以了。
第一步,新建一个自定义异常类 ApiException。
public class ApiException extends RuntimeException {    private IErrorCode errorCode;    public ApiException(IErrorCode errorCode) {        super(errorCode.getMessage());        this.errorCode = errorCode;    }    public ApiException(String message) {        super(message);    }    public ApiException(Throwable cause) {        super(cause);    }    public ApiException(String message, Throwable cause) {        super(message, cause);    }    public IErrorCode getErrorCode() {        return errorCode;    }}
第二步,新建一个断言处理类 Asserts,简化抛出 ApiException 的步骤。
public class Asserts {    public static void fail(String message) {        throw new ApiException(message);    }    public static void fail(IErrorCode errorCode) {        throw new ApiException(errorCode);    }}
第三步,新建一全局异常处理类 GlobalExceptionHandler,对异常信息进行解析,并封装到统一的返回对象 ResultObject 中。
@ControllerAdvicepublic class GlobalExceptionHandler {    @ResponseBody    @ExceptionHandler(value = ApiException.class)    public ResultObject handle(ApiException e) {        if (e.getErrorCode() != null) {            return ResultObject.failed(e.getErrorCode());        }        return ResultObject.failed(e.getMessage());    }}
全局异常处理类用到了两个注解,@ControllerAdvice 和 @ExceptionHandler。
@ControllerAdvice 是一个特殊的 @Component(可以通过源码看得到),用于标识一个类,这个类中被以下三种注解标识的方法:@ExceptionHandler,@InitBinder,@ModelAttribute,将作用于所有@Controller 类的接口上。
@Target({ElementType.TYPE})@Retention(RetentionPolicy.RUNTIME)@Documented@Componentpublic @interface ControllerAdvice {}
@ExceptionHandler 注解的作用就是标识统一异常处理,它可以指定要统一处理的异常类型,比如说我们自定义的 ApiException。
第四步,在需要校验的地方通过 Asserts 类抛出异常 ApiException。还拿用户登录这个接口来说明吧。
@Controller@Api(tags="用户")@RequestMapping("/users")public class UsersController {    @ApiOperation(value = "登录以后返回token")    @RequestMapping(value = "/login", method = RequestMethod.POST)    @ResponseBody    public ResultObject login(@Validated UsersLoginParam users, BindingResult result) {        String token = usersService.login(users.getUserLogin(), users.getUserPass());             Map<String, String> tokenMap = new HashMap<>();        tokenMap.put("token", token);        tokenMap.put("tokenHead", tokenHead);        return ResultObject.success(tokenMap);    }}
该接口需要查询数据库验证密码是否正确,如果密码不正确就抛出校验信息“密码不正确”。
@Servicepublic class UsersServiceImpl extends ServiceImpl<UsersMapper, Users> implements IUsersService {    public String login(String username, String password) {        String token = null;        //密码需要客户端加密后传递        UserDetails userDetails = loadUserByUsername(username);        if (!passwordEncoder.matches(password, userDetails.getPassword())) {            Asserts.fail("密码不正确");         }        // 其他代码省略        return token;    }}
第五步,通过 ApiPost 来测试一下接口,故意把密码输错。

也可以通过 debug 的形式,体验一下整个工作流程。

三、总结
实际开发中把两者结合在一起用,就可以弥补彼此的短板了,简单校验用 Hibernate Validator,复杂一点的逻辑校验,比如说需要数据库查询用全局异常处理来实现。
源码地址:https://github.com/itwanger/coding-more
参考链接:http://www.macrozheng.com

没有什么使我停留——除了目的,纵然岸旁有玫瑰、有绿荫、有宁静的港湾,我是不系之舟。
推荐阅读:
差点散伙!
撸了几行骚代码,解放了双手!
二哥的读者(女,从小就想当黑客,初中学编程,高中造火箭
二哥的读者(男,半年从机械转码,去携程实习了!
到顶部