springboot接口参数校验JSR303的实现

目录

• 一、在controller接口处理校验异常
• 二、统一异常处理
• 三、错误码枚举类
• 四、自定义参数校验注解

在 javax.validation.constraints包中定义了非常多的校验注解，引入依赖：

  org.springframework.boot
  spring-boot-starter-web

在实体类上添加对应注解即可，

实体类：

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 [email protected]
* @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;
    
}

一、在controller接口处理校验异常

message可指定校验说明，
接下来在对应的controller接口方法中使用@Valid 注解来指定接口参数校验，后面紧跟BindingResult来接收校验结果，当发生校验失败则封装并返回校验失败信息，如果通过校验则执行相关逻辑

@RequestMapping(value = "/save",method = RequestMethod.POST)
public R save(@Valid @RequestBody BrandEntity brand, BindingResult result){
    if(result.hasErrors()){
        Map 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();
}

二、统一异常处理

如果在每一个接口都进行参数校验异常处理，工作量比较大，并且代码比较繁琐，于是引入全局异常处理，
也就是@RestControllerAdvice+@ExceptionHandler注解的使用。

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

四、自定义参数校验注解

1、编写自定义校验注解

参照javax.validation.constraints的@NotBlank注解，

@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[] 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[] payload() default {};

    //注解参数
    int[] values() default {};
}

创建 ValidationMessages.properties 文件，在配置文件中配置自定义的校验规则提示信息：

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

2、编写自定义校验器

//statusValue就是自定义的注解，Integer就是自定义注解标注的字段类型
public class StatusValueConstraintValidator implements ConstraintValidator {
    
    private Set 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);
    }
}

3、关联自定义注解和自定义校验器

也就是在自定义注解的@Constraint注解中指明自定义的校验器，

4、测试自定义注解@statusValue

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

}

在对应的controller接口中使用@Validated注解来进行校验，showStatus字段传入值3，

测试结果如图：

到此这篇关于springboot接口参数校验JSR303的实现的文章就介绍到这了,更多相关springboot接口参数校验JSR303 内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！