ThreadLocal的使用与原理

  • Post author:
  • Post category:其他


1. 使用场景

  1. 使用单个线程保存上下文信息
  2. 可以使得本来线程不安全的类变得安全,例如DateFormat,如果每个线程只有一个DateFormat,那么就是安全的
  3. 承载一些线程的信息,放在在方法调用的时候来回传递参数

2. 使用方法

@Test
public void testThreadLocal() throws InterruptedException {
    AtomicInteger count = new AtomicInteger(0);
    ThreadFactory threadFactory = new ThreadFactory() {
        @Override
        public Thread newThread(Runnable r) {
            return new Thread(r,"Thread:" + count.incrementAndGet());
        }
    };
    ThreadLocal<Integer> idThreadLocal = new ThreadLocal<>();

    Thread thread1 = threadFactory.newThread(new LoginLogRunnable(1, idThreadLocal));
    Thread thread2 = threadFactory.newThread(new LoginLogRunnable(2, idThreadLocal));

    thread1.start();
    thread2.start();

    Thread.sleep(10000);

}

static class LoginLogRunnable implements Runnable {
    private Integer id;
    private ThreadLocal<Integer> idThreadLocal;

    public LoginLogRunnable(Integer id, ThreadLocal<Integer> idThreadLocal) {
        this.id = id;
        this.idThreadLocal = idThreadLocal;
    }

    @Override
    public void run() {
        idThreadLocal.set(id);
        try {
            Thread.sleep(100);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println(Thread.currentThread().getName() + ":" + idThreadLocal.get());
    }
}

例如上面的代码,我们只需要set进自己的值,当获取的时候通过ThreadLocal.get()就可以获取到我们set的值。

原理

当我们set的时候发生了什么?ThreadLocal如何实现一个修改数据的时候,不会影响到别的别的线程的数据。

这里写图片描述

如图所示,一个Thread会包含一个ThreadLocaMap用来保存一个线程的成员变量,而Entry的每个节点的value对应的是值。

当我们set值的时候,会获取当前线程的ThreadLocalMap,然后将将key为ThreadLocal实例、value为对应值的Entry,添加进ThreadLocalMap。get的时候也是相同的道理。

注意点

  1. 1.即使Entry的key是若引用,如果在set值以后不再访问,但由于value是强引用,所以也无法正常释放内存,有可能造成内存泄漏
  2. ThreadLocal如果不使用了的话,最好是调用remove方法释放掉对应的内存。



版权声明:本文为yao123long原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。