重构 - 从枚举生成字符串列表

Question

方法 getContinent() 需要一个 String 作为参数，该参数可以具有三个可能的值："Africa"、"Asia"和<代码>“欧洲”。

根据我得到的给定值，我必须匹配适当的枚举并获取其所有值的列表（作为字符串）。

我的代码中有重复。 有更好的方法吗？ 将其放入每个 enum 中是个好主意吗？

Stream.of(CountryAsia.values())
      .map(country -> country.getName())
      .collect(Collectors.toList());

： 我的代码

Main 类：

public class Main {
    public static void main(String[] args) {
        String continent = getContinent() //one of three possible values: "Africa" or "Asia" or "Europe"
        List<String> list = null;

        switch(continent) {
            case "Africa":
                list = Stream.of(CountryAfrica.values())
                        .map(country -> country.getName())
                        .collect(Collectors.toList());
                break;
            case "Asia":
                list = Stream.of(CountryAsia.values())
                        .map(country -> country.getName())
                        .collect(Collectors.toList());
                break;
            case "Europe":
                list = Stream.of(CountryEurope.values())
                        .map(country -> country.getName())
                        .collect(Collectors.toList());
                break;
        }

        System.out.println(list);
        //next instructions..
    }
}

CountryEurope 枚举：

public enum CountryEurope {
    DENMARK("Denmark"),
    ESTONIA("Estonia"),
    ICELAND("Iceland");

    CountryEurope(String name) {
        this.name = name;
    }

    private final String name;

    public String getName() {
        return name;
    }
}

CountryAsia 枚举：

public enum CountryAsia {
    AFGHANISTAN("Afghanistan"),
    KAZAKHSTAN("Kazakhstan"),
    UNITED_ARAB_EMIRATES("United Arab Emirates"),
    UZBEKISTAN("Uzbekistan");

    CountryAsia(String name) {
        this.name = name;
    }

    private final String name;

    public String getName() {
        return name;
    }
}

CountryAfrica 类：

public enum CountryAfrica {
    ALGIERIA("Algeria"),
    WESTERN_SAHARA("Western Sahara"),
    EGYPT("Egypt");

    CountryAfrica(String name) {
        this.name = name;
    }

    private final String name;

    public String getName() {
        return name;
    }
}

Answer 1

您可以使用 EnumSet.allOf() 以便获取 enum 常量的名称。

EnumSet.allOf() 需要一个 enum 的Class 作为参数，并返回其 EnumSet元素（这是特殊的实现专门为enum设计的Set）。

然后，我们可以使用流或纯文本将enum常量的Set转换为字符串列表 for 循环。

如果您使用的是Java 14 +，则可以使用切换表达式。否则，用方法调用替换重复的流语句。

为了使用这些enum中的每个元素（无论其类型如何）访问方法getName()，我们可以定义一个接口，其中所有他们将实施。这样它将作为所有国家/地区枚举的入口点。

public interface CountryData {
    String getName();
}

此外，它还可能包含一些与特定于域的信息相关的其他方法，这些信息可能在您的应用程序中有用，例如国家/地区或地区税率（所有这些行为都可以通过 CountryData 接口访问）。

结合上面提到的所有内容将给出非常简洁且富有表现力的代码：

public static List<String> getCountries(String continent) {
    return switch(continent) {
        case "Africa" -> getEnumNames(CountryAfrica.class);
        case "Asia" -> getEnumNames(CountryAsia.class);
        case "Europe" -> getEnumNames(CountryEurope.class);
        default -> Collections.emptyList();
    };
}

public static <T extends Enum<T> & CountryData> List<String> getEnumNames(Class<T> enumClas) {
    return EnumSet.allOf(enumClas).stream()
            .map(CountryData::getName)
            .collect(Collectors.toList());
}

上面的方法声明了一个有界类型参数 ; & CountryData>，这意味着任何类型 T 都是 Enum 类和接口 CountryData。

具有多个边界的类型变量是所有类型的子类型
列在绑定中。如果其中一个边界是类，则它必须是
首先指定。

有关泛型方法语法的信息，请查看本教程

main()

public static void main(String[] args) {
    System.out.println(getCountries("Europe"));
    System.out.println(getCountries("Africa"));
    System.out.println(getCountries("Asia"));
}

输出

[Denmark, Estonia, Iceland]
[Algeria, Western Sahara, Egypt]
[Afghanistan, Kazakhstan, United Arab Emirates, Uzbekistan]

将其放入每个枚举中是个好主意吗？

不，这并不是因为：

• 代码不会消除重复，相同的行会分散在多个文件中；
• 此代码并不特定于您的任何枚举，因此它必须驻留在一个位置（在用于处理此列表的类中）。

Answer 2

看起来您尝试在枚举中存储一些数据集合，恕我直言，这有点错误。我更愿意提供一张地图，其中键是“大陆”，值是国家/地区的集合。映射<字符串，集合>。

然后，你不需要使用开关，你可以使用Map#get()。