1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

public V put(K key, V value) {

    return putVal(hash(key), key, value, false, true);

}

final V putVal(int hash, K key, V value, boolean onlyIfAbsent,

                   boolean evict) {

    Node<K,V>[] tab; Node<K,V> p; int n, i;

    // 判断 table[] 是否为空，如果是空的就创建一个 table[]，并获取他的长度n

    if ((tab = table) == null || (n = tab.length) == 0)

        n = (tab = resize()).length;   

    // 如果单链表节点 Node<K,V> p == tab[i = (n - 1) & hash]) == null，

    // 就直接 put 进单链表中，说明此时并没有发生 Hash 冲突

    if ((p = tab[i = (n - 1) & hash]) == null)

        tab[i] = newNode(hash, key, value, null);

    else {

        // 说明索引位置已经放入过数据了，已经在单链表处产生了Hash冲突

        Node<K,V> e; K k;

        // 判断 put 的数据和之前的数据是否重复

        if (p.hash == hash &&

            // 进行 key 的 hash 值和 key 的 equals() 和 == 比较，如果都相等，则初始化数组 Node<K,V> e

            ((k = p.key) == key || (key != null && key.equals(k))))            

            e = p;

        // 判断是否是红黑树，如果是红黑树就直接插入树中

        else if (p instanceof TreeNode)

            e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);

        else {

            // 如果不是红黑树，就遍历每个节点，判断单链表长度是否大于等于 7，

            // 如果单链表长度大于等于 7，数组的长度小于 64 时，会优先选择扩容

            // 如果单链表长度大于等于 7，数组的长度大于 64 时，才会选择单链表--->红黑树

            for (int binCount = 0; ; ++binCount) {

                if ((e = p.next) == null) {

                    // 采用尾插法，在单链表中插入数据

                    p.next = newNode(hash, key, value, null);

                    // 如果 binCount >= 8 - 1

                    if (binCount >= TREEIFY_THRESHOLD - 1)

                        treeifyBin(tab, hash);

                        break;

                }

                // 判断索引每个元素的key是否可要插入的key相同，如果相同就直接覆盖

                if (e.hash == hash &&

                    ((k = e.key) == key || (key != null && key.equals(k))))

                    break;

                 p = e;

            }

        }

        // 说明数组或者单链表中有相同的key，因此只需要将value覆盖，并将oldValue返回即可

        if (e != null) {

            V oldValue = e.value;

            if (!onlyIfAbsent || oldValue == null)

                e.value = value;

                afterNodeAccess(e);

                return oldValue;

        }

    }

    // 说明没有key相同，因此要插入一个key-value，并记录内部结构变化次数

    ++modCount;

    // 判断是否扩容

    if (++size > threshold)

        resize();

    afterNodeInsertion(evict);

    return null;

}

1

2

// HashMap 中 size 变量

transient int size;