一步一步学多线程-ThreadLocal源码解析

　　上网查看了很多篇ThreadLocal的原理的博客，上来都是文字一大堆，费劲看了半天，大脑中也没有一个模型，想着要是能够有一张图明确表示出来ThreadLocal的设计该多好，所以就自己看了源码，画了一张图出来，立刻清晰不少。

根据源码代码理清ThreadLocal结构

 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();
 }

 ThreadLocalMap getMap(Thread t) {
     return t.threadLocals;
 }

 ThreadLocal.ThreadLocalMap threadLocals = null;

 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);
 }

　　通过上面的代码可以看出来，在每个线程里面都保存了ThreadLocalMap，而ThreaLocalMap里面又存储了以ThreadLocal为键值的Entry数组。

　　用一张图来表示更清楚

　　　　　　

ThreatLocal的Set步骤

　　先看ThreadLocal的set方法，我们看一下ThreatLocal具体是怎么操作的

 public void set(T value) {
     Thread t = Thread.currentThread();
     ThreadLocalMap map = getMap(t);
     if (map != null)
         map.set(this, value);
     else
         createMap(t, value);
 }

 void createMap(Thread t, T firstValue) {
     t.threadLocals = new ThreadLocalMap(this, firstValue);
 }

　　通过上面的代码可以看出来ThreadLocal的存储

　　1、  获取当前线程

　　2、  获取到当前线程中的ThreadLocalMap

　　3、  如果获取到的ThreadLocalMap不为null，就以当前ThreadLocal对象为键，以要插入的值为值，ThreadMapLocalMap中。

　　4、  如果获取的的ThreadLocalMap为null，则创建一个ThreadLocalMap对象

ThreadLocal的Get步骤

　　

 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();
 }

 private T setInitialValue() {
     T value = initialValue();
     Thread t = Thread.currentThread();
     ThreadLocalMap map = getMap(t);
     if (map != null)
         map.set(this, value);
     else
         createMap(t, value);
     return value;
 }

　　通过上面的代码可以看出来ThreadLocal的Get步骤为

　　1、  获取当前线程

　　2、  根据当前线程获取线程中的ThreadLocalMap

　　3、  如果获取到的ThreadLocalMap不为null,则以当前ThreadLocal为键，获取存储在ThreadLocalMap中的值并返回

　　4、  如果获取到的ThreadLocalMap为null，则创建一个以当前对象为键的对象，以ThreadLocal设置的初始值为值，存入ThreadLocalMap中，然后返回。

ThreaLocal中的初始值设置

 protected T initialValue() {
     return null;
 }

　　上面这个initalValue方法是在创建ThreadLocalMap时候的value值的默认初始值，代码中默认返回的是null，我们可以通过自己在代码中自己设计值。