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);
相关推荐
Cg1362691597415 分钟前
多态的定义
java·开发语言
云霄IT20 分钟前
新版电脑微信4.1.x.x小程序逆向之——寻找小程序存放位置目录和__APP__.wxapkg
java·微信·小程序
微信api接口介绍20 分钟前
微信社群管理开发
java·开发语言·网络·微信
渣哥24 分钟前
用错注入方式?你的代码可能早就埋下隐患
javascript·后端·面试
王中阳Go36 分钟前
我发现不管是Java还是Golang,懂AI之后,是真吃香!
后端·go·ai编程
亲爱的马哥1 小时前
再见,TDuckX3.0 结束了
前端·后端·github
我是天龙_绍1 小时前
redis 秒杀 分布式 锁
后端
AAA修煤气灶刘哥1 小时前
Spring AI 通关秘籍:从聊天到业务落地,Java 选手再也不用馋 Python 了!
后端·spring·openai
「QT(C++)开发工程师」1 小时前
C++语言编程规范-并发
java·linux·c++
Meteors.1 小时前
23种设计模式——迭代器模式 (Iterator Pattern)详解
java·设计模式·迭代器模式