ThreadLocal类介绍

ThreadLocal类维持了线程的封闭性,避免了创建线程之间共享的对象变量或者全局变量,保证了多线程环境下对象变量的使用安全性

ThrealLocal类提供了get和set等访问方法,这些方法给每个线程单独创建了该线程使用的变量的独立副本,因此使用get方法获取到总是当前线程在调用set方法时设置的变量值

以下以数据库的JDBC连接对象的使用来举例:

使用单个的JDBC连接对象在多线程环境下操作数据库是线程不安全的,由于数据库事务的存在, 当A用户在更改自己的个人信息,更新成功但没有进行事务提交,此时B用户抢占了A用户操作数据库的JDBC连接对象的执行权,B用户也修改了自己的个人信息,但由于异常情况,B用户修改数据失败而做了事务回滚,这样A用户修改的数据也就跟着一起回滚了, 所以单个数据库连接在多线程环境下肯定是不安全的

我们实际的解决方案是采用数据库连接池来,我们创建多个数据库连接对象,避免了多个线程操作同一数据库的JDBC连接对象,一个线程使用JDBC连接对象操作数据库,必须先向数据库连接池申请使用,并在使用结束后归还线程池才能被下一个线程使用.这样可以通过数据库连接池达到线程安全

我们下面使用ThreadLocal类来实现数据库连接对象的安全性,其本质就是为每个线程单独创建一个本线程可使用的JDBC连接对象,避免多个线程之间共享同一JDBC连接对象:

public class ConnectionHolderFactory {
    // 为本工厂类创建一个ThreadLocal的成员变量
    private static ThreadLocal<Connection> connectionHolder = new ThreadLocal<Connection>(){
        // 重写initialValue()方法, 方法泛型 T的对象, 父类中的initialValue()方法返回null
        @SneakyThrows
        @Override
        public Connection initialValue(){
            return DriverManager.getConnection("xxxx");
        }
    };
    
    //获取JDBC连接对象的工厂方法
    public static Connection getConnection(){
        // ThreadLocal<T>类的get()方法中调用了initialValue()方法, 返回泛型 T的对象
        return connectionHolder.get();
    }
}

当某个线程使用ConnectionHolderFactory.getConnection()方法想要获取JDBC连接对象时,就是调用ThreadLocal类的get()方法时,此时就会调用initialValue()方法来获取初始化一个JDBC连接对象, 从概念上看,你可以把ThreadLocal<T>视为 HashMap<Thread,T>对象,其中保存了特定于该线程的value值,但ThreadLocal的实现并非如此,这些特定于线程的值保存在Thread对象中,当线程执行结束时,销毁Thread对象时就会将这些值回收

下面提出了TheadLocal<T>的部分源码:

    // ThreadLocal类中的get()方法
    public T get() {
        Thread t = Thread.currentThread();
        // getMap(t)方法调用返回当前线程对象t中的成员变量ThreadLocalMap, 初始化值为null
        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();
    }


    // ThreadLocal类中初始化ThreadLocalMap集合的方法
    private T setInitialValue() {
        T value = initialValue();
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null){
            map.set(this, value);
        } else {
            //createMap()方法的作用是初始化ThreadLocalMap集合,并存入value值
            // ThreadLocalMap map = new ThreadLocalMap(this, value);
            createMap(t, value);
        }
        return value;
    }

ThreadLocal通过下面三点来保证我们使用的变量安全:

1. 每个线程的线程对象不同  Thread t =  Thread.CurrentThread() 即 t对象不同
2. 每个线程对象内部都有一个成员变量 ThreadLocalMap对象, 该对象本质上就是一个Map<k,v>集合对象
3. ThreadLocalMap中以(key,alue)键值对的形式存储数据, 其中key为当前线程创建的ThreadLocal对象, value即是我们想使用的对象变量 , 这样我们可以通过当前线程创建的ThreadLocal对象来获取我们想使用的线程安全对象数据

注意: 使用ThreadLocal设置线程变量,在线程退出时,要清理资源,否则会造成内存溢出, 可以在Interceptor拦截器中的afterCompletion()中进行资源清理