springboot手动触发参数校验,service层调用参数校验

背景

入参校验一般是在控制层通过javax.validation.constraints包下的规则注解如NotNull结合Valid与Validated实现,但是有时候我们的方法不提供给controller调用,这时候就无法触发自动参数校验,为此我们可以在不更改校验代码的前提下手动触发spring的参数校验。

代码实现

  • 触发工具类
java 复制代码
import javax.validation.ConstraintViolation;
import javax.validation.Validation;
import javax.validation.ValidationException;
import javax.validation.Validator;
import java.util.List;
import java.util.Set;

/**
 * @description validator参数校验类
 */
public class ValidateUtil {

    private static final Validator VALIDATOR = Validation.buildDefaultValidatorFactory().getValidator();

    /**
     * 校验实体类
     * @param t  参数
     * @param notNull t是否可以为空
     */
    public static <T> void validate(T t, boolean notNull) {
        if (t == null) {
            if (notNull) {
                throw new ValidationException("参数不能为空");
            }
            return;
        }
        Set<ConstraintViolation<T>> constraintViolations = VALIDATOR.validate(t);
        if (constraintViolations.size() < 1) {
            return;
        }
        for (ConstraintViolation<T> constraintViolation : constraintViolations) {
            throw new ValidationException(constraintViolation.getMessage());
        }
    }

    public static <T> void validate(T t) {
        validate(t, false);
    }

    public static <T> void validate(List<T> list) {
        validate(list, false);
    }

    /**
     * @description: 参数校验
     * @date 11:59 2024/6/7
     * @param list
     * @param notNull list是否可以为空
     **/
    public static <T> void validate(List<T> list, boolean notNull) {
        if (list == null || list.isEmpty()) {
            if (notNull) {
                throw new ValidationException("参数不能为空");
            }
            return;
        }

        list.forEach(item -> {
            Set<ConstraintViolation<T>> constraintViolations = VALIDATOR.validate(item);
            if (constraintViolations.size() < 1) {
                return;
            }
            for (ConstraintViolation<T> constraintViolation : constraintViolations) {
                throw new ValidationException(constraintViolation.getMessage());
            }
        });
    }

    /**
     * 通过组来校验实体类
     * @param t 参数
     * @param notNull t是否可以为空
     * @param groups 校验组
     */
    public static <T> void validate(T t, boolean notNull, Class<?>... groups) {
        if (t == null) {
            if (notNull) {
                throw new ValidationException("参数不能为空");
            }
            return;
        }
        Set<ConstraintViolation<T>> constraintViolations = VALIDATOR.validate(t, groups);
        if (constraintViolations.size() < 1) {
            return;
        }
        for (ConstraintViolation<T> constraintViolation : constraintViolations) {
            throw new ValidationException(constraintViolation.getMessage());
        }
    }

    public static <T> void validate(T t, Class<?>... groups) {
        validate(t, false, groups);
    }
}
  • 使用
java 复制代码
// 方法入参
public class RefundOrderParam {

    @ApiModelProperty(value = "总退款金额", required = true)
    @NotNull(message = "总退款金额不能为空", groups = {InsertGroup.class, UpdateGroup.class})
    private BigDecimal totalRefundMoney;

    @ApiModelProperty(value = "是否原路退", required = true)
    @NotNull(message = "是否原路退不能为空", groups = {InsertGroup.class, UpdateGroup.class})
    private Boolean originalRefund;
}
//触发校验
ValidateUtil.validate(request, UpdateGroup.class);
相关推荐
hqxstudying27 分钟前
java依赖注入方法
java·spring·log4j·ioc·依赖
·云扬·35 分钟前
【Java源码阅读系列37】深度解读Java BufferedReader 源码
java·开发语言
martinzh2 小时前
Spring AI 项目介绍
后端
Bug退退退1232 小时前
RabbitMQ 高级特性之重试机制
java·分布式·spring·rabbitmq
小皮侠2 小时前
nginx的使用
java·运维·服务器·前端·git·nginx·github
前端付豪2 小时前
20、用 Python + API 打造终端天气预报工具(支持城市查询、天气图标、美化输出🧊
后端·python
爱学习的小学渣2 小时前
关系型数据库
后端
武子康2 小时前
大数据-33 HBase 整体架构 HMaster HRegion
大数据·后端·hbase
前端付豪2 小时前
19、用 Python + OpenAI 构建一个命令行 AI 问答助手
后端·python
凌览2 小时前
斩获 27k Star,一款开源的网站统计工具
前端·javascript·后端