你知晓Java中的EnumMap吗?

提到Java中集合Map，你可能常用的是HashMap/ConcurrentHashMap，但是当Map的key是枚举类型时，是否想过使用EnumMap了？

那么当key是枚举类型时，EnumMap和HashMap的区别是啥？

• HashMap：使用Hash表存储，底层是数组，数组中存的是entry对象，默认长度是16。也就说每次向Map中加入对象，首先根据key值计算hash值，然后根据数组长度放到合适的位置，当发生冲突时，使用"开链法"，即在冲突位置加入链表；Java8在解决冲突时，又新增了红黑树的解决方案。总体来说，HashMap就是空间换时间的方式来提高访问效率。但是假如我们已知key的所有值时，是否还有必要每次都要计算hash值，然后找位置，解决冲突，甚者扩容了...
• EnumMap：既然已知了所有key，那么申请已知大小的数组，每次访问时根据枚举变量的ordinal值定位到数组的指定位置，存取即可，这样是不是既高效，且不会浪费存储中间。

下面就进入EnumMap的内部探秘吧

主要成员：

public class EnumMap<K extends Enum<K>, V> extends AbstractMap<K, V>
    implements java.io.Serializable, Cloneable {
	// key值的枚举类型
	private final Class<K> keyType;

	//根据key值的枚举类型，会预先缓存所有的枚举变量
	private transient K[] keyUniverse;

	/**
	 * 所有的vlaue值，数组的下标是枚举变量的序号（ordinal）
	 * 数组的内容是map中枚举变量对应的vlaue
	 */
	private transient Object[] vals;

	//当前map的大小，
	private transient int size = 0;
}

使用详解：

假设预先定了这样的Enum对象

public enum  WeekEnum {
    SUNDAY, MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY, SATURDAY;
}

构造方法：

// 构造方法使用示例
EnumMap<WeekEnum, String> weekEnumMap = new EnumMap<>(WeekEnum.class);
// 会预先获取所有的枚举值，然后根据枚举值的个数，创建相应大小的数组用来存放value
public EnumMap(Class<K> keyType) {
	this.keyType = keyType;
	keyUniverse = getKeyUniverse(keyType);
	vals = new Object[keyUniverse.length];
}

put方法：

// 检查枚举值类型，然后根据key的序号，将value存入相应下标的数组
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);
    }

get方法：

// 依然是先校验key值类型，然后根据key的序号去数组的位置取值
public V get(Object key) {
  return (isValidKey(key) ?
          unmaskNull(vals[((Enum<?>)key).ordinal()]) : null);
}