如何创建一个泛型方法来返回枚举值数组而不是其常量？

Question

我有一个枚举，其描述为字符串。这是枚举。

public enum tabs
{
    A("Actual data"),
    B("Bad data"),
    C("Can data"),
    D("Direct data");

    private String description;
    tabs(String desc)
    {
        this.description = desc;
    }


    public String getDescription()
    {
        return this.description;
    }
}

现在，如果我需要 A 的数据，我会这样做

tabs.A.description

我正在创建一个通用方法来返回值。

我已经设法利用下面的通用方法来接受枚举类型作为参数。 它返回枚举常量的数组。 但我实际上希望它返回一个值数组。即 {"Actual data", "Bad data", "Can data", "Direct data"}。

public static String[] getActualTabs(Class<? extends Enum<?>> e) 
{
    return Arrays.stream(e.getEnumConstants()).map(Enum::name).toArray(String[]::new);
}

我有哪些选择来实现这一目标？

Answer 1

您可以传递一个映射器Function 将枚举常量映射到Strings：

public static <T extends Enum<T>> String[] getActualTabs(Class<T> e, Function<? super T, String> mapper) 
{
    return Arrays.stream(e.getEnumConstants()).map(mapper).toArray(String[]::new);
}

用法：

System.out.println(Arrays.toString(getActualTabs(tabs.class,tabs::getDescription)));

输出：

[Actual data, Bad data, Can data, Direct data]

Answer 2

我首先将getDescription 拉出一个接口，并确保您项目的所有枚举都实现它：

public interface HasDescription {
    String getDescription();
}

然后，在你的通用方法中使用它：

public static String[] getActualTabs(Class<? extends Enum<?> & HasDescription> e) {
    return Arrays.stream(e.getEnumConstants())
                 .map(HasDescription::getDescription)
                 .toArray(String[]::new);
}

Answer 3

两个答案都很漂亮。

但还有另一种变体可能更优越，因为它的使用范围更广：

public static <T> String[] getActualTabs(T[] values, Function<? super T, String> mapper) {
    return Arrays.stream(values)
                 .map(mapper)
                 .toArray(String[]::new);
}

public static <T extends HasDescription> String[] getActualTabs(T[] values) {
    return Arrays.stream(values)
                 .map(HasDescription::getDescription)
                 .toArray(String[]::new);
    // or just: return getActualTabs(values, HasDescription::getDescription);
}

这种实现的好处是我们不必乱用反射。

相反，您可以这样称呼它们：

String[] tabs1 = getActualTabs(Tabs.values())
String[] tabs2 = getActualTabs(Tabs.values(), Tabs::getDescription)

请注意，重命名函数可能很有用，例如。 G。 getStringifiedValues() 或 toDescriptions()。

免责声明：未经测试，但应该可以工作，恕我直言。