前言
ThreadLocal是什么？有什么作用？我们直接说结论。
- ThreadLocal跟线程同步机制没有半毛钱关系。
- ThreadLocal提供了解决多线程环境下成员变量问题的解决方案，但是并不是用共享变量的方式。
例子1
public class Main {
public static void main(String[] args) {
// write your code here
Count count = new Count();
for (int i = 0 ; i < 3 ; i++){
new CountThread(count).start();
}
}
}
class Count{
private static ThreadLocal<Integer> count = new ThreadLocal<Integer>(){
@Override
protected Integer initialValue() {
return 0;
}
};
public void addCount(){
count.set(count.get()+1);
}
public void printCount(){
System.out.println(Thread.currentThread().getName()+"-"+count.get());
}
}
class CountThread extends Thread{
private Count count;
public CountThread(Count count){
this.count = count;
}
@Override
public void run() {
for (int i = 0 ;i < 5 ; i++){
count.addCount();
count.printCount();
}
}
}
这个例子中，Count类是计数器类，CountThread是执行计数的线程，用来模拟多线程情况下的计数效果。下面就是输出结果。
Thread-0-1
Thread-0-2
Thread-0-3
Thread-0-4
Thread-0-5
Thread-1-1
Thread-1-2
Thread-1-3
Thread-2-1
Thread-2-2
Thread-2-3
Thread-2-4
Thread-2-5
Thread-1-4
Thread-1-5
很明显，运行结果是符合我们预期的效果。从结果上看，线程间对变量的访问操作做到了隔离，每个计数线程都启到了计数的功能，并不相互影响。那么如果把整型换成引用类型呢？工作线程访问变量也能起到隔离的作用吗？那我们就再写个例子验证下。
例子2
public static void main(String[] args) {
Ref ref = new Ref();
for (int i = 0 ; i < 2 ; i++){
new RefThread(ref).start();
}
}
class A{}
class Ref{
private static A a = new A();
private static ThreadLocal<A> ref = new ThreadLocal<A>(){
@Override
protected A initialValue() {
return a;
}
};
public void change(){
ref.set(new A());
}
public void printAddress(){
System.out.println(Thread.currentThread().getName()+"-"+ref.get());
}
}
class RefThread extends Thread{
private Ref ref;
public RefThread(Ref ref){
this.ref = ref;
}
@Override
public void run() {
for (int i = 0 ; i < 2 ; i++) {
ref.printAddress();
ref.change();
ref.printAddress();
}
}
}
例子2和例子1大同小异，区别在于变量类型变成了引用类型，通过打印内存地址来判断在线程内变量使用的连续性，和多线程环境下的变量隔离性。我们看下输出结果。
Thread-0-com.loubinfeng.A@c1e719b
Thread-0-com.loubinfeng.A@2d99e68
Thread-0-com.loubinfeng.A@2d99e68
Thread-0-com.loubinfeng.A@6e61aeb
Thread-1-com.loubinfeng.A@c1e719b
Thread-1-com.loubinfeng.A@189e7143
Thread-1-com.loubinfeng.A@189e7143
Thread-1-com.loubinfeng.A@16598c20
从结果上看很明显，同个线程变量初始是指向统一内存地址，后续变化是连续的，多线程环境下变量也是隔离的。也就是不同是引用类型还是基础数据类型，使用ThreadLocal都能解决多线程环境的变量问题。
例子3
从例子2中，我们发现两个线程变量副本初始都指向同一内存地址，所以我们改例子2，如下：
class A{
int p;
public A add(){
p++;
return this;
}
}
class Ref{
private static A a = new A();
private static ThreadLocal<A> ref = new ThreadLocal<A>(){
@Override
protected A initialValue() {
return a;
}
};
public void change(){
ref.set(ref.get().add());
}
public void printAddress(){
System.out.println(Thread.currentThread().getName()+"-"+ref.get()+"-"+ref.get().p);
}
}
class RefThread extends Thread{
private Ref ref;
public RefThread(Ref ref){
this.ref = ref;
}
@Override
public void run() {
for (int i = 0 ; i < 2 ; i++) {
ref.printAddress();
ref.change();
ref.printAddress();
}
}
}
我们在A类中添加了一个int类型的p变量，那么我们多线程操作这个变量，能做到隔离吗？我们看下结果：
Thread-0-com.loubinfeng.A@6d8b82ec-0
Thread-1-com.loubinfeng.A@6d8b82ec-0
Thread-0-com.loubinfeng.A@6d8b82ec-1
Thread-1-com.loubinfeng.A@6d8b82ec-2
Thread-0-com.loubinfeng.A@6d8b82ec-2
Thread-1-com.loubinfeng.A@6d8b82ec-2
Thread-0-com.loubinfeng.A@6d8b82ec-3
Thread-1-com.loubinfeng.A@6d8b82ec-4
很明显，结果差强人意，因为操作的是同一内存，如果要达到p变量的隔离效果，只需要微调下，如下：
class Ref{
private static A a = new A();
private static ThreadLocal<A> ref = new ThreadLocal<A>(){
@Override
protected A initialValue() {
return a;
}
};
public void change(){
A a = new A();
a.p = ref.get().p+1;
ref.set(a);
}
public void printAddress(){
System.out.println(Thread.currentThread().getName()+"-"+ref.get()+"-"+ref.get().p);
}
}
Thread-0-com.loubinfeng.A@14e672aa-0
Thread-0-com.loubinfeng.A@1a218e46-1
Thread-0-com.loubinfeng.A@1a218e46-1
Thread-0-com.loubinfeng.A@1e1e19da-2
Thread-1-com.loubinfeng.A@14e672aa-0
Thread-1-com.loubinfeng.A@45669fee-1
Thread-1-com.loubinfeng.A@45669fee-1
Thread-1-com.loubinfeng.A@4ba17932-2
这个例子想说明，使用ThreadLocal，变量又是引用类型时，请注意变量的内存地址，因为变量副本复制是引用，而不是真正的内存。
ThreadLocal工作原理及api
看到这里，大家一定对ThreadLocal的工作原理很好奇。它是怎么做到变量的隔离而没有涉及变量的共享同步。
因为线程同步机制是多线程共享同一个变量，而ThreadLocal是为每一个线程创建一个单独的变量副本，所以每个线程都可以独立的改变自己所拥有的变量副本，同时还不影响其他线程的对应副本。
ThreadLocal定义了4个方法供开发者调用：
- get()：返回此线程局部变量的当前线程副本中的值。
- initialValue()：返回此线程局部变量的当前线程的“初始值”。
- remove()：移除此线程局部变量当前线程的值。
- set(T value)：将此线程局部变量的当前线程副本中的值设置为指定值。
ThreadLocal，ThreadLocalMap和Thread之间关系
再探究ThreadLocal工作机制源码之前，我们先要搞清ThreadLocal，ThreadLocalMap和Thread三者的角色关系。
- ThreadLocalMap是ThreadLocal的一个内部类
- 每一个Thread对象中都有一个ThreadLocalMap类型的变量，用来存储变量副本
- ThreadLocalMap的key是ThreadLocal类型的，value是变量类型。
补充下，ThreadLocalMap是实现变量副本机制的关键类。突然出现这个类，估计一头雾水，下节我们讲探究源码，大家会对这个类有进一步的了解。
ThreadLocal源码实现
public void set(T value) {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
}
这是ThreadLocal的set方法的源码实现，实现逻辑是非常清晰的，先获取到了当前的线程对象，然后得到线程内部的ThreadLocalMap对象，如果是空的话，就创建这个对象。不是空的话，就将值存入这个map中，key则为当前这个ThreadLocal对象。
private void set(ThreadLocal<?> key, Object value) {
// We don't use a fast path as with get() because it is at
// least as common to use set() to create new entries as
// it is to replace existing ones, in which case, a fast
// path would fail more often than not.
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)]) {
ThreadLocal<?> k = e.get();
if (k == key) {
e.value = value;
return;
}
if (k == null) {
replaceStaleEntry(key, value, i);
return;
}
}
tab[i] = new Entry(key, value);
int sz = ++size;
if (!cleanSomeSlots(i, sz) && sz >= threshold)
rehash();
}
这个是ThreadLocalMap的set方法，跟随上面的思路，ThreadLocal的set方法最终执行了ThreadLocalMap的set方法。这里要说明的是，虽然都是key-value结构，但是和集合Map解决散列冲突的方式是不一样的。集合Map的put采用是拉链式的，而ThreadLocalMap的set采用的是开放定址法。
public T get() {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null) {
ThreadLocalMap.Entry e = map.getEntry(this);
if (e != null) {
@SuppressWarnings("unchecked")
T result = (T)e.value;
return result;
}
}
return setInitialValue();
}
这是ThreadLocal的get方法，跟他的set方法如出一辙，也是先得到线程对象，在获取Thread中的ThreadLocalMap对象，然后根据key取值，如果map为空，直接返回ThreadLocal 的初始值。
/**
* Get the entry associated with key. This method
* itself handles only the fast path: a direct hit of existing
* key. It otherwise relays to getEntryAfterMiss. This is
* designed to maximize performance for direct hits, in part
* by making this method readily inlinable.
*
* @param key the thread local object
* @return the entry associated with key, or null if no such
*/
private Entry getEntry(ThreadLocal<?> key) {
int i = key.threadLocalHashCode & (table.length - 1);
Entry e = table[i];
if (e != null && e.get() == key)
return e;
else
return getEntryAfterMiss(key, i, e);
}
/**
* Version of getEntry method for use when key is not found in
* its direct hash slot.
*
* @param key the thread local object
* @param i the table index for key's hash code
* @param e the entry at table[i]
* @return the entry associated with key, or null if no such
*/
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)
expungeStaleEntry(i);
else
i = nextIndex(i, len);
e = tab[i];
}
return null;
}
当然追踪get方法，最终还是执行到了ThreadLocalMap的getEntry方法。如果所对应的key就是我们要找的元素，则返回，否则就调用getEntryAfterMiss方法。
这部分源码就不过分展开了，大略介绍下。
ThreadLocal方法的缺陷
ThreadLocal会存在内存泄漏的情况。因为每个Thread都有一个ThreadLocalMap实例，这个map的key类型是ThreadLocal，但是是个弱引用。弱饮用有利于GC回收。当这个key为null时，GC就会回收这部分空间，但是value却不一定被回收，因为有可能他和当前线程还存在强引用关系。这就是问题所在。
当然源码中也考虑到了这种情况，已经做了很多优化的地方，但是还是不能100%避免。这是我们就要显示的调用ThreadLocal的remove方法进行处理。
总结
ThreadLocal是解决多线程变量问题的另一种思路，核心理念就是为每个线程创建变量副本，用空间换时间的思路。
