通過自定義枚舉注解
@EnumValidator(value = SexEnums.class)，
private Integer sex;
就可以校驗入?yún)⒅当仨氃谥付杜e類中
原創(chuàng):http://m.itdecent.cn/p/32f0d6e3afbb

目錄

• 自定義枚舉校驗注解
• 使用枚舉校驗注解
• 演示
• 附

自定義枚舉校驗注解

1.定義自定義枚舉@EnumValidator

package com.enums.validate.base;

import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.*;

/**
 * 枚舉值校驗注解
 *
 * @author: zetting
 * @date:2018/12/18
 */
@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.FIELD, ElementType.METHOD})
@Constraint(validatedBy = EnumValidatorClass.class)
public @interface EnumValidator {
    Class<?> value();

    String message() default "入?yún)⒅挡辉谡_枚舉中";

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

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

2.定義實(shí)現(xiàn)枚舉值校驗處理類EnumValidatorClass

 
package com.enums.validate.base;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

/**
 * 枚舉值校驗注解實(shí)現(xiàn)
 *
 * @author: zetting
 * @date:2018/12/18
 */
public class EnumValidatorClass implements ConstraintValidator<EnumValidator, Object>, Annotation {
    private Logger log = LoggerFactory.getLogger(this.getClass());
    private List<Object> values = new ArrayList<>();

    @Override
    public void initialize(EnumValidator enumValidator) {
        Class<?> clz = enumValidator.value();
        Object[] ojects = clz.getEnumConstants();
        try {
            Method method = clz.getMethod("getValue");
            if (Objects.isNull(method)) {
                throw new Exception(String.format("枚舉對象{}缺少字段名為value的字段",
                        clz.getName()));
            }
            Object value = null;
            for (Object obj : ojects) {
                value = method.invoke(obj);
                values.add(value);
            }
        } catch (Exception e) {
            log.error("[處理枚舉校驗異常]", e);
        }
    }


    @Override
    public Class<? extends Annotation> annotationType() {
        return null;
    }

    @Override
    public boolean isValid(Object value, ConstraintValidatorContext constraintValidatorContext) {
        return Objects.isNull(value) || values.contains(value) ? true : false;
    }
}

使用枚舉校驗注解

1.定義要校驗的枚舉值

package com.enums.validate.modules.enums;


/**
 * 性別枚舉
 *
 * @author: zetting
 * @date: 2018/12/20 22:41
 */
public enum SexEnums {
    /**
     * 未知
     */
    UNKNOWN(0),

    /**
     * 男
     */
    MALE(1),

    /**
     * 女
     */
    FEMALE(2);

    private int value;

    SexEnums(int value) {
        this.value = value;
    }

    public int getValue() {
        return value;
    }}

2.在相應(yīng)要檢驗的字段加上@EnumValidator注解

package com.enums.validate.modules.dto;

import com.enums.validate.aop.Request;
import com.enums.validate.base.EnumValidator;
import com.enums.validate.modules.enums.SexEnums;
import org.hibernate.validator.constraints.NotBlank;

/**
 * 入?yún)⑿ｒ炚埱? *
 * @author: zetting
 * @date: 2018/12/19 22:16
 */
public class ValidateRequest extends Request {
    /**
     * 用戶名
     */
    @NotBlank
    private String userName;

    /**
     * 性別
     */
    @EnumValidator(value = SexEnums.class)
    private Integer sex;

    /**
     * 地址
     */
    private String address;

    public String getUserName() {
        return userName;
    }

    public void setUserName(String userName) {
        this.userName = userName;
    }

    public Integer getSex() {
        return sex;
    }

    public void setSex(Integer sex) {
        this.sex = sex;
    }

    public String getAddress() {
        return address;
    }

    public void setAddress(String address) {
        this.address = address;
    }
}

2.controller層正常使用、無需添加任何東西

package com.enums.validate.modules.controller;

import com.enums.validate.base.Response;
import com.enums.validate.modules.dto.ValidateRequest;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * 枚舉校驗
 */
@RestController
public class MyController {

    @GetMapping(value = "/validate")
    public Response validate(ValidateRequest request) {
        return Response.success();
    }
}

演示

1.入?yún)⒉辉倜杜e值中

入?yún)⒉辉僦付ǖ拿杜e中.jpg

2.入?yún)⒃诿杜e值中

入?yún)⒃谥付ǖ拿杜e中.jpg

附

gitee源碼：
https://gitee.com/zetting/my-gather/tree/master/springboot-enums-validate