JDK原始碼-HashMap-remove方法(JDK7和JDK8)
阿新 • • 發佈:2018-11-09
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進行判斷型別後再相應刪除。