ThreadLocal原理分析

初步理解ThreadLocal

ThreadLocal是线程本地变量，是隔离的思想来解决线程安全问题的一种方法
ThreadLocal是HashMap思想在多线程的应用
ThreadLocal是作为Key为ThreadLocalMap获取数据而备

一、ThreadLocal原理总结

大致原理总结：
在每个Thread中包含一个ThreadLocalMap，ThreadLocalMap的key是 ThreadLocal对象的弱引用，value是独享数据。

用ThreadLocal的弱引用作为key

为什么使用弱引用？
我们看看官方文档的说法：
我们来对比一下两种引用
key 使用弱引用：引用的ThreadLocal的对象被回收了，由于ThreadLocalMap持有ThreadLocal的弱引用，即使没有手动删除，ThreadLocal也会被回收。value在下一次ThreadLocalMap调用set、get、remove的时候会被清除。

key 使用强引用：引用的ThreadLocal的对象被回收了，但是ThreadLocalMap还持有ThreadLocal的强引用，如果没有手动删除，ThreadLocal不会被回收，导致Entry内存泄漏。

结论：
ThreadLocal容易出现内存泄漏的根源是：由于ThreadLocalMap的生命周期跟Thread一样长，如果没有手动删除对应key就会导致内存泄漏，而不是因为弱引用。

下面是引用图

二、源码分析

一、Thread类中维护了一个threadLocals的变量，是ThreadLocalMap类型

二、ThreadLocalMap的源码中，定义了Entry类,key是ThreadLocal类型，并且是弱引用的ThreadLocal，值是object

2.1 存储

看一下set方法：
1：获取当前线程
2：根据线程调用getMap，返回ThreadLocalMap
3、如果map!=null，调用map.set(this,value)，this指的是ThreadLocal作为key
4、如果map==null，创建map

这个getMap方法返回的就是上文提到的Thread类中维护的那个threadLocals变量

ThreadLocal的set最终调用了ThreadLocalMap的set方法，如下图
思考：ThreadLocal作为Key是如何避免哈希冲突的呢？
ThreadLocal的set方法中的463行给出了答案，我们重点关注下面这行代码：

 int i = key.threadLocalHashCode & (len-1);

set源码解析
1、根据key的哈希码和数组长度求元素放置的位置，即数组下标
2、 从i开始往后一直遍历到数组最后一个Entry
3、 循环如果key相等，覆盖value
4、 如果key为null,用新key、value覆盖，同时清理历史key=null的陈旧数据
5、如果超过阀值，就需要再哈希了

在ThreadLocal中有这么一个变量，计算threadLocalHashCode的：

nextHashCode又是怎么实现的呢？看下图，每一次生成的key的hash值都不同，避免了哈希冲突

下面是创建map的源码

2.2 取值

1）从当前线程中获取ThreadLocalMap，查询当前ThreadLocal变量实例对应的Entry，如果不为null,获取value,返回
2）如果map为null,即还没有初始化，走初始化方法

2.3 移除

1、获取ThreadLocalMap
2、调用ThreadLocalMap的remove方法删除变量
3、循环判断key相等，调用Entry的clear方法
4、清除陈旧数据

remove方法的具体实现：

private void remove(ThreadLocal<?> key) {
    Entry[] tab = table;
    int len = tab.length;
    int i = key.threadLocalHashCode & (len-1);
    for (Entry e = tab[i];
         e != null;
         e = tab[i = nextIndex(i, len)]) {
        if (e.get() == key) {
            e.clear();//调用Entry的clear方法
            expungeStaleEntry(i);//清除陈旧数据
            return;
        }
    }
}