@Validated自定义条件注解

【背景】

  在做微信小程序对接的时候,我们常常会传入环境,譬如: 要打开的小程序版本。正式版为"release",体验版为"trial",开发版为"develop",但是每次都要去写校验太麻烦了,我想直接用@Validated注解来做,但是里面提供的都不太合适,所以自己搞了一个自定义注解。

【实现】

/**
 * @author yueF_L
 * @date 2023-03-21 14:01
 * 自定义条件注解
 */
@Target({ElementType.FIELD, ElementType.METHOD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER, ElementType.TYPE_USE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(validatedBy = {ContainValueConstraintValidator.class})
public @interface ContainValue {

    String message() default "数组不能为空";

    Class[] groups() default {};

    Class[] payload() default {};

    /**
     * 自定义一个类型来存放数据(数组)
     *
     * @return
     */
    String[] values() default {};

}

 这里引用了类 #ContainValueConstraintValidator,实现比较逻辑。

package com.spark.common.core.validate.constraint;

import com.spark.common.core.validate.ContainValue;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.HashSet;
import java.util.Set;

/**
 * @author yueF_L
 * @date 2023-03-21 14:03
 * 自定义条件注解实现方法
 */
public class ContainValueConstraintValidator implements ConstraintValidator {

    /**
     * set存储
     */
    private Set set = new HashSet<>();

    /**
     * ContainValue拿到的是注解中的数据
     */
    @Override
    public void initialize(ContainValue constraintAnnotation) {
        //拿到注解中自定义的数据,且是数组型的
        String[] values = constraintAnnotation.values();
        //放在数组里,遍历判断
        for (String value : values) {
            set.add(value);
        }
    }


    @Override
    public boolean isValid(String vStr, ConstraintValidatorContext constraintValidatorContext) {
        if (set.contains(vStr)) {
            return true;
        }
        return false;
    }
}

使用方式:

@Validated自定义条件注解_第1张图片

你可能感兴趣的:(java,java)