注解校验参数接收类字段两者必传其一

最新推荐文章于 2023-10-20 15:37:33 发布

C__jx

最新推荐文章于 2023-10-20 15:37:33 发布

阅读量630

点赞数

文章标签： java 开发语言

本文链接：https://blog.csdn.net/C__jx/article/details/131162858

版权

文章介绍了一个自定义的Java注解@AtLeastOneNotEmpty，用于在对象中确保至少有一个指定的字段不为空。注解支持多个字段的校验，并提供了对应的验证器逻辑，通过检查字段值是否为空来确定验证是否通过。该机制可以增强数据验证功能，适用于表单提交或数据处理场景。

摘要由CSDN通过智能技术生成

先上注解：

@AtLeastOneNotEmpty

使用：

@AtLeastOneNotEmpty(fields = {"name1","name2"},message = "name1,name2必填其一")
@AtLeastOneNotEmpty(fields = {"name3","name4"},message = "name3,name4必填其一")

上源码：

AtLeastOneNotEmpty 注解类

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(
    validatedBy = {AtLeastOneNotEmptyValidator.class}
)
@Repeatable(AtLeastOneNotEmpties.class)
@Documented
public @interface AtLeastOneNotEmpty {
    String message() default "至少有一个属性不可为空";

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

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

    String[] fields();
}

支持多个校验：

AtLeastOneNotEmpties



@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface AtLeastOneNotEmpties {
    AtLeastOneNotEmpty[] value();
}

逻辑实现代码：

public class AtLeastOneNotEmptyValidator extends AbstractValidator<AtLeastOneNotEmpty, Object> {
    private String[] fields;

    public AtLeastOneNotEmptyValidator() {
    }

    public void initialize(AtLeastOneNotEmpty atLeastOneNotEmpty) {
        this.fields = atLeastOneNotEmpty.fields();
    }

    public boolean check(Object object, WrapperHibernateConstraintValidatorContext wrapperHibernateConstraintValidatorContext) {
        if (object == null) {
            return true;
        } else {
            try {
                String[] var3 = this.fields;
                int var4 = var3.length;

                for(int var5 = 0; var5 < var4; ++var5) {
                    String fieldName = var3[var5];
                    Object value = this.getFieldValue(object, fieldName);
                    if (this.notEmpty(value)) {
                        return true;
                    }
                }

                return false;
            } catch (Exception var8) {
                throw new IllegalStateException(var8);
            }
        }
    }

    private Object getFieldValue(Object object, String fieldName) throws IllegalAccessException {
        if (object == null) {
            return null;
        } else {
            Class<?> clazz = object.getClass();
            Field[] classFields = clazz.getDeclaredFields();
            Field[] var5 = classFields;
            int var6 = classFields.length;

            for(int var7 = 0; var7 < var6; ++var7) {
                Field field = var5[var7];
                if (field.getName().equals(fieldName)) {
                    field.setAccessible(true);
                    return field.get(object);
                }
            }

            return null;
        }
    }

    private boolean notEmpty(Object value) {
        if (Objects.isNull(value)) {
            return false;
        } else {
            return value instanceof String ? StringUtils.isNotBlank((String)value) : true;
        }
    }
}