ThreadLocal详解

想要获取更多文章能够访问个人博客 -  代码无止境。

什么是ThreadLocal

ThreadLocal在《Java核心技术 卷一》中被称做线程局部变量（PS：关注公众号itweknow，回复“Java核心技术”获取该书），咱们能够利用ThreadLocal建立只能由同一线程读和写的变量。所以就算两个线程正在执行同一段代码，而且这段代码具备对ThreadLocal变量的引用，这两个线程也没法看到彼此的ThreadLocal变量。

简单使用

1.建立ThreadLocal，只须要new一个ThreadLocal对象便可。

private ThreadLocal<String> myThreadLocal = new ThreadLocal<String>();

2.设置值

myThreadLocal.set("I'm a threadLocal");

3.获取值

myThreadLocal.get();

4.清除，有些状况下咱们在使用完线程局部变量后，须要即时清理，不然会致使程序运行错误。

myThreadLocal.remove();

假如咱们如今要利用AOP打印方法的耗时，这个时候咱们须要在@Before方法中记录方法开始执行的时间，而后在@AfterReturning方法中打印出来耗时时间。咱们写在切面里的方法可能慧在多个线程中同时执行，因此此时咱们须要ThreadLocal来记录开始执行的时间。

1.咱们须要在切面类中定义一个ThreadLocal。

private ThreadLocal<Long> threadLocal = new ThreadLocal();

2.在@Before方法中记录开始时间。

long startTime = System.currentTimeMillis();
threadLocal.set(startTime);

3.在@AfterReturning方法中取出开始时间，并计算耗时。

long startTime = threadLocal.get();
long spendTime = System.currentTimeMillis() - startTime;
threadLocal.remove();
System.out.println("方法执行时间：" + spendTime + "ms");

这里只是借这个场景和你们一块儿熟悉一下ThreadLocal的用法，整个打印方法耗时的实现你能够在Github上找到，若是你想了解AOP能够参考这篇文章《使用 Spring Boot AOP 实现 Web 日志处理和分布式锁》。

原理解析

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

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的get()和set()方法的源码，能够看到ThreadLocal是将值存放在ThreadLocalMap中。其实在每一个线程中都维护着一个threadLocals变量（ThreadLocalMap类型），当使用set()方法的时候其实是将值存在当前线程的threadLocals中的，调用get()方法也是从当前线程中取值的，这样就作到了线程间的隔离。
看到这里想必你也奇怪，在设置值和取值的时候都没有任何与key有关的东西，那么当一个线程有多个ThreadLocal的时候是如何作到一一对应的呢？那咱们就一块儿来看下这个ThreadLocalMap类吧。

static class ThreadLocalMap {
    /**
     * The initial capacity -- MUST be a power of two.
     */
    private static final int INITIAL_CAPACITY = 16;

    /**
     * The table, resized as necessary.
     * table.length MUST always be a power of two.
     */
    private Entry[] table;

    /**
     * The number of entries in the table.
     */
    private int size = 0;

    /**
     * The next size value at which to resize.
     */
    private int threshold; // Default to 0
}

由上面可见在ThreadLocalMap中维护着table，size以及threshold三个属性。table是一个Entry数组主要用来保存具体的数据，size是table的大小，而threshold这表示当table中元素数量超过该值时，table就会扩容。了解了ThreadLocalMap的结构以后，咱们就来看下其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) {
            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();
}

经过上面的代码分析得出，整个的设值过程以下：

1. 经过ThreadLocal的threadLocalHashCode值定位到table中的位置i。
2. 若是table中i这个位置是空的，那么就新建立一个Entry对象放置在i这个位置。
3. 若是table中i这个位置不为空，则取出来i这个位置的key。
4. 若是这个key恰好就是当前ThreadLocal对象，则直接修改该位置上Entry对象的value。
5. 若是这个key不是当前TreadLocal对象，则寻找下一个位置的Entry对象，而后重复上述步骤进行判断。

对于get方法也是一样的原理从ThreadLocalMap中获取值。那么ThreadLocal是如何生成threadLocalHashCode值的呢？

public class ThreadLocal<T> {
    private final int threadLocalHashCode = nextHashCode();
    private static final int HASH_INCREMENT = 0x61c88647;
    private static int nextHashCode() {
        return nextHashCode.getAndAdd(HASH_INCREMENT);
    }
}

可见咱们在初始化一个ThreadLocal对象的时候都为其会生成一个threadLocalHashCode值，每初始化一个ThreadLocal该值就增长0x61c88647。这样就能够作到每一个ThreadLocal在ThreadLocalMap中找到一个存储值的位置了。

结束语

在文章的最后分享一次以前遇到的一个与ThreadLocal有关的坑，有一次在写分页的时候使用了PageHeler插件，引包的时候错误地引用了MybatisPlus下的PagerHelper，而MybatisPlus下的PageHelper在ThreadLocal中存储了SQL分页信息在使用以后没有移除，因此执行了分页的SQL以后在当前线程中执行的SQL都会出现问题。因此你们在使用ThreadLocal的过程当中千万要注意在适当的时候须要清除。本文主要介绍了Java中的线程局部变量ThreadLocal的使用，而且和你们一块儿稍微了解了一下源码。但愿对你们可以有所帮助。

PS:学习不止，码不停蹄！若是您喜欢个人文章，就关注我吧！