Effective Java 第三版——37. 使用EnumMap替代序数索引

Tips
《Effective Java, Third Edition》一书英文版已经出版，这本书的第二版想必不少人都读过，号称Java四大名著之一，不过第二版2009年出版，到如今已经将近8年的时间，但随着Java 6，7，8，甚至9的发布，Java语言发生了深入的变化。
在这里第一时间翻译成中文版。供你们学习分享之用。

37. 使用EnumMap替代序数索引

有时可能会看到使用ordinal方法（条目 35）来索引到数组或列表的代码。 例如，考虑一下这个简单的类来表明一种植物：

class Plant {
    enum LifeCycle { ANNUAL, PERENNIAL, BIENNIAL }
    final String name;
    final LifeCycle lifeCycle;

    Plant(String name, LifeCycle lifeCycle) {
        [this.name](http://this.name) = name;
        this.lifeCycle = lifeCycle;
    }

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

如今假设你有一组植物表明一个花园，想要列出这些由生命周期组织的植物(一年生，多年生，或双年生)。为此，须要构建三个集合，每一个生命周期做为一个，并遍历整个花园，将每一个植物放置在适当的集合中。一些程序员能够经过将这些集合放入一个由生命周期序数索引的数组中来实现这一点：

// Using ordinal() to index into an array - DON'T DO THIS!

Set<Plant>[] plantsByLifeCycle =

    (Set<Plant>[]) new Set[Plant.LifeCycle.values().length];

for (int i = 0; i < plantsByLifeCycle.length; i++)

    plantsByLifeCycle[i] = new HashSet<>();

for (Plant p : garden)

    plantsByLifeCycle[p.lifeCycle.ordinal()].add(p);

// Print the results

for (int i = 0; i < plantsByLifeCycle.length; i++) {

    System.out.printf("%s: %s%n",

        Plant.LifeCycle.values()[i], plantsByLifeCycle[i]);

}

这种方法是有效的，但充满了问题。 由于数组不兼容泛型（条目 28），程序须要一个未经检查的转换，而且不会干净地编译。 因为该数组不知道索引表明什么，所以必须手动标记索引输出。 可是这种技术最严重的问题是，当你访问一个由枚举序数索引的数组时，你有责任使用正确的int值; int不提供枚举的类型安全性。 若是你使用了错误的值，程序会默默地作错误的事情，若是你幸运的话，抛出一个ArrayIndexOutOfBoundsException异常。

有一个更好的方法来达到一样的效果。 该数组有效地用做从枚举到值的映射，所以不妨使用Map。 更具体地说，有一个很是快速的Map实现，设计用于枚举键，称为java.util.EnumMap。 下面是当程序重写为使用EnumMap时的样子：

// Using an EnumMap to associate data with an enum

Map<Plant.LifeCycle, Set<Plant>>  plantsByLifeCycle =

    new EnumMap<>(Plant.LifeCycle.class);

for (Plant.LifeCycle lc : Plant.LifeCycle.values())

    plantsByLifeCycle.put(lc, new HashSet<>());

for (Plant p : garden)

    plantsByLifeCycle.get(p.lifeCycle).add(p);

System.out.println(plantsByLifeCycle);

这段程序更简短，更清晰，更安全，运行速度与原始版本至关。 没有不安全的转换; 无需手动标记输出，由于map键是知道如何将本身转换为可打印字符串的枚举; 而且不可能在计算数组索引时出错。 EnumMap与序数索引数组的速度至关，其缘由是EnumMap内部使用了这样一个数组，但它对程序员的隐藏了这个实现细节，将Map的丰富性和类型安全性与数组的速度相结合。 请注意，EnumMap构造方法接受键类型的Class对象：这是一个有限定的类型令牌（bounded type token），它提供运行时的泛型类型信息（条目 33）。

经过使用stream（条目 45）来管理Map，能够进一步缩短之前的程序。 如下是最简单的基于stream的代码，它们在很大程度上重复了前面示例的行为：

// Naive stream-based approach - unlikely to produce an EnumMap!

System.out.println(Arrays.stream(garden)

        .collect(groupingBy(p -> p.lifeCycle)));

这个代码的问题在于它选择了本身的Map实现，实际上它不是EnumMap，因此它不会与显式EnumMap的版本的空间和时间性能相匹配。 为了解决这个问题，使用Collectors.groupingBy的三个参数形式的方法，它容许调用者使用mapFactory参数指定map的实现：

// Using a stream and an EnumMap to associate data with an enum

System.out.println(Arrays.stream(garden)

        .collect(groupingBy(p -> p.lifeCycle,

() -> new EnumMap<>(LifeCycle.class), toSet())));

这样的优化在像这样的示例程序中是不值得的，可是在大量使用Map的程序中多是相当重要的。

基于stream版本的行为与EmumMap版本的行为略有不一样。 EnumMap版本老是为每一个工厂生命周期生成一个嵌套map类，而若是花园包含一个或多个具备该生命周期的植物时，则基于流的版本才会生成嵌套map类。 所以，例如，若是花园包含一年生和多年生植物但没有两年生的植物，plantByLifeCycle的大小在EnumMap版本中为三个，在两个基于流的版本中为两个。

你可能会看到数组索引(两次)的数组，用序数来表示从两个枚举值的映射。例如，这个程序使用这样一个数组来映射两个阶段到一个阶段转换（phase transition）（液体到固体表示凝固，液体到气体表示沸腾等等）：

// Using ordinal() to index array of arrays - DON'T DO THIS!

public enum Phase {

    SOLID, LIQUID, GAS;

    public enum Transition {

        MELT, FREEZE, BOIL, CONDENSE, SUBLIME, DEPOSIT;

        // Rows indexed by from-ordinal, cols by to-ordinal

        private static final Transition[][] TRANSITIONS = {

            { null,    MELT,     SUBLIME },

            { FREEZE,  null,     BOIL    },

            { DEPOSIT, CONDENSE, null    }

        };

        // Returns the phase transition from one phase to another

        public static Transition from(Phase from, Phase to) {

            return TRANSITIONS[from.ordinal()][to.ordinal()];

        }

    }

}

这段程序能够运行，甚至可能显得优雅，但外观多是骗人的。 就像前面显示的简单的花园示例同样，编译器没法知道序数和数组索引之间的关系。 若是在转换表中出错或者在修改Phase或Phase.Transition枚举类型时忘记更新它，则程序在运行时将失败。 失败多是ArrayIndexOutOfBoundsException，NullPointerException或（更糟糕的）沉默无提示的错误行为。 即便非空条目的数量较小，表格的大小也是phase的个数的平方。

一样，能够用EnumMap作得更好。 由于每一个阶段转换都由一对阶段枚举来索引，因此最好将关系表示为从一个枚举（from 阶段）到第二个枚举（to阶段）到结果（阶段转换）的map。 与阶段转换相关的两个阶段最好经过将它们与阶段转换枚举相关联来捕获，而后能够用它来初始化嵌套的EnumMap：

// Using a nested EnumMap to associate data with enum pairs

public enum Phase {

   SOLID, LIQUID, GAS;

   public enum Transition {

      MELT(SOLID, LIQUID), FREEZE(LIQUID, SOLID),

      BOIL(LIQUID, GAS),   CONDENSE(GAS, LIQUID),

      SUBLIME(SOLID, GAS), DEPOSIT(GAS, SOLID);

      private final Phase from;

      private final Phase to;

      Transition(Phase from, Phase to) {

         this.from = from;

         [this.to](http://this.to) = to;

      }

      // Initialize the phase transition map

      private static final Map<Phase, Map<Phase, Transition>>

        m = Stream.of(values()).collect(groupingBy(t -> t.from,

         () -> new EnumMap<>(Phase.class),

         toMap(t -> [t.to](http://t.to), t -> t,

            (x, y) -> y, () -> new EnumMap<>(Phase.class))));

      public static Transition from(Phase from, Phase to) {

         return m.get(from).get(to);

      }

   }

}

初始化阶段转换的map的代码有点复杂。map的类型是Map<Phase, Map<Phase, Transition>>，意思是“从（源）阶段映射到从（目标）阶段到阶段转换映射。”这个map的map使用两个收集器的级联序列进行初始化。 第一个收集器按源阶段对转换进行分组，第二个收集器使用从目标阶段到转换的映射建立一个EnumMap。 第二个收集器((x, y) -> y))中的合并方法未使用；仅仅由于咱们须要指定一个map工厂才能得到一个EnumMap，而且Collectors提供伸缩式工厂，这是必需的。 本书的前一版使用显式迭代来初始化阶段转换map。 代码更详细，但能够更容易理解。

如今假设想为系统添加一个新阶段：等离子体或电离气体。 这个阶段只有两个转变：电离，将气体转化为等离子体; 和去离子，将等离子体转化为气体。 要更新基于数组的程序，必须将一个新的常量添加到Phase，将两个两次添加到Phase.Transition，并用新的十六个元素版本替换原始的九元素阵列数组。 若是向数组中添加太多或太少的元素或者将元素乱序放置，那么若是运气不佳：程序将会编译，但在运行时会失败。 要更新基于EnumMap的版本，只需将PLASMA添加到阶段列表中，并将IONIZE(GAS, PLASMA)和DEIONIZE(PLASMA, GAS)添加到阶段转换列表中：

// Adding a new phase using the nested EnumMap implementation

public enum Phase {

    SOLID, LIQUID, GAS, PLASMA;

    public enum Transition {

        MELT(SOLID, LIQUID), FREEZE(LIQUID, SOLID),

        BOIL(LIQUID, GAS),   CONDENSE(GAS, LIQUID),

        SUBLIME(SOLID, GAS), DEPOSIT(GAS, SOLID),

        IONIZE(GAS, PLASMA), DEIONIZE(PLASMA, GAS);

        ... // Remainder unchanged

    }

}

该程序会处理全部其余事情，而且几乎不会出现错误。 在内部，map的map是经过数组的数组实现的，所以在空间或时间上花费不多，以增长清晰度，安全性和易于维护。

为了简便起见，上面的示例使用null来表示状态更改的缺失(其从目标到源都是相同的)。这不是很好的实践，极可能在运行时致使NullPointerException。为这个问题设计一个干净、优雅的解决方案是很是棘手的，并且结果程序足够长，以致于它们会偏离这个条目的主要内容。

总之，使用序数来索引数组很不合适：改用EnumMap。 若是你所表明的关系是多维的，请使用EnumMap <...，EnumMap <... >>。 应用程序员应该不多使用Enum.ordinal（条目 35），若是使用了，也是通常原则的特例。