ThreadLocal的简单介绍

2024-03-09 14:53:59

我们在接触一个新东西时，首先应该是先用起来，然后再去探究内部原理

Thread Local的使用还是比较简单的，类似Map，各种put/get

它的核心方法如下：

  • public void set(T value)：保存当前副本到ThreadLocal中，每个线程单独存放
  • public T get()：取出刚才保存的副本，每个线程只会取出自己的副本
  • protected T initialValue()：初始化副本，作用和set一样，不过initialValue会自动执行，如果get()为空
  • public void remove()：删除刚才保存的副本

1. 例子 - 数据隔离

这里我们用SimpleDateFormat举例，因为这个类是线程不安全的（后面有空再单独开篇），如果不做隔离，会有各种各样的并发问题

我们先来看下线程不安全的例子,代码如下：

public class ThreadLocalDemo {

    // 线程不安全：在多个线程中执行时，有可能解析出错
    private SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
    public void parse(String dateString){
        try {
            System.out.println(simpleDateFormat.parse(dateString));
        } catch (ParseException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        ExecutorService service = Executors.newFixedThreadPool(10);
        ThreadLocalDemo demo = new ThreadLocalDemo();
        for (int i = 0; i < 30; i++) {
            service.execute(()->{
                demo.parse("2020-01-01");
            });
        }
    }
}

多次运行，可能会出现下面的报错：

Exception in thread "pool-1-thread-4" java.lang.NumberFormatException: empty String
  • 1

关于SimpleDateFormat的不安全问题，在源码注释里有提到，如下：

Date formats are not synchronized. It is recommended to create separate format instances for each thread. If multiple threads access a format concurrently, it must be synchronized externally.
  • 1

意思就是建议多线程使用时，要么每个线程单独创建，要么加锁

下面我们分别用加锁和单独创建来解决

线程安全的例子：加锁

public class ThreadLocalDemo {

    // 线程安全1：加内置锁
    private SimpleDateFormat simpleDateFormatSync = new SimpleDateFormat("yyyy-MM-dd");
    public void parse1(String dateString){
        try {
           synchronized (simpleDateFormatSync){
               System.out.println(simpleDateFormatSync.parse(dateString));
           }
        } catch (ParseException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        ExecutorService service = Executors.newFixedThreadPool(10);
        ThreadLocalDemo demo = new ThreadLocalDemo();
        for (int i = 0; i < 30; i++) {
            service.execute(()->{
                demo.parse1("2020-01-01");
            });
        }
    }
}

线程安全的例子：通过ThreadLocal为每个线程创建一个副本

public class ThreadLocalDemo {

    // 线程安全2：用ThreadLocal创建对象副本，做数据隔离
    // 下面这个代码可以简化，通过 withInitialValue
    private static ThreadLocal<SimpleDateFormat> threadLocal = new ThreadLocal<SimpleDateFormat>(){
        // 初始化方法，每个线程只执行一次；比如线程池有10个线程，那么不管运行多少次，总的SimpleDateFormat副本只有10个
        @Override
        protected SimpleDateFormat initialValue() {
            // 这里会输出10次，分别是每个线程的id
            System.out.println(Thread.currentThread().getId());
            return new SimpleDateFormat("yyyy-MM-dd");
        }
    };
    public void parse2(String dateString){
        try {
            System.out.println(threadLocal.get().parse(dateString));
        } catch (ParseException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        ExecutorService service = Executors.newFixedThreadPool(10);
        ThreadLocalDemo demo = new ThreadLocalDemo();
        for (int i = 0; i < 30; i++) {
            service.execute(()->{
                demo.parse2("2020-01-01");
            });
        }
    }
}

有的朋友可能会有疑问，这个例子为啥不直接创建局部变量呢？

这是因为如果创建局部变量，那么调用一次就会创建一个SimpleDateFormat，性能会比较低

而通过ThreadLocal为每个线程创建一个副本，那么基于这个线程的后续所有操作，都是访问这个副本，无需再次创建

2. 例子 - 传参优化

有时候，我们需要在多个方法之间进行传参（比如用户信息），此时就面临一个问题：

  • 如果将要传递的参数设置为全局变量，那么线程不安全
  • 如果将要传递的参数设置为局部变量，那么传参会很麻烦

这时就需要用到ThreadLocal了，正如开篇讲得，它的作用就是融合全局和局部的优点，使得线程也安全，传参也方便

下面是例子：

public class ThreadLocalDemo2 {

