【java】自定义校验Bean数字只能为1和5

小半 • 2023年2月5日上午10:45 • 技术随笔 • 阅读 163

1.校验状态接口

package com.test;

import com.test.ValidStatusValidator;
import javax.validation.Constraint;
import javax.validation.Payload;
import javax.validation.constraints.NotNull;
import java.lang.annotation.*;

import static java.lang.annotation.ElementType.*;

@Target({METHOD, FIELD,ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(
        validatedBy = {ValidStatusValidator.class}
)
public @interface ValidStatus {

    // 默认错误提示
    String message() default "状态错误";

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

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


    @Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    @interface List {
        NotNull[] value();
    }
}

2.校验器

package com.test;

import com.google.common.collect.Lists;
import com.test.ValidStatus;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.List;

public class ValidStatusValidator implements ConstraintValidator<ValidStatus,Integer> {

    private List<Integer> status = Lists.newArrayList(1,5);

    @Override
    public void initialize(ValidStatus constraintAnnotation) {

    }

    @Override
    public boolean isValid(Integer s, ConstraintValidatorContext constraintValidatorContext) {
        if(s != null && !Func.contains(this.status.iterator(),s)){
            return false;
        }
        return true;
    }
}

3.校验的bean

package com.test;

import com.xiaoi.ibn.faq.app.validation.annotation.ValidStatus;
import lombok.Data;

import javax.validation.constraints.NotNull;
import java.io.Serializable;
import java.util.List;
@Data
public class Status implements Serializable {
    @ValidStatus(message = "仅允许输入1或5")
    @NotNull(message = "status不能为空")
    private int status;
}