SpringBoot——校验参数是否包含在枚举属性值中

末蓝、 2024-03-26 19:06 117阅读 0赞

## 需求 ##

在一些枚举类调用中，我们需要对一些枚举的属性值做包含关系的判断，看是否匹配。通过反射机制获取对应方法，再获取对应的枚举属性值。下面是示例实现。

## 代码实现 ##

### 工具类 ###

@Slf4j
    public class EnumOperateUtil {
        
    
        /**
         * 校验枚举，是否包含在枚举属性值中
         *
         * @param enumClass  枚举class
         * @param methodName 属性值调用方法名称
         * @param value      目标属性值
         * @return
         */
        public static boolean validInclude(Class enumClass, String methodName, String value) {
        
            //获取枚举list
            List enumList = EnumUtils.getEnumList(enumClass);
            //循环处理
            for (int i = 0; i < enumList.size(); i++) {
        
                //获取枚举
                Object en = enumList.get(i);
                //获取枚举的class
                Class<?> eClass = en.getClass();
                try {
        
                    //反射获取方法
                    Method method = eClass.getMethod(methodName);
                    //调用枚举方法
                    Object invokeRes = method.invoke(en, null);
                    //获取方法调用结果，即属性值，判断
                    if (invokeRes.toString().equals(value)) {
        
                        //相同返回true
                        return true;
                    }
                } catch (Exception e) {
        
                    //异常处理
                    log.error("validInclude()失败：错误信息：枚举{}方法执行失败！{}", methodName, e.getMessage());
                }
            }
            //不同或异常则返回false
            return false;
        }
    }

### 枚举类 ###

public enum ColourEnum {
        
    	RED(1, "红色"),
    	YELLOW(2, "黄色"),
    	GREEN(3, "绿色");
    	private Integer code;
    	private String desc;
    	ColourEnmu(Integer code, String desc) {
        
    		this.code = code;
    		this.desc = desc;
    	}
    	public Integer getCode() {
        
    		return code;
    	}
    	public String getDesc() {
        
    		return desc;
    	}
    }

### 使用方法 ###

Integer code = 4;
    Boolean includeCheckFlag = EnumOperateUtil.validInclude(ColourEnum.class, "getCode", code);