ThreadLocal概念和源码讲解

ThreadLocal概念

ThreadLocal<T> 提供了线程的局部变量，每个线程都可以通过set(),get() 来对这个局部变量进行操作，但不会和其他线程的局部变量冲突，实现了线程的数据隔离。也就是说线程操作了这个ThreadLocal<T>变量那么它就独自拥有这个操作的值，其他线程访问不了。ThreadLocal<T>的泛型变量可以封装各种类型，是线程封装的线程变量。

ThreadLocal内部封装了ThreadLocalMap内部类，而Thread类内部聚合了一个ThreadLocalMap：ThreadLocalMap内部封装了一个Entry类用于存储数据（如下图），可以看出Entry继承了一个虚引用，Entry的存储结构是键值对，而这个键就是一个ThreadLocal对象，值就是我们需要存储的线程私有变量。

static class ThreadLocalMap {

       
        static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;

            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }

 

从ThreadLocal获取值

public T get() {
        Thread t = Thread.currentThread();  //获取当前线程
        ThreadLocalMap map = getMap(t);  //获取线程内聚合的ThreadLocalMap
        if (map != null) {
            ThreadLocalMap.Entry e = map.getEntry(this);  //通过ThreadLocal对象获取Entry
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;  //获取值
                return result;
            }
        }
        return setInitialValue();  //若map为null，初始化
    }

可以看到，首先通过Thread.currentThread()获取当前线程，然后再从线程里获取一个ThreadLocalMap，每个线程都会维护一个ThreadLocalMap的引用；然后再从该map里获取值，key是一个ThreadLocal对象。

往ThreadLoacal中set值

public void set(T value) {
        Thread t = Thread.currentThread(); //获取当前线程
        ThreadLocalMap map = getMap(t);  //获取当前线程关联的map
        if (map != null)
            map.set(this, value);  //往map插入值，key是一个ThreadLocal对象
        else
            createMap(t, value);  //若map为null，新建一个当前对象的ThreadLocalMap并存入value
    }

 

在spring中应用场景

在Spring的获取数据库连接池的连接用到，即当线程需要操作mapper时，mapper封装了sqlsession, sqlsession需要获取连接collection，基于连接获取statement才能对数据库进行操作；线程获取连接的时候不是直接去连接池随机获取一个连接，这样会导致spring事务乱套，因为事务必须基于同一个连接，如果两个操作组成一个事务同时去连接池获取两个不同的连接，那么两个操作就是相互隔离的无法组成事务。

所以获取连接是通过一个事务管理器，事务管理器去连接池取一个，那么问题来了，当不同的线程事务取连接可能取到相同的连接，这样又会导致事务乱套，这时候就需要用到ThreadLocal<Collection>了，每一个线程来取的时候就将这个连接通过ThreadLocal内置到自己线程中私有了，而其他线程是无法得到的，当其他线程来的时候发现自己的ThreadLocal<Collection>为null则又会重新从连接池获取一个连接给自己私有，spring通过这样控制事务的一致性。如下图：