ConstraintValidator类如何实现自定义注解校验前端传参


Posted in Java/Android onJune 18, 2021

前言

今天项目碰到这么一个问题,前端传递的json格式到我的微服务后端转换为vo类,其中有一个Integer的字段后端希望它在固定的几个数里面取值,例如只能取值1、2、4。

一般咱们的思路是啥呢,找一些spring为我们提供的类似@Length、@NotBlank这些注解加在参数上面。

像下面这样

ConstraintValidator类如何实现自定义注解校验前端传参

不过我这个校验一时间想不起来用哪个注解了,咋整呢?行吧,咱不求人,自己实现一个。

补充一句话,千万别直接拿着实体类往后传递到service层校验哈,太low了这样子。

一、利用@Constraint定义注解

import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import javax.validation.Constraint;
import javax.validation.Payload;
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(
        validatedBy = {IllegalNumberValidator.class}
)
public @interface IllegalNumber {
    /**
     * 允许前端取的几个值
     * 
     */
    int[] acceptValues();
    /**
     * 标识此字段是否为必选项
     * 
     */
    boolean required() default true;
 
    /**
     * 标识此字段是否为必选项
     * 
     */
    String message() default "数字不合法,不在要求的取值范围之内";
 
    /**
     * 标识要校验的值所属组,这个后面详细解释
     * 
     */
    Class<?>[] groups() default {};
    
    /**
     * 这个字段一般不需要我们关注
     * 
     */
    Class<? extends Payload>[] payload() default {};
}

二、增强注解

1.编写增强类

注意到刚才注解中的@Constraint注解了吗

ConstraintValidator类如何实现自定义注解校验前端传参

validatedBy属性标识这个注解要被哪个类所增强,我们把增强类IllegalNumberValidator定义出来

import com.google.common.collect.Lists;
import org.springframework.util.StringUtils;
import java.util.List;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
public class IllegalNumberValidator implements ConstraintValidator<IllegalNumber, Integer> {
    private final List<Integer> valueList = Lists.newArrayList();
    private boolean require = false;
    @Override
    public void initialize(IllegalNumber constraintAnnotation) {
        require = constraintAnnotation.required();
        int[] ints = constraintAnnotation.acceptValues();
        for (int anInt : ints) {
            valueList.add(anInt);
        }
    }
    @Override
    public boolean isValid(Integer number, ConstraintValidatorContext constraintValidatorContext) {
        // 如果是必选的话,假设为我们传递的参数为空那肯定不行
        if (require) {
            if (number == null) {
                return false;
            }
            return valueList.contains(number);
        } else {
            // 如果不为必选参数,为空返回true,不为空还是得校验
            if (StringUtils.isEmpty(number)) {
                return true;
            } else {
                return valueList.contains(number);
            }
        }
    }
}

增强类继承ConstraintValidator类,实现的initialize()方法是初始化方法,啥意思呢,啥目的呢?在你真正执行校验之前,可以做一些准备性工作,发生在要校验的值上面的注解的IllegalNumber 已经给咱们传进来了。我做的初始化工作就是load一下Integer类型的可选值,方便一会执行真正的校验。

然后在isValid()方法中你可以做真正的校验了,很简单,我看下传递的Integer类型的值是不是acceptValues里面的可选值就行了。

定义一个前端传递的类,方便调试注解

import lombok.Data;
import org.hibernate.validator.constraints.Length;
import javax.validation.constraints.NotNull;
@Data
public class TestVO {
    @NotNull
    @IllegalNumber(acceptValues = {0, 1,2},required = true,message = "请正确取值")
    private Integer number;
    @NotNull
    @Length(min = 1)
    private String password;
}

定义接口,用来接收前端传递的json数据并parse为TestVO类

/**
     * 测试自定义注解
     *
     * @param vo json将会映射的实体
     * @return 默认信息
     */
    @PostMapping(value = "/v1.0/test2", name = "测试自定义注解")
    public String test2(@Valid @RequestBody TestVO vo) {
        log.info("get vo success , detail message is:{}", vo);
        return RETURN_MESSAGE;
    }

注意,如果说前端传递数据不符合注解的校验,其实是会抛出异常的来自@Constraint注解实现的注解都有此特点,例如@Length、@Max等。咱们需要在异常抛出的时候给出拦截 这里咱们做一个通用拦截:

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.ResponseStatus;
import java.util.Objects;
import javax.validation.ConstraintViolationException;
@ControllerAdvice
public class RestResponseEntityExceptionHandler {
    private static final Logger LOG = LoggerFactory.getLogger(RestResponseEntityExceptionHandler.class);
    @Autowired
    private ApplicationContext applicationContext;
    @ExceptionHandler({ConstraintViolationException.class})
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    @ResponseBody
    public String handleConstraintViolationException(ConstraintViolationException e) {
        LOG.info("ConstraintViolationException intercept success:{}", e.getMessage());
        return e.getMessage();
    }
    @ExceptionHandler({MethodArgumentNotValidException.class})
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    @ResponseBody
    public String handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {
        LOG.info("MethodArgumentNotValidException intercept success:{}", e.getMessage());
        return Objects.requireNonNull(e.getBindingResult().getFieldError()).getDefaultMessage();
    }
}

2.测试效果

下面测试一下。打开postman。直接干!取值的限定是0、1、2。咱们先试下错误的

ConstraintValidator类如何实现自定义注解校验前端传参

ok,再试下正确的

