ConstraintValidator類如何實現(xiàn)自定義注解校驗前端傳參

更新時間：2021年06月18日 10:49:42 作者：追夢狗蛋

這篇文章主要介紹了ConstraintValidator類實現(xiàn)自定義注解校驗前端傳參的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

前言

今天項目碰到這么一個問題，前端傳遞的json格式到我的微服務后端轉(zhuǎn)換為vo類，其中有一個Integer的字段后端希望它在固定的幾個數(shù)里面取值，例如只能取值1、2、4。

一般咱們的思路是啥呢，找一些spring為我們提供的類似@Length、@NotBlank這些注解加在參數(shù)上面。

像下面這樣

在這里插入圖片描述

不過我這個校驗一時間想不起來用哪個注解了，咋整呢？行吧，咱不求人，自己實現(xiàn)一個。

補充一句話，千萬別直接拿著實體類往后傳遞到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 "數(shù)字不合法,不在要求的取值范圍之內(nèi)";
 
    /**
     * 標識要校驗的值所屬組，這個后面詳細解釋
     * 
     */
    Class<?>[] groups() default {};
    
    /**
     * 這個字段一般不需要我們關注
     * 
     */
    Class<? extends Payload>[] payload() default {};
}

二、增強注解

1.編寫增強類

注意到剛才注解中的@Constraint注解了嗎

在這里插入圖片描述

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) {
        // 如果是必選的話,假設為我們傳遞的參數(shù)為空那肯定不行
        if (require) {
            if (number == null) {
                return false;
            }
            return valueList.contains(number);
        } else {
            // 如果不為必選參數(shù)，為空返回true，不為空還是得校驗
            if (StringUtils.isEmpty(number)) {
                return true;
            } else {
                return valueList.contains(number);
            }
        }
    }
}

增強類繼承ConstraintValidator類，實現(xiàn)的initialize()方法是初始化方法，啥意思呢，啥目的呢？在你真正執(zhí)行校驗之前，可以做一些準備性工作，發(fā)生在要校驗的值上面的注解的IllegalNumber 已經(jīng)給咱們傳進來了。我做的初始化工作就是load一下Integer類型的可選值，方便一會執(zhí)行真正的校驗。

然后在isValid()方法中你可以做真正的校驗了，很簡單，我看下傳遞的Integer類型的值是不是acceptValues里面的可選值就行了。

定義一個前端傳遞的類，方便調(diào)試注解

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數(shù)據(jù)并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;
    }

注意，如果說前端傳遞數(shù)據(jù)不符合注解的校驗，其實是會拋出異常的來自@Constraint注解實現(xiàn)的注解都有此特點，例如@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。咱們先試下錯誤的

在這里插入圖片描述

ok,再試下正確的

在這里插入圖片描述

3.注解中的groups參數(shù)詳解

groups參數(shù)，代表所屬組的意思。演示下怎么用，大家也就知道這個參數(shù)啥意思了。建立Group1接口

public interface Group1 {
}

建立Group2接口

public interface Group2 {
}

給TestVO增加一個參數(shù)，方便一會進行比較

@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; 
}

使用注解的時候標明所屬組：

在這里插入圖片描述

接口處也進行標識：

在這里插入圖片描述

現(xiàn)在咱們分別測試下兩個接口，看groups參數(shù)是否能生效

test2接口

在這里插入圖片描述

test3接口

在這里插入圖片描述

ok，相信大家對此參數(shù)已經(jīng)掌握了，這里不再多余贅述。

總結(jié)

本篇介紹了自定義注解的另外一種手法，其實還有許許多多的手法，例如利用反射實現(xiàn)、利用攔截器實現(xiàn)等等。遇見的時候咱們再介紹。以上僅為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家

您可能感興趣的文章:

Springboot事件和bean生命周期執(zhí)行機制實例詳解
這篇文章主要介紹了Springboot事件和bean的生命周期執(zhí)行機制,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2023-03-03
SpringBoot之跨域過濾器配置允許跨域訪問方式
這篇文章主要介紹了SpringBoot之跨域過濾器配置允許跨域訪問方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2022-07-07
關于ResponseEntity類和HttpEntity及跨平臺路徑問題
這篇文章主要介紹了關于ResponseEntity類和HttpEntity及跨平臺路徑問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2024-07-07
java ssm框架實現(xiàn)分頁功能的示例代碼(oracle)
這篇文章主要介紹了java ssm框架實現(xiàn)分頁功能的示例代碼(oracle)，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-03-03
使用Lombok子類繼承父類,父類屬性不生效問題及解決
在使用Lombok庫時,若子類繼承父類,父類的屬性可能不會自動生效,為解決此問題,可通過在父類上添加@Getter和@Setter注解,或使用@SuperBuilder注解來確保父類屬性在子類中有效,同時,需注意確保Lombok版本一致且正確配置了相關插件
2024-10-10
Java-Io-RandomAccessFile任意位置讀寫數(shù)據(jù)的操作小結(jié)
RandomAccessFile類支持隨機訪問方式，可以跳轉(zhuǎn)到文件的任意位置讀寫數(shù)據(jù),這個類在文件隨機讀取時有很大的優(yōu)勢,可利用多線程完成對一個大文件的讀寫，本文給大家介紹Java-Io-RandomAccessFile(任意位置讀寫數(shù)據(jù))的相關知識,需要的朋友可以參考下
2022-05-05
5個主流的Java開源IDE工具詳解
這篇文章主要介紹了5個主流的Java開源IDE工具,無論如何，Java在當今使用的編程語言中始終排在前三名，在TIOBE索引中涉及700萬到1000萬的程序員和開發(fā)者
2020-07-07
SpringBoot配置Redis連接池的實現(xiàn)步驟
本文主要介紹了SpringBoot配置Redis連接池的實現(xiàn)步驟,詳細的講解了連接池的作用、配置方式、連接池參數(shù)說明,具有一定的參考價值,感興趣的可以了解一下
2025-03-03
基于TCP通信丟包原因總結(jié)(推薦)
下面小編就為大家?guī)硪黄赥CP通信丟包原因總結(jié)(推薦)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2017-08-08
Java神操作：圖片快速轉(zhuǎn)換PDF秘籍
想掌握Java神操作,將圖片快速轉(zhuǎn)換成PDF嗎？這份秘籍將一步步引導你,輕松實現(xiàn)這一目標,別等了,跟著我們一起,解鎖這項超實用技能吧！
2024-02-02