Spring Boot 中自定义注解实现可选 int 值的校验

正文:

在实际的应用开发中,我们经常需要对接口的输入进行校验,以确保数据的合法性。在Spring Boot项目中,通过使用Hibernate Validator等校验框架,可以轻松地实现各种校验需求。本文将介绍如何自定义一个注解@InIntValues,并通过对应的校验器InIntValuesValidator来实现对输入是否为指定可选int值的校验。

1. 创建自定义可选int值校验注解

首先,我们创建一个自定义注解@InIntValues,用于标记需要进行可选int值校验的字段或参数。以下是注解的定义:

java 复制代码
import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.*;

/**
 * 可选int值范围校验注解
 */
@Documented
@Constraint(validatedBy = InIntValuesValidator.class)
@Target({ElementType.FIELD, ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
public @interface InIntValues {
    // 默认错误提示信息
    String message() default "必须是指定的int值之一";

    // 可选的int值数组
    int[] values();

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

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

2. 创建可选int值校验器

接下来,我们需要实现可选int值校验的逻辑。为此,创建一个InIntValuesValidator类,实现ConstraintValidator<InIntValues, Integer>接口:

java 复制代码
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.Arrays;

/**
 * 可选int值范围校验器
 */
public class InIntValuesValidator implements ConstraintValidator<InIntValues, Integer> {

    private int[] allowedValues;

    @Override
    public void initialize(InIntValues constraintAnnotation) {
        this.allowedValues = constraintAnnotation.values();
    }

    /**
     * 校验方法
     * @param value 待校验的整数
     * @param context 校验上下文
     * @return 校验结果,true表示通过,false表示不通过
     */
    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
        // 如果值为null,由 @NotNull 校验
        if (value == null) {
            return true;
        }
        // 判断值是否在可选的int值数组中
        return Arrays.stream(allowedValues).anyMatch(val -> val == value);
    }
}

在这个校验器中,我们使用@InIntValues注解的values()属性来指定可选的int值数组。在isValid方法中,我们使用Java 8的Stream API来判断被验证的值是否在这个可选的int值数组中。

3. 在实体类或DTO中应用注解

现在,你可以在需要进行可选int值校验的字段上应用@InIntValues注解:

java 复制代码
public class YourRequestDTO {
    
    @NotBlank(message = "姓名不能为空")
    @Chinese
    private String name;

    @InIntValues(values = {1, 2, 3}, message = "必须是指定的int值之一")
    private Integer selectedValue;

    // 其他字段...
}

在这个例子中,selectedValue字段通过@InIntValues注解进行了可选int值校验。当请求中的selectedValue字段不是指定的可选int值之一时,将触发校验错误,并返回指定的错误提示信息。

通过这样的自定义注解和校验器,你可以在Spring Boot项目中轻松实现对字段值是否为指定可选int值的校验,提高系统的数据质量和安全性。

相关推荐
荔枝吻9 分钟前
【沉浸式解决问题】idea开发中mapper类中突然找不到对应实体类
java·intellij-idea·mybatis
snoopyfly~25 分钟前
Ubuntu 24.04 LTS 服务器配置:安装 JDK、Nginx、Redis。
java·服务器·ubuntu
追逐时光者29 分钟前
一款开源免费、通用的 WPF 主题控件包
后端·.net
挺菜的1 小时前
【算法刷题记录(简单题)003】统计大写字母个数(java代码实现)
java·数据结构·算法
蜗牛沐雨1 小时前
警惕 Rust 字符串的性能陷阱:`chars().nth()` 的深坑与高效之道
开发语言·后端·rust
&Sinnt&1 小时前
Git 版本控制完全指南:从入门到精通
git·后端
掘金-我是哪吒2 小时前
分布式微服务系统架构第156集:JavaPlus技术文档平台日更-Java线程池使用指南
java·分布式·微服务·云原生·架构
亲爱的非洲野猪2 小时前
Kafka消息积压的多维度解决方案:超越简单扩容的完整策略
java·分布式·中间件·kafka
陈随易2 小时前
MoonBit助力前端开发,加密&性能两不误,斐波那契测试提高3-4倍
前端·后端·程序员