ConstraintValidator类如何实现自定义注解校验前端传参

3.注解中的groups参数详解

groups参数,代表所属组的意思。演示下怎么用,大家也就知道这个参数啥意思了。 建立Group1接口

public interface Group1 {
}

建立Group2接口

public interface Group2 {
}

给TestVO增加一个参数,方便一会进行比较

@Data
public class TestVO {
    @NotNull
    @IllegalNumber(acceptValues = {0, 1,2},required = true,message = "请正确取值",groups = Group1.class)
    private Integer number;
    @NotNull
    @IllegalNumber(acceptValues = {0, 1,2},required = true,message = "请正确取值ha",groups = Group2.class)
    private Integer number2;
    @NotNull
    @Length(min = 1)
    private String password; 
}

使用注解的时候标明所属组:

ConstraintValidator类如何实现自定义注解校验前端传参

接口处也进行标识:

ConstraintValidator类如何实现自定义注解校验前端传参

现在咱们分别测试下两个接口,看groups参数是否能生效

test2接口

ConstraintValidator类如何实现自定义注解校验前端传参

test3接口

ConstraintValidator类如何实现自定义注解校验前端传参

ok,相信大家对此参数已经掌握了,这里不再多余赘述。

总结

本篇介绍了自定义注解的另外一种手法,其实还有许许多多的手法,例如利用反射实现、利用拦截器实现等等。遇见的时候咱们再介绍。 以上仅为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木

Java/Android 相关文章推荐
Java实现多文件上传功能
Jun 30 Java/Android
swagger如何返回map字段注释
Jul 03 Java/Android
SpringBoot整合RabbitMQ的5种模式实战
Aug 02 Java/Android
java多态注意项小结
Oct 16 Java/Android
alibaba seata服务端具体实现
Feb 24 Java/Android
Spring Boot 使用 Spring-Retry 进行重试框架
Apr 24 Java/Android
Java设计模式中的命令模式
Apr 28 Java/Android
Android 中的类文件和类加载器详情
Jun 05 Java/Android
springboot实现string转json json里面带数组
Jun 16 Java/Android
MyBatis XPathParser解析器使用范例详解
Jul 15 Java/Android
app场景下uniapp的扫码记录
Jul 23 Java/Android
OpenFeign实现远程调用
Aug 14 Java/Android
SpringBoot项目中控制台日志的保存配置操作
Jun 18 #Java/Android
浅谈@Value和@Bean的执行顺序问题
Jun 16 #Java/Android
SpringBoot2 参数管理实践之入参出参与校验的方式
Jun 16 #Java/Android
SpringBoot生成License的实现示例
Springboot如何使用logback实现多环境配置?
解决tk mapper 通用mapper的bug问题
一篇带你入门Java垃圾回收器
You might like
php与XML、XSLT、Mysql的结合运用实现代码
2009/11/19 PHP
解析php多线程下载远程多个文件
2013/06/25 PHP
PHP高手需要要掌握的知识点
2014/08/21 PHP
PHP结合jQuery插件ajaxFileUpload实现异步上传文件实例
2020/08/17 PHP
PHP 表单提交及处理表单数据详解及实例
2016/12/27 PHP
Laravel 模型关联基础教程详解
2019/09/17 PHP
JavaScript 放大镜 移动镜片效果代码
2011/05/09 Javascript
模拟jQuery中的ready方法及实现按需加载css,js实例代码
2013/09/27 Javascript
JS获取键盘上任意按键的值(实例代码)
2013/11/12 Javascript
javascript批量修改文件编码格式的方法
2015/01/27 Javascript
JavaScript检测鼠标移动方向的方法
2015/05/22 Javascript
jquery拼接ajax 的json和字符串拼接的方法
2017/03/11 Javascript
javascript中的replace函数(带注释demo)
2018/01/07 Javascript
jQuery实现图片上传预览效果功能完整实例【测试可用】
2018/05/28 jQuery
vant(ZanUi)结合async-validator实现表单验证的方法
2018/12/06 Javascript
Nautil 中使用双向数据绑定的实现
2019/10/02 Javascript
JS实现瀑布流效果
2020/03/07 Javascript
谈谈我在vue-cli3中用预渲染遇到的坑
2020/04/22 Javascript
python编写暴力破解zip文档程序的实例讲解
2018/04/24 Python
解决python 未发现数据源名称并且未指定默认驱动程序的问题
2018/12/07 Python
python 接口实现 供第三方调用的例子
2019/08/13 Python
Python object类中的特殊方法代码讲解
2020/03/06 Python
python获取响应某个字段值的3种实现方法
2020/04/30 Python
tensorflow 20:搭网络,导出模型,运行模型的实例
2020/05/26 Python
使用Nibabel库对nii格式图像的读写操作
2020/07/01 Python
如何把python项目部署到linux服务器
2020/08/26 Python
python3中calendar返回某一时间点实例讲解
2020/11/18 Python
python3访问字典里的值实例方法
2020/11/18 Python
高中生学习总结的自我评价范文
2013/10/13 职场文书
酒店工作职员求职简历的自我评价
2013/10/23 职场文书
岗位职责范本
2013/11/23 职场文书
学校办公室主任职责
2013/12/27 职场文书
运动会开幕词
2015/01/28 职场文书
本溪关门山导游词
2015/02/09 职场文书
房地产置业顾问岗位职责
2015/04/11 职场文书
分享Python获取本机IP地址的几种方法
2022/03/17 Python