ThreadLocal源代码解析

最开始的时候，理解的ThreadLocal，我的理解是这样的：

ThreadLocal<Thread,Map<k,v>> K的话是线程，V的话是线程的副本，是一个Map，里面可以方式多个变量的副本，后来发现是错的。

今天翻了一下代码，画了一个类图。

 

关于ThreadLocalMap.table这个属性，类型是Entry[]

用于真正存放数据的Entry，继承于WeakReference。想想ThreadLocal的使用场景,用“空间换时间”，若空间全部被强应用，则很容易造成内存泄露(使用结束后没有显示调用remove方法)。采用弱引用，在关键时刻，或在调用方没有显示的置空的情况下，也可回收部分内存，以便程序继续运行。其余基本都与HashMap的内部实现类似，内部都通过数组实现。而对键值key，HashMap存的是用户自定义的值，ThreadLocal存的线程对象hashcode值，仅此而已。

 

关于ThreadLocal中的方法

public void set(T value)

设置当前线程副本中的线程局部变量的值。

public T get()

返回此线程局部变量的当前线程副本中的值。

如果这是线程第一次调用该方法，则创建并初始化此副本。

public void remove()

移除此线程局部变量的值。

该方法是JDK 5.0新增的方法。目的是为了减少内存的占用。当线程结束后，对应该线程的局部变量将自动被垃圾回收，所以显式调用该方法清除线程的局部变量并不是必须的操作，但它可以加快内存回收的速度。

protected T initialValue()

返回此线程局部变量的当前线程的初始值。

线程第一次使用get()方法访问变量的时候。如果线程先于get方法调用set(T)方法，则不会在线程中再调用initialValue方法。该方法是一个protected的方法，显然是为了让子类覆盖而设计的。因该缺省实现只返回null；如果程序员希望将线程局部变量初始化为 null 以外的某个值，则必须为 ThreadLocal 创建子类，并重写此方法。

 

关于ThreadLocal属性的get操作

1、获得当前线程对象

2、从当前线程中获得ThreadLocalMap的引用

3、然后通过当前ThreadLocal属性的hashCode找到对应的Entry，Entry是一个数组，真正存放数据的地方，返回结果

 

使用ThreadLocal注意的问题

1、变量污染问题

在使用线程池的时候，要非常主要remove的操作，如果没有，一个线程处理完一个任务，之后处理另外一个任务，变量就会有问题。

2、内存问题

主要放入ThreadLocal中的数据大小，虽然是弱引用，但是也会导致GC问题的。