Java-ThreadLocal
在并发编程里,有时候我们需要让每个线程保存自己的私有数据,不想被其他线程干扰,这时候,`ThreadLocal` 就派上用场了。ThreadLocal 不是线程,而是为线程准备的私有数据仓库。
一、ThreadLocal是什么?核心作用是什么?它是怎么解决多线程共享变量的冲突的?
用一句话解释:每个线程有自己的小抽屉,别人看不见。什么时候用?典型场景:用户上下文、数据库连接、日志 TraceId 等。
二、核心结构
Thread、ThreadLocalMap、ThreadLocal三者关系
[Thread] └─ ThreadLocalMap├─ (ThreadLocalA, valueA)├─ (ThreadLocalB, valueB)
ThreadLocalMap
挂在线程身上,不是 ThreadLocal
身上。ThreadLocal
自己只是钥匙,负责把值放到线程的小抽屉里。
三、源代码核心
get和set源码最关键的逻辑
public T get() {Thread t = Thread.currentThread();ThreadLocalMap map = getMap(t);...
}
public void set(T value) {Thread t = Thread.currentThread();ThreadLocalMap map = getMap(t);...
}
Thread.currentThread()
找到自己。ThreadLocalMap
是真正存东西的地方。key 是 ThreadLocal
自己(this),value 是你要放入的值。
四、WeakReference与内存泄漏
- 强引用:我们常常new出来的对象就是强引用类型,只要强引用存在,垃圾回收器将永远不会回收被引用的对象,哪怕内存不足的时候
- 软引用:使用SoftReference修饰的对象被称为软引用,软引用指向的对象在内存要溢出的时候被回收
- 弱引用:使用WeakReference修饰的对象被称为弱引用,只要发生垃圾回收,若这个对象只被弱引用指向,那么就会被回收
- 虚引用:虚引用是最弱的引用,在 Java 中使用 PhantomReference 进行定义。虚引用中唯一的作用就是用队列接收对象即将死亡的通知
ThreadLocalMap
里 key 是弱引用
static class Entry extends WeakReference<ThreadLocal<?>> {Object value;
}
如果ThreadLocal没引用了,key会被GC回收,但value还在,形成泄漏,因此推荐remove(),用完手动清理。
五·一、小例子
示例一:一个线程写私有值,这是验证 ThreadLocal 是当前线程私有存取容器 的最小闭环。
ThreadLocal<String> tl = new ThreadLocal<>();//造钥匙,准备给某个线程存私货
tl.set("hello");//把值塞到当前main线程的ThreadLocalMap里
System.out.println(tl.get()); // 用同一把钥匙去翻找,能拿回“hello”
示例二:多个线程隔离示例
ThreadLocal<Integer> tl = new ThreadLocal<>();//造钥匙
//下边启动五个线程
for (int i = 0; i < 5; i++) {int num = i;new Thread(() -> {tl.set(num);//把自己的 num 值放到自己线程的小抽屉(ThreadLocalMap)里System.out.println(Thread.currentThread().getName() + " -> " + tl.get());}).start();
}
虽然 5 个线程都用的是 同一把 tl
,但是它们访问的 ThreadLocalMap
是自己线程私有的,所以:线程1放0,线程2放1......线程5放4。每个线程对同一个 ThreadLocal
只能拿到自己放进去的值。
五·二、再来
注释都写在代码中,有些长拖着看
import java.util.ArrayList;
import java.util.List;public class ThreadLocalTest {private List<String> messages = new ArrayList<>();public static final ThreadLocal<ThreadLocalTest> holder = ThreadLocal.withInitial(ThreadLocalTest::new); // 定义了一个 ThreadLocal,类型是 ThreadLocal<ThreadLocalTest>,每个线程里面都有自己的一份 ThreadLocalTest 对象。public static void add(String msg) { // 这个对象里有个 List<String> messages,相当于:每个线程都有一份“自己的私有 List”。holder.get().messages.add(msg); // 找当前线程自己的 ThreadLocalTest 对象。messages.add(msg):把消息塞进去。}public static List<String> clear() { // 拿到当前线程的List然后打印一下size,remove():把这个线程的 ThreadLocal 清理掉,防止内存泄漏。List<String> messages = holder.get().messages;holder.remove();System.out.println("size is " + messages.size());return messages;}// 下边启动了 10 个线程,每个线程往自己的 ThreadLocalTest 的 messages 里加一条 "msgX",再把自己那份 messages 打印出来。public static void main(String[] args) {Thread[] threads = new Thread[10];for (int i = 0; i < 10; i++) {int j = i;threads[i] = new Thread(() -> {ThreadLocalTest.add("msg" + j);List<String> messages = holder.get().messages;System.out.println(messages);});threads[i].start();}}
}
ThreadLocal
保证了:每个线程有自己的 ThreadLocalTest1
对象,互不干扰;同一个静态 holder
,放进去的却是每个线程自己的 ThreadLocalMap
里的一份,所以即使你写了 static
,每个线程也完全隔离,这就是 ThreadLocal 的意义。