判断某个值是否存在指定枚举类中

描述

在web项目开发过程中,对于一些类型字段通常会用 1，2，3 等数值表示。
例如：支付功能选择支付方式 1:余额 2:支付宝 3:微信 4:苹果支付等。
那么该字段只能存储 1，2，3，4这几个值，由于该值是前端传到后台的，无法保证正确因此后端通常会进行校验该值。

我的思路

对于该类型的字段通常会使用枚举类表示，如：

public enum PayRecordPayWayEnum {

	BALANCE(1,"余额"),          // 1:余额
	ALI(2,"支付宝"),            // 2:支付宝
	WECHAT(3, "微信"),          // 3:微信
	APP_STORE(4,"苹果支付")     // 4:苹果支付
	;

	private final Integer code;
	private final String description;

	private PayRecordPayWayEnum(Integer code, String description) {
		this.code = code;
		this.description = description;
	}

	public Integer getCode() { // 注意此方法名，在接下来的工具类中会使用到
		return code;
	}

	public String getDescription() {
		return description;
	}
}

我们需要的是，通过一个工具来判断某个code是否存在于该枚举类中。
解决方法：
通过java反射机制,注意需要保证枚举类中的 getCode() 和下面工具类中的一致。

public class EnumUtil {

    private static final Logger logger = LoggerFactory.getLogger(EnumUtil.class);

    /**
     * 判断某个枚举是否包某个code值
     * @param enumClass 需要判断是否存在那个枚举类中
     * @param code 需要判断的值
     * @return 包含返回true，否则返回false
     */
    public static boolean isInclude(Class enumClass, int code){
        List enumList = EnumUtils.getEnumList(enumClass);
        for (int i = 0;i<enumList.size(); i++){
            Object en = enumList.get(i);
            Class<?> enClass = en.getClass();
            try {
                Method method = enClass.getMethod("getCode"); // 需要与枚举类方法对应
                Object invoke = method.invoke(en, null);
                if(Integer.parseInt(invoke.toString()) == code) {
                    return true;
                }
            }catch (Exception e){
                logger.error("枚举执行getCode方法失败...");
            }
        }
        return false;
    }
    
    public static void main(String[] args) {
        boolean include1 = isInclude(PayRecordPayWayEnum.class, 3);
        System.out.println(include1); // true
        
        boolean include2 = isInclude(PayRecordPayWayEnum.class, 5);
        System.out.println(include2); // false
    }
}

最后

如果在这里获得过启发和思考，希望点赞支持！对于内容有不同的看法欢迎来信交流。
技术栈 >> java
邮箱 >> [email protected]