Question

假设我有以下java enum

public enum Color {
  RED(10),
  GREEN(22),
  BLUE(33);

  private int value;

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

  public int intValue() {
    return value;
  }
}

为了能够通过给定的整数值获取颜色实例，我需要添加如下方法：

public static Color getInstance(int value) {
  switch (value) {
    case 10: return RED;
    case 22: return GREEN;
    case 33: return BLUE;
    default: throw new IllegalArgumentException("Invalid color value");
  }
}

是否有可能从IDE自动生成此方法？（最好是IntelliJ）？

Answer 1

您可以为具有enum值的所有类型的枚举实现中央实用程序方法，而不是重复每个int中的代码。这是可能的，因为enum可以实现接口，因此您可以定义访问int值的统一方式：

public interface IntValued {
    int intValue();
}
/** get the particular {@link IntValued} {@code enum} constant. */
public static <T extends Enum<T>&IntValued> T get(Class<T> type, int value) {
    return type.cast(GET.get(type).apply(value));
}
private static ClassValue<IntFunction<Enum>> GET = new ClassValue<IntFunction<Enum>>() {
    protected IntFunction<Enum> computeValue(Class<?> type) {
        return prepare(type);
    }
};
// invoked only once per enum type
static IntFunction<Enum> prepare(Class<?> type) {
    Enum[] values=type.asSubclass(Enum.class).getEnumConstants();
    if(values.length==0) return i -> null;
    IntSummaryStatistics s=Arrays.stream(values)
        .mapToInt(o -> ((IntValued)o).intValue())
        .summaryStatistics();
    int min=s.getMin(), max=s.getMax();
    if((max-min+1)<s.getCount()*2) {
        Enum[] linear=new Enum[max-min+1];
        for(Enum e: values) linear[((IntValued)e).intValue()-min]=e;
        return i -> i<min||i>max? null: linear[i-min];
    }
    Map<Integer, Enum> map = Arrays.stream(values).collect(
        Collectors.toMap(o -> ((IntValued)o).intValue(), Function.identity()));
    return map::get;
}

这使用ClassValue，这是一种将自定义元数据与类相关联的JRE方法。 ClassValue只负责调用初始化代码一次，并且仍然允许垃圾收集/类卸载。上面的代码动态地确定要使用的查找结构。如果特定int类型的enum值是密集的，则它使用数组，否则它将转换为Map。

您可以按如下方式使用它：

public enum Color implements IntValued {
    RED(10),
    GREEN(22),
    BLUE(33);

    private int value;

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

    public int intValue() {
      return value;
    }
}

Color color=get(Color.class, 22);
System.out.println(color);

请注意，与生成的switch语句不同，此解决方案在这些int值发生更改时不会中断。

Answer 2

如果您使用的是Java 8，则可以使用default中的interface方法向任何类添加功能 - 甚至是enum。

看一下here，我只需将enum添加到implement ReverseLookup即可向其添加反向查询。

Java枚举自动生成getInstance方法？

2 个答案: