springboot介面引數校驗JSR303的實現

<dependency>
  <groupId>org.springframework.boot</groupId>
  <artifactId>spring-boot-starter-web</artifactId>
</dependency>

package com.zhmsky.mallproduct.entity;

import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;

import java.io.Serializable;
import java.util.Date;
import lombok.Data;

import javax.validation.constraints.NotBlank;

/**
* 品牌
* 
* @author zhmsky
* @email zhmsky@gmail.com
* @date 2022-07-30 16:40:28
*/
@Data
@TableName("pms_brand")
public class BrandEntity implements Serializable {
    private static final long serialVersionUID = 1L;

    /**
    * 品牌名
    */
    @NotBlank(message = "品牌名不能為空")
    private String name;
    
}

@RequestMapping(value = "/save",method = RequestMethod.POST)
public R save(@Valid @RequestBody BrandEntity brand, BindingResult result){
    if(result.hasErrors()){
        Map<String, String> map = new HashMap<>();
        //獲取校驗結果
        for (FieldError fieldError : result.getFieldErrors()) {
            //獲取錯誤提示
            String message = fieldError.getDefaultMessage();
            //校驗錯誤的欄位
            String field = fieldError.getField();
            //封裝錯誤
            map.put(field,message);
        }
        return R.error(20001,"引數校驗失敗").put("errorMap",map);
    }
	brandService.save(brand);
    return R.ok();
}

package com.zhmsky.mallproduct.exception;

import com.zhmsky.common.utils.R;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.BindingResult;
import org.springframework.validation.FieldError;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.util.HashMap;
import java.util.Map;

/**
 * 同意例外處理類
 * @author zhmsky
 * @date 2022/8/1 21:36
 */
@Slf4j
@RestControllerAdvice(basePackages = {"com.zhmsky.mallproduct.controller"})
public class MallProductControllerAdvice {

    //具體的某個異常型別處理
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public R handleValidException(MethodArgumentNotValidException e){
        log.error("資料異常:{},異常型別:{}",e.getMessage(),e.getClass());
        BindingResult result = e.getBindingResult();
        Map<String, String> map = new HashMap<>();
        for (FieldError fieldError : result.getFieldErrors()) {
            map.put(fieldError.getDefaultMessage(),fieldError.getField());
        }
        return R.error().put("errorMap",map);
    }

    //丟擲的所有異常型別處理
    @ExceptionHandler(Throwable.class)
    public R handleException(Throwable e){
        //TODO 處理邏輯
        return R.error();
    }
}

package com.zhmsky.common.exception;

/**
 * @author zhmsky
 * @date 2022/8/1 21:56
 */
public enum ErrorCodeEnum {
  
    UNKNOWN_EXCEPTION(10000, "系統未知錯誤"),

    VALID_EXCEPTION(10001, "引數格式校驗失敗");

    /**
     * 錯誤碼
     */
    private Integer code;

    /**
     * 錯誤提示
     */
    private String msg;

    ErrorCodeEnum(Integer code, String msg) {
        this.code = code;
        this.msg = msg;
    }

    public Integer getCode(){
        return this.code;
    }

    public String getMsg(){
        return this.msg;
    }
}

@Documented
@Constraint(validatedBy = { })
@Target({ METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE })
@Retention(RUNTIME)
@Repeatable(List.class)
public @interface NotBlank {
    //校驗規則提示資訊
	String message() default "{javax.validation.constraints.NotBlank.message}";
    
    //分組校驗
	Class<?>[] groups() default { };

    //負載
	Class<? extends Payload>[] payload() default { };

	/**
	 * Defines several {@code @NotBlank} constraints on the same element.
	 *
	 * @see NotBlank
	 */
	@Target({ METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE })
	@Retention(RUNTIME)
	@Documented
	public @interface List {
		NotBlank[] value();
	}
}

@Documented
@Constraint(validatedBy = {StatusValueConstraintValidator.class})
@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE})
@Retention(RUNTIME)
public @interface statusValue {
    //校驗規則提示資訊（在組態檔中設定）
    String message() default "{com.zhmsky.common.validator.statusValue.message}";

    Class<?>[] groups() default {};

    Class<? extends Payload>[] payload() default {};

    //註解引數
    int[] values() default {};
}

com.zhmsky.common.validator.statusValue.message=error.......xxxx

//statusValue就是自定義的註解，Integer就是自定義註解標註的欄位型別
public class StatusValueConstraintValidator implements ConstraintValidator<statusValue, Integer> {
    
    private Set<Integer> set = new HashSet<>();
    
    //初始化方法
    @Override
    public void initialize(statusValue constraintAnnotation) {
        //欄位標註的註解裡面設定的值（校驗規則）
        int[] values = constraintAnnotation.values();
        for (int value : values) {
            set.add(value);
        }
    }
    
    //判斷是否校驗成功
    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
        //value就是欄位欄位實際接收到的值
        return set.contains(value);
    }
}

@Data
@TableName("pms_brand")
public class BrandEntity implements Serializable {
	private static final long serialVersionUID = 1L;
    
    	/**
	 * 品牌名
	 */
	@NotBlank(message = "品牌名不能為空")
	private String name;
    
    
	/**
	 * 顯示狀態[0-不顯示；1-顯示]，指定欄位值只能為 0 或 1
	 */
	@statusValue(values={0,1})
	private Integer showStatus;

}