Java EnumMap工作原理及实现(二)

1.概述

A specialized Map implementation for use with enum type keys. All of the keys in an enum map must come from a single enum type that is specified, explicitly or implicitly, when the map is created. Enum maps are represented internally as arrays. This representation is extremely compact and efficient.

EnumMap是是一种键为枚举类型的特殊的Map实现。所有的Key也必须是一种枚举类型，EnumMap是使用数组来实现的。

1 2 3 4 5 6 7 8

EnumMap<Course, String> map = new EnumMap<Course, String>(Course.class); map.put(Course.ONE, "语文"); map.put(Course.ONE, "政治"); map.put(Course.TWO, "数学"); map.put(Course.THREE, "英语"); for(Entry<Course, String> entry : map.entrySet()) { System.out.println(entry.getKey() + ": " + entry.getValue()); }

输出结果为：

1 2 3

ONE: 政治 TWO: 数学 THREE: 英语

其具体实现的结构如下图所示：

2. put和get方法

put方法通过key的ordinal将值存储到对应的地方，get方法则根据key的ordinal获取对应的值。

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

public V put(K key, V value) { // 类型检查 typeCheck(key); // 获取key的序号 int index = key.ordinal(); Object oldValue = vals[index]; // 赋值 vals[index] = maskNull(value); // 若之前的值为空，则size++ if (oldValue == null) size++; return unmaskNull(oldValue); } public V get(Object key) { return (isValidKey(key) ? unmaskNull(vals[((Enum<?>)key).ordinal()]) : null); }

3. 遍历

EnumMapIterator的迭代这样实现的：

1 2 3 4 5 6 7 8 9 10 11 12

public boolean hasNext() { while (index < vals.length && vals[index] == null) index++; return index != vals.length; } public Map.Entry<K,V> next() { if (!hasNext()) throw new NoSuchElementException(); lastReturnedEntry = new Entry(index++); return lastReturnedEntry; }

通过hasNext跳过空的数组，也就是说，保证了遍历顺序与Enum中key的先后顺序一致。

本文转载自：点击打开链接

参考资料

What is EnumMap in Java