ThreadLocal的原理及使用场景

ThreadLocal一般被用来存储线程变量的副本，各个线程存储数据都是独立的，互不产生影响。由于这种隔离性，使得线程之间没有资源竞争的关系，因此也不需要做线程同步。

ThreadLocal的原理

ThreadLocal主要提供一下集中方法：

public T get() { }  //取数据
public void set(T value) { }  //存数据
public void remove() { }  //删除数据
protected T initialValue() { } // 初始化的数据，用于子类自定义初始化值

下面我们将从ThreadLocal的源码，分析其实现如何做到变量的线程本地存储。

1. get方法

get方法首先获取调用者的线程，然后通过getMap方法获取ThreadLocalMap，ThreadLocalMap是一个内部类，内部包含一个数据结构Entry，Entry内部是ThreadLocal的引用和存储的值。如果ThreadLocalMap的值是空的，则需要初始化值并返回。如果为空，需要从ThreadLocalMap中取出Entry，如果Entry不为空，将Value返回。

我们还可以看一下getMap(t)这个方法，其仅仅把线程的threadLocals返回，由此可知，实际上用户存储的变量是依附在线程上的，这也是实现线程和本地变量绑定的最关键地方。

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

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

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

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

2. remove

remove方法仅仅把依附的线程上的关于ThreadLocal存储在ThreadLocalMap上的变量清空而已

public void remove() {
         ThreadLocalMap m = getMap(Thread.currentThread());
         if (m != null)
             m.remove(this);
}

4. set()

set方法首先从线程上取出依附的ThreadLocalMap，如果线程上已经存在ThreadLocalMap，则将值和此ThreadLocal绑定，如果不存在，创建一个ThreadLocal，初始化ThreadLocal和Value绑定，so easy~

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

6. initialValue

这是个被保护的方法，等待子类继承重写，如果没有重写此方法，在没有调用过set的情况下调用get，会返回空。

    protected T initialValue() {
        return null;
    }

注：如果想要深入了解，可以参看ThreadLocalMap这个类的设计，他负责数据的存储和删除，应该算是ThreadLocal的最重要的Point。

ThreadLocal的使用场景

ThreadLocal主要用在数据库连接管理，session管理等上面

private static ThreadLocal<Connection> connectionHolder = new ThreadLocal<Connection>() {
public Connection initialValue() {
    return DriverManager.getConnection(DB_URL);
}
};
 
public static Connection getConnection() {
return connectionHolder.get();
}

private static final ThreadLocal threadSession = new ThreadLocal();
 
public static Session getSession() throws InfrastructureException {
    Session s = (Session) threadSession.get();
    try {
        if (s == null) {
            s = getSessionFactory().openSession();
            threadSession.set(s);
        }
    } catch (HibernateException ex) {
        throw new InfrastructureException(ex);
    }
    return s;
}