Java从入门到放弃（十二）集合框架之HashMap源码(3)

ゞ浴缸里的玫瑰 2022-05-25 01:42 147阅读 0赞

### 1、romove方法 ###

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) {   //如果table对应的位置有值
                Node<K,V> node = null, e; K k; V v;     //node节点就是要替换的节点 
                if (p.hash == hash &&((k = p.key) == key || (key != null && key.equals(k)))) //如果所在位置第一个节点的key与删除的key相等
                    node = p;             //直接替换对应的节点
                else if ((e = p.next) != null) {      //节点存在下一个节点
                    if (p instanceof TreeNode)     //属于红黑树结构
                        node = ((TreeNode<K,V>)p).getTreeNode(hash, key);
                    else {
                        do {
                            if (e.hash == hash &&((k = e.key) == key ||(key != null && key.equals(k)))) {    //key值，hash相等
                                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)   
                        ((TreeNode<K,V>)node).removeTreeNode(this, tab, movable);    //删除红黑树对应节点
                    else if (node == p)
                        tab[index] = node.next;         //在节点上值只有一个节点，即第6行的情况
                    else
                        p.next = node.next;            //直接把删除节点的前一个节点的next指向删除节点的nwxt节点，就相当于把删除节点排除在链表之外了
                    ++modCount;
                    --size;
                    afterNodeRemoval(node);
                    return node;            //返回node节点
                }
            }
            return null;
        }

romoveNode方法是先通过对应的key找到node节点，然后再根据参数value和matchValue的值，这两个值是在精确删除key-value键值对，如果matchValue是true，并且value不为空，就会匹配key和value的值，都相等的时才会删除对应的节点。

### 2、replace方法 ###

public V replace(K key, V value) {
            Node<K,V> e;
            if ((e = getNode(hash(key), key)) != null) {           //获取对应的节点，不为null就替换value值，返回旧的value值
                V oldValue = e.value;   
                e.value = value;
                afterNodeAccess(e);
                return oldValue;
            }
            return null;
        }

###  ###

final Node<K,V> getNode(int hash, Object key) {
            Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
            if ((tab = table) != null && (n = tab.length) > 0 &&(first = tab[(n - 1) & hash]) != null) {  //数组索引位置不为null
                if (first.hash == hash && ((k = first.key) == key || (key != null && key.equals(k))))     
                    return first;                     //如果第一个节点的key就和对应的key相等，直接返回
                if ((e = first.next) != null) {        
                    if (first instanceof TreeNode)
                        return ((TreeNode<K,V>)first).getTreeNode(hash, key);
                    do {
                        if (e.hash == hash &&((k = e.key) == key || (key != null && key.equals(k))))
                            return e;
                    } while ((e = e.next) != null);         //遍历链表或者红黑树的节点寻找对应的节点值
                }
            }
            return null;
        }

这个方法就是替换对应key的value值，还是一样，就是先根据key确定在数组中的位置，然后再数组的链表或者红黑树遍历查找对应的值，对value进行更改。

### 3、clear方法 ###

public void clear() {
            Node<K,V>[] tab;
            modCount++;
            if ((tab = table) != null && size > 0) {   
                size = 0;
                for (int i = 0; i < tab.length; ++i)   //遍历数组置为null
                    tab[i] = null;
            }
        }

### 4、循环迭代 ###

HashMap没有和ArrayList一样的iterator()方法，只能通过获取带键值对的set集合，然后再获取迭代器。

public Set<Map.Entry<K,V>> entrySet() {
            Set<Map.Entry<K,V>> es;
            return (es = entrySet) == null ? (entrySet = new EntrySet()) : es;   //返回一个EntrySet内部类对象
        }
    
        final class EntrySet extends AbstractSet<Map.Entry<K,V>> {
            public final int size()                 { return size; }     //集合的长度
            public final void clear()               { HashMap.this.clear(); }     //清空集合
            public final Iterator<Map.Entry<K,V>> iterator() {           //返回一个迭代器对象
                return new EntryIterator();
            }
            public final boolean contains(Object o) {           //判断是否包含对应的对象
                if (!(o instanceof Map.Entry))
                    return false;
                Map.Entry<?,?> e = (Map.Entry<?,?>) o;
                Object key = e.getKey();
                Node<K,V> candidate = getNode(hash(key), key);
                return candidate != null && candidate.equals(e);
            }
            public final boolean remove(Object o) {      //移除o对象，
                if (o instanceof Map.Entry) {
                    Map.Entry<?,?> e = (Map.Entry<?,?>) o;
                    Object key = e.getKey();
                    Object value = e.getValue();
                    return removeNode(hash(key), key, value, true, true) != null;
                }
                return false;
            }
            public final Spliterator<Map.Entry<K,V>> spliterator() {
                return new EntrySpliterator<>(HashMap.this, 0, -1, 0, 0);
            }
            public final void forEach(Consumer<? super Map.Entry<K,V>> action) {   //函数式编程的一个方法
                Node<K,V>[] tab;
                if (action == null)
                    throw new NullPointerException();
                if (size > 0 && (tab = table) != null) {
                    int mc = modCount;
                    for (Node<K, V> e : tab) {
                        for (; e != null; e = e.next)
                            action.accept(e);
                    }
                    if (modCount != mc)
                        throw new ConcurrentModificationException();
                }
            }
        }

这里通过entrySet方法返回一个内部类对象，这个entrySet的一个set集合，可以生产一个迭代器对象，如下

HashMap<String,String> map  = new HashMap();
            Set<Map.Entry<String, String>> entrySet = map.entrySet();
            Iterator<Map.Entry<String, String>> iterator = entrySet.iterator();

然后就可以循环的遍历map的键值对。也可以只获取key的集合或者value的集合，也一样，获取key的集合使用keySet()方法，返回一个keySet的内部类的对象。获取values就使用values（）方法，返回一个values的内部类对象。如下：

HashMap<String,String> map  = new HashMap();
            Set<String> keySet = map.keySet();
            Iterator<String> keyIterator = keySet.iterator();
    
            Collection<String> values = map.values();
            Iterator<String> valueIterator = values.iterator();