按照上面的分析,此时 slotToExpunge
值为 3 , staleSlot
值为 5 , i 为 6
假设,假设这个时候如果不进行交换,而是直接回收的话,此时位置为 5 的数据就被回收掉,然后接下来要插入一个 key 为 15 的数据,此时 15 mod 10 算出来是 5 ,正好这个时候位置为 5 的被回收完毕,这个位置就被空出来了,那么此时就会这样:
同样的 key 值竟然出现了两次?!
这肯定是不希望看到的结果,所以一定要进行数据交换
在上面代码中有一行代码 cleanSomeSlots(expungeStaleEntry(slotToExpunge), len);
,说明接下来的处理是交给了 expungeStaleEntry
,接下来去分析一下 expungeStaleEntry
expungeStaleEntry
private int expungeStaleEntry(int staleSlot) { Entry[] tab = table; int len = tab.length; // expunge entry at staleSlot tab[staleSlot].value = null; tab[staleSlot] = null; size--; // Rehash until we encounter null Entry e; int i; for (i = nextIndex(staleSlot, len); (e = tab[i]) != null; i = nextIndex(i, len)) { ThreadLocal<?> k = e.get(); // 如果 k == null ,说明 value 就应该被回收掉 if (k == null) { // 此时直接将 e.value 置为 null // 这样就将 thread -> threadLocalMap -> value 这条引用链给打破 // 方便了 GC e.value = null; tab[i] = null; size--; } else { // 这个时候要重新 hash ,因为采用的是开放地址法,所以可以理解为就是将后面的元素向前移动 int h = k.threadLocalHashCode & (len - 1); if (h != i) { tab[i] = null; // Unlike Knuth 6.4 Algorithm R, we must scan until // null because multiple entries could have been stale. while (tab[h] != null) h = nextIndex(h, len); tab[h] = e; } } } return i; }
因为是在 replaceStaleEntry
方法中调用的此方法,传进来的值是 staleSlot
,继续上图,经过 replaceStaleEntry
之后,它的数据结构是这样:
此时传进来的 staleSlot
值为 6 ,因为此时的 key 为 null ,所以接下来会走 e.value = null
,这一步结束之后,就成了:
接下来 i 为 7 ,此时的 key 不为 null ,那么就会重新 hash : int h = k.threadLocalHashCode & (len - 1);
,得到的 h 应该是 5 ,但是实际上 i 为 7 ,说明出现了 hash 冲突,就会继续向下走,最终的结果是这样:
可以看到,原来的 key 为 null ,值为 V5 的已经被回收掉了。我认为之所以回收掉之后,还要再次进行重新 hash ,就是为了防止 key 值重复插入情况的发生
假设 key 为 25 的并没有进行向前移动,也就是它还在位置 7 ,位置 6 是空的,再插入一个 key 为 25 ,经过 hash 应该在位置 5 ,但是有数据了,那就向下走,到了位置 6 ,诶,竟然是空的,赶紧插进去,这不就又造成了上面说到的问题,同样的一个 key 竟然出现了两次?!
而且经过 expungeStaleEntry
之后,将 key 为 null 的值,也设置为了 null ,这样就方便 GC
分析到这里应该就比较明确了,在 expungeStaleEntry
中,有些地方是帮助 GC 的,而通过源码能够发现, set 方法调用了该方法进行了 GC 处理, get 方法也有,不信你瞅瞅:
get
private Entry getEntry(ThreadLocal<?> key) { int i = key.threadLocalHashCode & (table.length - 1); Entry e = table[i]; // 如果能够找到寻找的值,直接 return 即可 if (e != null && e.get() == key) return e; else // 如果找不到,则调用 getEntryAfterMiss 方法去处理 return getEntryAfterMiss(key, i, e); } private Entry getEntryAfterMiss(ThreadLocal<?> key, int i, Entry e) { Entry[] tab = table; int len = tab.length; // 一直探测寻找下一个元素,直到找到的元素是要找的 while (e != null) { ThreadLocal<?> k = e.get(); if (k == key) return e; if (k == null) // 如果 k == null 说明有 value 没有及时回收 // 调用 expungeStaleEntry 方法去处理,帮助 GC expungeStaleEntry(i); else i = nextIndex(i, len); e = tab[i]; } return null; }
get 和 set 方法都有进行帮助 GC ,所以正常情况下是不会有内存溢出的,但是如果创建了之后一直没有调用 get 或者 set 方法,还是有可能会内存溢出
所以最保险的方法就是,使用完之后就及时 remove 一下,加快垃圾回收,就完美的避免了垃圾回收
我 ThreadLocal 虽然没办法做到 100% 的解决内存泄漏问题,但是我能做到 80% 不也应该夸夸我嘛