在 Java 枚举中覆盖 valueof() 和 toString()

Question

enum 中的值是需要在其中包含空格的单词，但枚举的值中不能包含空格，因此它们都是捆绑在一起的。我想覆盖 toString() 在我告诉它的地方添加这些空间。

当我在添加空格的同一字符串上使用 valueOf() 时，我还希望枚举提供正确的枚举。

例如：

public enum RandomEnum
{
     StartHere,
     StopHere
}

在值为StartHere 的RandomEnum 上调用toString()，返回字符串"Start Here"。对同一字符串 ("Start Here") 调用 valueof() 返回枚举值 StartHere。

我该怎么做？

Answer 1

我认为你不会让 valueOf("Start Here") 工作。 但就空间而言...尝试以下...

static private enum RandomEnum {
    R("Start There"), 
    G("Start Here"); 
    String value;
    RandomEnum(String s) {
        value = s;
    }
}

System.out.println(RandomEnum.G.value);
System.out.println(RandomEnum.valueOf("G").value);

Start Here
Start Here

Answer 2

您可以试用此代码。由于您无法覆盖 valueOf 方法，因此您必须定义一个自定义方法（下面的示例代码中的 getEnum），它返回您需要的值并更改您的客户端以改为使用此方法。

public enum RandomEnum {

    StartHere("Start Here"),
    StopHere("Stop Here");

    private String value;

    RandomEnum(String value) {
        this.value = value;
    }

    public String getValue() {
        return value;
    }

    @Override
    public String toString() {
        return this.getValue();
    }

    public static RandomEnum getEnum(String value) {
        for(RandomEnum v : values())
            if(v.getValue().equalsIgnoreCase(value)) return v;
        throw new IllegalArgumentException();
    }
}

Answer 3

以下是 valueOf() 的一个很好的通用替代方法

public static RandomEnum getEnum(String value) {
  for (RandomEnum re : RandomEnum.values()) {
    if (re.description.compareTo(value) == 0) {
      return re;
    }
  }
  throw new IllegalArgumentException("Invalid RandomEnum value: " + value);
}

Answer 4

您可以在枚举中使用静态映射，将字符串映射到枚举常量。在“getEnum”静态方法中使用它。这样就无需在每次您想从其 String 值中获取枚举时遍历枚举。

public enum RandomEnum {

    StartHere("Start Here"),
    StopHere("Stop Here");

    private final String strVal;
    private RandomEnum(String strVal) {
        this.strVal = strVal;
    }

    public static RandomEnum getEnum(String strVal) {
        if(!strValMap.containsKey(strVal)) {
            throw new IllegalArgumentException("Unknown String Value: " + strVal);
        }
        return strValMap.get(strVal);
    }

    private static final Map<String, RandomEnum> strValMap;
    static {
        final Map<String, RandomEnum> tmpMap = Maps.newHashMap();
        for(final RandomEnum en : RandomEnum.values()) {
            tmpMap.put(en.strVal, en);
        }
        strValMap = ImmutableMap.copyOf(tmpMap);
    }

    @Override
    public String toString() {
        return strVal;
    }
}

只要确保映射的静态初始化发生在枚举常量声明的下方即可。

顺便说一句 - 'ImmutableMap' 类型来自 Google guava API，在这种情况下我绝对推荐它。

---

编辑 - 根据 cmets：

1. 此解决方案假定每个分配的字符串值都是唯一且非空的。鉴于枚举的创建者可以控制这一点，并且字符串对应于唯一且非空的枚举值，这似乎是一个安全的限制。
2. 我添加了问题中要求的“toSTRing()”方法

Answer 5

试试这个，但我不确定它是否会在任何地方都有效:)

public enum MyEnum {
    A("Start There"),
    B("Start Here");

    MyEnum(String name) {
        try {
            Field fieldName = getClass().getSuperclass().getDeclaredField("name");
            fieldName.setAccessible(true);
            fieldName.set(this, name);
            fieldName.setAccessible(false);
        } catch (Exception e) {}
    }
}

Answer 6

Java 8 实现怎么样？ （null 可以替换为你的默认枚举）

public static RandomEnum getEnum(String value) {
    return Arrays.stream(RandomEnum.values()).filter(m -> m.value.equals(value)).findAny().orElse(null);
}

或者你可以使用：

...findAny().orElseThrow(NotFoundException::new);

Answer 7

你仍然可以选择在你的枚举中实现这个：

public static <T extends Enum<T>> T valueOf(Class<T> enumType, String name){...}