JDK源码-HashMap-remove方法（JDK7和JDK8）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Lin_wj1995/article/details/83895429

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进行判断类型后再相应删除。