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);
相关推荐
m0_5719575828 分钟前
Java | Leetcode Java题解之第543题二叉树的直径
java·leetcode·题解
魔道不误砍柴功3 小时前
Java 中如何巧妙应用 Function 让方法复用性更强
java·开发语言·python
NiNg_1_2343 小时前
SpringBoot整合SpringSecurity实现密码加密解密、登录认证退出功能
java·spring boot·后端
闲晨3 小时前
C++ 继承:代码传承的魔法棒,开启奇幻编程之旅
java·c语言·开发语言·c++·经验分享
种树人202408193 小时前
如何在 Spring Boot 中启用定时任务
spring boot
Chrikk4 小时前
Go-性能调优实战案例
开发语言·后端·golang
幼儿园老大*4 小时前
Go的环境搭建以及GoLand安装教程
开发语言·经验分享·后端·golang·go
canyuemanyue4 小时前
go语言连续监控事件并回调处理
开发语言·后端·golang
杜杜的man4 小时前
【go从零单排】go语言中的指针
开发语言·后端·golang
测开小菜鸟4 小时前
使用python向钉钉群聊发送消息
java·python·钉钉