如何使用ENUM从三个字符串中返回随机字符串？

Question

通过使用下面的代码，我可以找出我所在的数据中心，它工作正常。。

public enum DatacenterEnum {

    DEV, DC1, DC2, DC3;

    private static DatacenterEnum compareLocation() {
        String ourhost = getHostName();

        for (DatacenterEnum dc : values()) {
            String namepart = "." + dc.name().toLowerCase() + ".";
            if (ourhost.indexOf(namepart) >= 0) {
            return dc;
            }
        }
        return null;// I don't want to do this now.
    }
}

但它可能无法找到任何数据中心，因此目前我正在返回null。

是否有任何直接的方式或单行命令，我可以在ENUM中随机返回DC1或DC2或DC3而不是返回null？

我知道一种方法是列出字符串，然后随机选择0到2（包括0到2）之间的任何整数，然后找到字符串。但它的代码太多了，事实上，它不仅仅是想看看有没有其他方法可以做到这一点？

我可以在枚举中直接使用任何简单而直接的方法吗？

Answer 1

您可以使用值（）方法，它返回一个数组。然后只需使用Math.random（）返回一个随机实例。

以下是一个示例：

public static void main (String[] args) {
    String[] arr = {"DEV","DC1","DC2","DC3"}; //returned by Enum.values(), you get the idea
    String randElement = arr[(int) ((Math.random() * 3) +1)]; 
    System.out.println(randElement);
}

基本上可以归结为生成一个介于1和n之间的随机数：）

Answer 2

这是一个适用于任何枚举的通用解决方案。

单一排除的方便方法：

public static <E extends Enum<E>> E getRandom(Class<E> aEnum, E exclude) {
    return getRandom(aEnum, Collections.singletonList(exclude));
}

用于一个或多个排除的通用方法：

public static <E extends Enum<E>> E getRandom(Class<E> aEnum, List<E> exclude){
    //Convert set of enums into a list
    List<E> enums = new ArrayList<E>(EnumSet.allOf(aEnum));

    //Remove items from the list that you want to exclude
    enums.removeAll(exclude);

    int size = enums.size();
    if(size != 0){
        //Get random enum
        int randomIndex = new Random().nextInt(size);
        return enums.get(randomIndex);
    } else {
        throw new IllegalArgumentException("Empty Enumeration after excludes");
    }
}

对于你的例子，你可以打电话

EnumUtil.getRandom(DatacenterEnum.class, DatacenterEnum.DEV);

Answer 3

这里有一句话：

return DataCenterEnum.values()[new Random().nextInt(3) + 1)];

对于那些需要对代码进行更严格控制的人，这里有一个更安全的版本，它不依赖于枚举实例的顺序：

return new DataCenterEnum[]{DC1, DC2, DC3}[new Random().nextInt(3)];