HashMap詳解七

使用 Iterator 遍歷

通過 HashMap.entrySet().iterator() 方法獲取迭代器, 使用 next 方法對 HashMap 進行遍歷.

HashMap<String, String> map = new HashMap<>();
Iterator it = map.entrySet().iterator();
while(it.hasNext()) {
    Map.Entry<String, String> entry = it.next();
}

下面詳細講解各個方法的作用, 其實迭代器之所以能遍歷元素節點, 主要是應用了內部類. 通過內部類可以訪問外部類的變數和方法, 從而完成遍歷節點.

---

entrySet()

/**
 * 直接返回 EntrySet 的例項
 * 注意這裡 entrySet 不是靜態方法, 而 EntrySet 是非靜態的內部類, 所以可以直接 new 例項
 */
public Set<Map.Entry<K,V>> entrySet() {
    Set<Map.Entry<K,V>> es;
    return (es = entrySet) == null ? (entrySet = new EntrySet()) : es;
}

---

EntrySet

/**
 * EntrySet 繼承於 AbstractSet
 */
final class EntrySet extends AbstractSet<Map.Entry<K,V>> {
    ...

    /**
     * 返回 EntryIterator 例項, 這也是屬於 HashMap 的非靜態內部類
     */
    public final Iterator<Map.Entry<K,V>> iterator() {
        return new EntryIterator();
    }
    ...
}

---

EntryIterator

/**
 * HashMap 的非靜態內部類
 */
final class EntryIterator extends HashIterator
        implements Iterator<Map.Entry<K,V>> {
    /**
     * next 方法呼叫父類 HashIterator 的 nextNode 方法, 返回下一個元素
     */
    public final Map.Entry<K,V> next() { return nextNode(); }
}

---

HashIterator

/**
 * HashMap 的內部抽象類
 */
abstract class HashIterator {
    Node<K,V> next;        // next entry to return
    Node<K,V> current;     // current entry
    int expectedModCount;  // for fast-fail
    int index;             // current slot

    /**
     * 建構函式, 從 0 開始遍歷 HashMap 的儲存陣列, 一直到非空元素
     */
    HashIterator() {
        expectedModCount = modCount;
        Node<K,V>[] t = table;
        current = next = null;
        index = 0;
        if (t != null && size > 0) { // advance to first entry
            do {} while (index < t.length && (next = t[index++]) == null);
        }
    }

    public final boolean hasNext() {
        return next != null;
    }

    final Node<K,V> nextNode() {
        Node<K,V>[] t;
        Node<K,V> e = next;
        if (modCount != expectedModCount)
            throw new ConcurrentModificationException();
        if (e == null)
            throw new NoSuchElementException();

        // 從根節點開始遍歷連結串列, 其中樹也當成連結串列結構來遍歷, 一直到尾節點
        if ((next = (current = e).next) == null && (t = table) != null) {
            // 連結串列遍歷完全後, 重新讀取陣列的下一個非空元素
            do {} while (index < t.length && (next = t[index++]) == null);
        }
        return e;
    }
}

以上就是 HashMap 的遍歷方法, 它不是按照插入節點的先後順序進行遍歷, 而是按照陣列結構來遍歷.

---