线程内部存储---TheadLocal从使用到源码分析

ThreadLocal是什么？

ThreadLocalThreadLocal是一个线程内部用于存储数据的类，通过它可以在指定的线程中存储数据，数据存储以后，只有在该线程中可以获取到存储的数据，对于其它线程来说无法获取到数据。个人认为是一个线程内部的存储机制。

如何使用？

ThreadLocal<Boolean> threadLocal = new ThreadLocal<>(); 这样就在一个线程创建了ThreadLocal这个对象，这个对象支持范性的，也就是说我们可以存储的自己想存的任意类型。

threadLocal.set(true); 这样我们就在当前所在线程，存储了true。

new Thread("Thread#1") { @Override public void run() { System.out.println(threadLocal.get()); }; }.start(); 我们在一个分线程去打印这个刚才存储的值会发现是null，因为这是在两个线程操作的。如果在当前线程获取则为正确刚才存的值。 如果想移除这个数据也很简单： threadLocal.remove(); 这就是threadLocal的使用，其实很简单。 主要记得是区分线程的就ok。

内部源码是如何实现的呢？

这就是treadLocal中的所有方法了，其实我们最关心的就是set和get方法。

我们先看set:

public void set(T value) { Thread t = Thread.currentThread(); ThreadLocalMap map = getMap(t); if (map != null) map.set(this, value); else createMap(t, value); } 这就是set方法，对就这么几行。 首先会通过Thread.currentThread();这个方法获取当前线程的Thread。然后通过getMap()获取 ThreadLocalMap对象。 ThreadLocalMap getMap(Thread t) { return t.threadLocals; } getMap中也很简单，就是返回这个线程的threadLocals, 这个ThreadLocal就是 Thread中的一个变量ThreadLocal.ThreadLocalMap threadLocals = null; 其实还是ThreadLocal中的内部类ThreadLocalMap；这里一会在分析，先顺着思路往下走。 获取map之后会根据判断如果不是null就进行set,也就是存储，如果是null就会调用createMap()方法进行创建这个ThreadMap。接下来在看是如何set和createMap的。

创建其实很简单,直接就是new一个。 void createMap(Thread t, T firstValue) { t.threadLocals = new ThreadLocalMap(this, firstValue); } set方法就稍微复杂了些，因为这也是核心内容。set方法在内部类ThreadLocalMap中，所以接下来分析下这个ThreadLocalMap类。

ThreadLocalMap

这个类在构造中创建了一个数组， new Entry[INITIAL_CAPACITY]; ,Entry里面就是一个object的对象,然后里面主要getEntry和set方法进行存取和读取。

““ private void set(ThreadLocal key, Object value) {

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) {//这里判断这个k和传进来的key是否相等 e.value = value;//进行存储后return 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(); }

““ 看主要代码 判断k和传进来的key相等话就存储这个value get方法：

“` private Entry getEntry(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) return (T)e.value; } return setInitialValue(); } get里面也是要先获取当前的Thread这也就是为什么ThreadLocal获取和存储的都是只在当前线程的。

然后也是getMap方法获取这个ThreadLocalMap，这也就是为什么里面就一行代码，也要写成一个方来，因为这是中思路，其他地方获取直接调用就行，日后扩展的话一样方便。

再然后就是通过这个map去调用上面说的getEntry方法。

至此，我们就知道ThreadLocal的总体工作流程和思路了。