1. 程式人生 > >JDK原始碼-HashMap-remove方法(JDK7和JDK8)

JDK原始碼-HashMap-remove方法(JDK7和JDK8)

remove方法原始碼相應簡單很多

測試程式碼

/**
 * 測試remove操作的區別
 */
@Test
public void remove(){
    HashMap<String, String> map = new HashMap();
    map.put("aa","aa");
    map.remove("aa");
}

JDK 7

public V remove(Object key) {
	Entry<K,V> e = removeEntryForKey(key);
	return (e == null ?
null : e.value); }

進入 removeEntryForKey 方法

final Entry<K,V> removeEntryForKey(Object key) {
	int hash = (key == null) ? 0 : hash(key);
	int i = indexFor(hash, table.length);
	Entry<K,V> prev = table[i]; //當前判斷節點的上一個節點
	Entry<K,V> e = prev; //當前判斷節點

	while (e != null) {
		Entry<
K,V>
next = e.next; //當前判斷節點的下一個節點 Object k; if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) { //找到key對應的節點 modCount++; size--; if (prev == e) table[i] = next; else prev.next = next; e.recordRemoval(this);//HashMap對該方法不執行任何程式碼,LinkedHashMap可以利用該方法維護linked的指標
return e; } prev = e; e = next; } return e; }

JDK 8

public V remove(Object key) {
	Node<K,V> e;
	return (e = removeNode(hash(key), key, null, false, true)) == null ?
		null : e.value;
}

進入 removeNode 方法

final Node<K,V> removeNode(int hash, Object key, Object value,
						   boolean matchValue, boolean movable) {
	Node<K,V>[] tab; Node<K,V> p; int n, index;
	if ((tab = table) != null && (n = tab.length) > 0 &&
		(p = tab[index = (n - 1) & hash]) != null) {
		//node是需要刪除key的對應的節點
		Node<K,V> node = null, e; K k; V v;
		//是否table[index]對應的node符合remove
		if (p.hash == hash &&
			((k = p.key) == key || (key != null && key.equals(k))))
			node = p;
		//否則開始往下個節點查詢
		else if ((e = p.next) != null) {
			//如果是紅黑樹,則通過紅黑樹的方法得到node
			if (p instanceof TreeNode)
				node = ((TreeNode<K,V>)p).getTreeNode(hash, key);
			//連結串列查詢刪除key對應的node
			else {
				do {
					if (e.hash == hash &&
						((k = e.key) == key ||
						 (key != null && key.equals(k)))) {
						node = e;
						break;
					}
					p = e;
				} while ((e = e.next) != null);
			}
		}
		if (node != null && (!matchValue || (v = node.value) == value ||
							 (value != null && value.equals(v)))) {
			if (node instanceof TreeNode) //是否是紅黑樹的node
				((TreeNode<K,V>)node).removeTreeNode(this, tab, movable);
			else if (node == p) //node是否是table[i]的node
				tab[index] = node.next;
			else //連結串列的node
				p.next = node.next; //此時的p是node的父節點
			++modCount;
			--size;
			afterNodeRemoval(node);
			return node;
		}
	}
	return null;
}

總結

JDK 7的刪除方法中由於維護了三個指標,分別是當前節點,當前節點的上一個節點,當前節點的下一個節點。所以刪除過程修改引用即可。

JDK8相對於JDK7,因為加入了紅黑樹的結構,如果刪除時跟JDK1.7稍微不同一點,8中主要通過獲取需要刪除的node,然後再對node進行判斷型別後再相應刪除。