    // 参数传递，程序繁琐
    public void fun1(int age){
        System.out.println(age);
        fun2(age);
    }
    private void fun2(int age){
        System.out.println(age);
        fun3(age);
    }
    private void fun3(int age){
        System.out.println(age);
    }

    public static void main(String[] args) {
        ExecutorService service = Executors.newFixedThreadPool(10);
        ThreadLocalDemo2 demo = new ThreadLocalDemo2();
        for (int i = 0; i < 30; i++) {
            final int j = i;
            service.execute(()->{
                demo.fun1(j);
            });
        }
    }
}

这段代码可能没有实际意义，但是意思应该到了，就是表达传递参数的繁琐性

下面我们看下用ThreadLocal来解决这个问题

public class ThreadLocalDemo2 {

    // 简化，ThreadLocal当全局变量来使用
    private static ThreadLocal<Integer> threadLocal = new ThreadLocal<Integer>();
    public void fun11(){
        System.out.println(threadLocal.get());
        fun22();
    }
    private void fun22(){
        System.out.println(threadLocal.get());
        fun33();
    }
    private void fun33(){
        int age = threadLocal.get();
        System.out.println(age);
    }

    public static void main(String[] args) {
        ExecutorService service = Executors.newFixedThreadPool(10);
        ThreadLocalDemo2 demo = new ThreadLocalDemo2();
        for (int i = 0; i < 30; i++) {
            final int j = i;
            service.execute(()->{
                try{
                    threadLocal.set(j);
                    demo.fun11();
                }finally {
                    threadLocal.remove();
                }
            });
        }
    }
}

可以看到，这里我们不再把age参数传来传去，而是为每个线程创建一个副本age

这样所有方法都可以访问到副本，同时也保证了线程安全

不过要注意的是，这次的使用和上次不同，这次多了remove方法，它的作用就是删除上面set的副本，这个下面再介绍

3. 内部原理

先来说说它是怎么做到数据隔离的

我们先来看下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);
}

可以看到，值是存在map里的（key是ThreadLocal对象，value就是为线程单独创建的副本)

而这个map是怎么来的呢？再来看下面的代码

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

可以看到，最终还是回到了Thread里面，这就是为啥线程之间实现了隔离，而线程内部实现了共享（因为是线程内的属性，只有当前线程可见）

我们再看下get()方法，如下:

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

可以看到，先找到当前线程内的map,然后再根据key取出value

最后一行的setInitialValue，就是在get为空时，重新执行的初始化动作

为什么要用ThreadLocal作为key，而不是线程id呢

是为了存储多个变量

如果用了线程id作为key，那么map里一个线程只能存放一个变量

而用了ThreadLocal作为key，那么可以一个线程存放多个变量（通过创建多个ThreadLocal）

如下所示：

private static ThreadLocal<Integer> threadLocal1 = new ThreadLocal<Integer>();
private static ThreadLocal<Integer> threadLocal2 = new ThreadLocal<Integer>();

public void test(){
    threadLocal1.set(1);
    threadLocal2.set(2);
    System.out.println(threadLocal1.get());
    System.out.println(threadLocal2.get());
}

再来说下它的内存泄漏问题

我们先来看下ThreadLocalMap内部代码：

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

可以看到，内部节点Entry继承了弱引用（在垃圾回收时，如果一个对象只有弱引用，则会被回收），然后在构造函数中通过super(k)将key设置为弱引用

因此在垃圾回收时，如果外部没有指向ThreadLocal的强引用，那么就会直接把key回收掉

此时key=null，而value还在，但是又取不出来，久而久之，就会出现问题

解决办法就是remove，通过在finally中remove，将副本从ThreadLocal中删除，此时key和value都被删除

总结

  1. ThreadLocal直译为线程本地变量，它的作用就是通过为每个线程单独创建一个副本，来保证线程间的数据隔离和简化方法间的传参
  2. 数据隔离的本质：Thread内部持有ThreadLocalMap对象，创建的副本都是存在这里，所以每个线程之间就实现了隔离
  3. 内存泄漏的问题：因为ThreadLocalMap中的key是弱引用，所以垃圾回收时，如果key指向的对象没有强引用，那么就会被回收，此时value还存在，但是取不出来，时间长了，就有问题（当然如果线程退出，那value还是会被回收）
  4. 使用场景：面试等场合
上一篇：ThreadLocal用法及原理


下一篇：并发编程之：ThreadLocal