Effective Java之EnumMap代替序数索引(三十三)

Map的实现类有很多种，EnumMap从名字我们可以看出这个Map是给枚举类用的。它的key为枚举元素，value自定义。在工作中我们也可以用其他的Map来实现我们关于枚举的需求，但是为什么要用这个EnumMap呢？
因为它的性能高！为什么性能高？
因为它的内部是用数组的数据结构来维护的！我们可以看一下它的源码实现：

Put方法

public V put(K key, V value) {  
    typeCheck(key);  

    int index = key.ordinal();  
    Object oldValue = vals[index];  
    vals[index] = maskNull(value);  
    if (oldValue == null)  
        size++;  
    return unmaskNull(oldValue);  
}  

typeCheck是用来检查key的类型的，因为key只能为枚举元素。接下来的这一句int index = key.ordinal();key.ordinal()这个就是我们上面说的枚举类型的序号，然后被当做数组的下标，放到vals这个数组里。那么get方法呢？

get方法

public V get(Object key) {  
    return (isValidKey(key) ?  
            unmaskNull(vals[((Enum<?>)key).ordinal()]) : null);  
}  

注意这一句话：vals[((Enum< ? >)key).ordinal()]。这个不就是取得下标，根据下标获取数组中的值吗？！

remove方法

public V remove(Object key) {  
    if (!isValidKey(key))  
        return null;  
    int index = ((Enum<?>)key).ordinal();  
    Object oldValue = vals[index];  
    vals[index] = null;  
    if (oldValue != null)  
        size--;  
    return unmaskNull(oldValue);  
}  

remove方法的实现也是挺简单的，就是把相应下标的元素变为null，等着GC回收。

下面是使用EnumMap的一个例子：

EnumMap<EnumTest01, String> enumMap = new EnumMap<EnumTest01, String>(EnumTest01.class);  
enumMap.put(EnumTest01.DELETE, "dsdsd");  
enumMap.put(EnumTest01.UPDATE, "qqqqqq");  
for (Map.Entry<EnumTest01, String> entry : enumMap.entrySet()) {  
    System.out.println(entry.getValue() + entry.getKey().getEnumDesc());  
}