Java ThreadLocal 实现原理

ThreadLocal 线程本地变量,算是Java开发中比较常用的API了,今天我们来一探究竟

使用场景

ThreadLocal 适用于每个线程需要自己独立的实例且该实例需要在多个方法中被使用,也就是变量在线程间隔离,而在同一线程共享的场景。例如管理Connection,我们希望每个线程只使用一个Connection实例,这个时候用ThreadLocal就很合适。

public class ThreadLocalDemo {
    private static final ThreadLocal<Object> threadLocal = new ThreadLocal<>();

    public static void main(String[] args) {
        threadLocal.set(new Object());
        someMethod();
    }

    static void someMethod() {
        // 获取在threadLocal中存储的对象
        threadLocal.get();
        
        // 清除ThreadLocal中数据
        threadLocal.remove();
    }
}

还有之前写过的一篇动态切换数据源 https://www.jianshu.com/p/0a485c965b8b,AOP 通过 ThreadLocal 保存当前线程需要访问的数据源的key,AbstractRoutingDataSource 再通过 ThreadLocal 中的数据切换到指定的数据源,对业务代码毫无入侵

原理

在我们了解了如何使用之后,来看下 ThreadLocal 是如何实现的

ThreadLocal.get()

我们从get方法来分析,可以看到方法中获取当前线程,并通过当前线程得到一个 ThreadLocalMap,我们可以暂时把这个ThreadLocalMap 理解为我们熟悉的HashMap,然后通过 this(当前ThreadLocal对象)作为key,从Map中获取Entry

图1

我们再来看下,ThreadLocalMap 以及ThreadLocalMap.Entry 中的核心成员变量,ThreadLocalMap 中实现了一个简单的hash表

图2

看到这里你可能还不是很清晰,结合下面这张图理解一下,每个线程(Thread对象)中有一个ThreadLocalMap,使线程之间的数据天然隔离,ThreadLocalMap 有一张hash表 Entry[],每个 Entry 中对应存储着一个ThreadLocal实例 - value,这样使得不同的ThreadLocal 对象之间也形成了隔离

图片来自网络
ThreadLocalMap 中的hash表

我们通过 ThreadLocalMap.set() 来了解下内部的hash表是如何实现的

图3

线性探测是指当发生hash冲突时,利用固定的算法寻找一定步长的下个位置(ThreadLocal中发生hash冲突时,index+1),依次判断,直至找到能够存放的位置

如果线程中操作了大量的 ThreadLocal 对象,势必会造成hash冲突,这是没有必要的性能开销,如果可以的话,我们可以只保留一个ThreadLocal对象

关于 ThreadLocal 的一些思考

  1. 为什么要使用弱引用

图3中,我们看到hash表中会出现 key == null的Entry,这是因为 ThreadLocalMap.Entry 的key (Entry 对ThreadLocal设置了弱引用,可以回顾一下图2)

弱引用的对象拥有更短暂的生命周期。在GC时,一旦发现了对象只具有弱引用,这个对象一定被回收

这么做的原因:如果ThreadLocal 对象需要被回收时(此时并没有调用ThreadLocal.remove),线程中的ThreadLocalMap 一直强引用着 ThreadLocal对象,这会让 ThreadLocal对象 以及对应的value对象内存无法释放,导致内存泄漏。这算是ThreadLocal的一种容错机制,这样做使得了ThreadLocal对象得到了回收,但是value的内存并没有释放,所以ThreadLocalMap 的get、set方法中都会去尝试清理ThreadLocal已经被回收的entry。

  1. 使用过后不及时remove会怎么样

很多博客中都强调了,ThreadLocal.remove的重要性。举个例子,我们新启了一个线程在这个线程中使用了ThreadLocal,我们并没有调用remove,这会导致存储的value对象一直没有办法被回收,直到线程被销毁

  1. 线程池中也需要remove吗

以web线程池为例,如果每次都在过滤器中操作同一个ThreadLocal.set,然后业务代码中get,似乎没什么问题。计算出的hash值都是一样的,槽位也是一样的会覆盖上一次的值。确实业务不会有问题,但是还是推荐大家在使用完之后remove,因为这样会让无用的value对象早点被回收,在很多java源码中都会看到,对一些不再使用的对象进行如下的help GC操作

object = null // help GC

所以我们也需要让无用的对象失去引用,帮助GC

  1. 综上所述

ThreadLocal 使用过后要及时remove,帮助JVM释放内存

参考

https://www.jianshu.com/p/98b68c97df9b

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,271评论 5 466
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,725评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,252评论 0 328
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,634评论 1 270
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,549评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 47,985评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,471评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,128评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,257评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,233评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,235评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,940评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,528评论 3 302
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,623评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,858评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,245评论 2 344
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,790评论 2 339

推荐阅读更多精彩内容