ThreadLocal的原理图
在线程任务Runnable中,使用一个及其以上ThreadLocal对象保存多个线程的一个及其以上私有值,即一个ThreadLocal对象可以保存多个线程一个私有值。
(重点)每一个线程Thread对象,都有一个threadLocals属性;
核心属性,因为每个Thread对象(线程)都拥有自己私有的threadLocals属性,ThreadLocal对象通过操作(set和get)每个线程自身私有属性threadLocals(数据结构:ThreadLocal.ThreadLocalMap)做到让每个线程拥有自己的私有变量;
threadLocals数据结构是ThreadLocal的内部类ThreadLocalMap,这个Map中的key是当前ThreadLocal对象,value是线程想要隔离的值,值的类型是ThreadLocal的泛型。
ThreadLocal源码实现
(1)set方法
ThreadLocal<String> local=new ThreadLocal<String>(); t.set("herock"); 为啥直接设置value就知道该value属于哪个线程?仔细体会上面的文字。
首先获取到当前线程,通过getMap(t)传入线程获得该线程持有的实例变量ThreadLocalMap。
getMap(t)返回线程t的实例变量threadLocals。
如果threadLocals为null就创建map,如果不为null以ThreadLocal实例为key,以所存值为value存入map;
(2)ThreadLocal的get方法
public T get() {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null) {
ThreadLocalMap.Entry e = map.getEntry(this);//这里的this是ThreadLocal对象
if (e != null) {
@SuppressWarnings("unchecked")
T result = (T)e.value;//获得value
return result;
}
}
return setInitialValue();
}
防止内存泄漏
ThreadLocal 被线程的实例变量引用,在线程未死之前, ThreadLocal 变量无法被回收,会导致内存泄漏?其实不然,在源程序中已经有相应的措施了。
ThreadLocalMap 的每个 Entry 都是一个对 键 的弱引用;
看下面的数据结构就知道与HashMap不一样, 没有next, 所以不是用拉链法解决冲突, 在set方法中用的是开放地址法解决key的冲突.
static class Entry extends WeakReference<ThreadLocal<?>> { /** The value associated with this ThreadLocal. */ Object value; Entry(ThreadLocal<?> k, Object v) { super(k); value = v; } }
ThreadLocalMap中的key是弱引用(会在下一次垃圾回收时回收),从而避免上文所述 ThreadLocal 被引用不能被回收而造成的内存泄漏的问题。
但是,这里又可能出现另外一种内存泄漏的问题。
ThreadLocalMap 维护 ThreadLocal 变量与具体value的映射,当 ThreadLocal 变量被回收后,该Entry的key变为 null,该 Entry 无法被移除。从而使得实例被该 Entry 引用而无法被回收造成内存泄漏。
再看一次set方法
针对该问题,ThreadLocalMap 的 set 方法中:
首先,根据key(ThreadLocal对象)的hashcode与len-1进行按位与计算出在ThreadLocalMap中哈希数组的index下标值,tab[i]即为链表的头结点;
然后遍历以tab[i]的头结点链表中的元素,如果有元素等于了该Entry(弱引用的ThreadLocal),就把value付给该entry对应的value;
(重点)如果遍历的元素中有值为null(因为弱引用在垃圾回收时被回收了),就把该位置的元素的设置为null(通过 replaceStaleEntry 方法将所有key为 null 的 Entry 的值value设置为 null,从而使得该值可被回收,然后在把想要插入的key-value插入进去);
如果遍历完了都没有与待插入元素相等的链表节点,就将该键值对进行插入;
另外,会在 rehash 方法中通过 expungeStaleEntry 方法将键和值为 null 的 Entry 设置为 null 从而使得该 Entry 可被回收。通过这种方式,ThreadLocal 可防止内存泄漏。
private void set(ThreadLocal<?> key, Object value) { Entry[] tab = table; int len = tab.length; int i = key.threadLocalHashCode & (len-1); for (Entry e = tab[i]; e != null; e = tab[i = nextIndex(i, len)]) { ThreadLocal<?> k = e.get(); if (k == key) { e.value = value; return; } if (k == null) {
//这里是链表中某个Entry元素的key为null时,这里就要进行清理,然后再把想要插入的值key和value插入进去 replaceStaleEntry(key, value, i); return; } } tab[i] = new Entry(key, value); int sz = ++size; if (!cleanSomeSlots(i, sz) && sz >= threshold) rehash(); }
总结
- ThreadLocal 并不解决线程间共享数据的问题
- 每个线程持有一个 Map 并维护了 ThreadLocal 对象与具体实例的映射,该 Map 由于只被持有它的线程访问,故不存在线程安全以及锁的问题
- ThreadLocalMap 的 Entry 为弱引用,避免了 ThreadLocal 对象无法被回收的问题
- ThreadLocalMap 的 set 方法通过调用 replaceStaleEntry 方法回收键为 null 的 Entry 对象的值(即为具体实例)以及 Entry 对象本身从而防止内存泄漏
- ThreadLocal 适用于变量在线程间隔离且在方法间共享的场景,比如一个Servlet类中有一个属性,有方法对该属性进行写操作和读操作,为了隔离
参考来源:
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 提示词工程——AI应用必不可少的技术
· Open-Sora 2.0 重磅开源!
· 周边上新:园子的第一款马克杯温暖上架