这可能是ThreadLocal说得最清楚的一篇文章

本文原创地址,我的博客https://jsbintask.cn/2019/04/01/jdk/jdk8-threadlocal/
)(食用效果最佳),转载请注明出处!

前言

ThreadLocal是jdk中一个非常重要的工具,它可以控制堆内存中的对象只能被指定线程访问,如果你经常阅读源码,基本在各大框架都能发现它的踪影。而它最经典的应用就是事务管理,同时它也是面试中的常客。

原理

我们知道,堆内存是共享的,为什么ThreadLocal能够控制指定线程访问呢? 如图:


ThreadLocal
  1. 调用ThreadLocal的get方法。
  2. 获取当前线程t1.
  3. 获取t1的成员变量ThreadLocalMap
  4. 根据ThreadLocal的hashcode计算出ThreadLocalMap中Entry[]数组的索引。
  5. 返回索引位置的值。
    这样我们就很容易理解了,为什么只有当前线程才能获取到某些值,因为这是这些值都直接保存在当前线程的成员变量ThreadLocalMap中,而ThreadLocal在这个过程中充当的角色则是提供它独一无二的hashcode值,这样我们就能计算出我们保存的值在ThreadLocalMap的位置。

源码分析

我们从构建一个ThreadLocal到调用它的set,get方法完整的分析一遍它的源码。

构造器

当我们使用new ThreadLocal<>() new一个ThreadLocal对象时,它初始化了一个成员变量threadLocalHashCode,这个成员变量代表当前ThreadLocal的hashcode值,而它肯定是唯一的:

ThreadLocal

  1. ThreadLocal内部有一个静态hashCode生成器nextHashCode
  2. 每次新new一个ThreadLocal对象,调用这个生成器同步方法获取hashcode。
    因为依赖于静态成员变量nextHashCode的关系,所以它的hashcode肯定唯一!

set(T t)

public void set(T value) {
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
}
  1. 获取当前线程t。
  2. 从t中获取ThreadLocalMap map。
    ThreadLocal
  3. 如果map不为空,将当前值value放入map。
  4. 如果map为空,新建一个ThreadLocalMap放入线程t。
    ThreadLocalMap是ThreadLocal中的内部类,它的结构如下:
public class ThreadLocalMap {
    
    static class Entry extends WeakReference<ThreadLocal<?>> {
        Object value;

        Entry(ThreadLocal<?> k, Object v) {
            super(k);
            value = v;
        }
    }

    private Entry[] table;
    
    private int size = 0;
    
    private static final int INITIAL_CAPACITY = 16;
    
    private int threshold; // Default to 0
}

类似于ArrayList内部的构造,它内部有一个Entry数组table,并且Entry继承自弱引用,所以每一个Entry中保存着两个值,ThreadLocal,value,value即是我们要保存的值。这里值得注意的是,它保存的ThreadLocal为弱引用(gc时会被标记清理).
接着,我们回过头详细分析第三步,ThreadLocalMap的set方法:

private void set(ThreadLocal<?> key, Object value) {
            Entry[] tab = table;
            int len = tab.length;
            int i = key.threadLocalHashCode & (len-1);  // 1

            for (Entry e = tab[i];
                 e != null;         // 2
                 e = tab[i = nextIndex(i, len)]) {
                ThreadLocal<?> k = e.get();

                if (k == key) {   // 3
                    e.value = value;
                    return;
                }

                if (k == null) {   // 4
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }

            tab[i] = new Entry(key, value);  // 5
            int sz = ++size;
            if (!cleanSomeSlots(i, sz) && sz >= threshold)   // 6
                rehash();
        }
  1. 根据ThreadLocal的hashCode计算出在entry中的索引i。
  2. 取出i对应的Entry值e。
  3. 如果e的key等于当前ThreadLocal,代表已经有一个一样的ThreadLocal在这个entry设值,直接替换这个entry上的value。
  4. e上面的ThreadLocal为null,代表垃圾收集器准备回收这个Entry了,我们需要放入一个新的Entry。重新计算数组大小,重新hash。
  5. i位置还没有初始化(第一次set这个ThreadLocal),直接将value放到i的位置。
  6. 扩容Entry数组。

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();
}
  1. 获取当前线程。
  2. 从当前线程中获取ThreadLocalMap
  3. 从ThreadLocalMap中找出ThreadLocal对应的Entry.
  4. 如果Entry不为null,直接返回Entry中的value
  5. 返回初始值。
    其中,ThreadLocalMap的get(ThreadLocal tl)如下:
    ThreadLocal

    它和我们一开始的分析一样,根据ThreadLocal的hashcode成员变量计算出索引位置i,得到Entry。这里同样有特殊情况,如果得到的Entry的key和当前ThreadLocal不相等,代表这个Entry将被垃圾收集处理,调用getEntryAfterMiss rehash,计算数组大小。

注意事项

从上面的代码分析中,我们知道,ThreadLocalMap的生命周期和当前线程同步,如果当前线程被销毁,则map中的所有引用均被销毁。但如果当前线程不被销毁呢(线程池,tomcat处理请求等)?Entry中保存了ThreadLocal的弱引用以及value,gc时可能清理掉ThreadLocal,而这个value确再没有访问之地,这个时候就会造成内存泄漏!
所以我们需要手动调用remove方法清理掉当前线程ThreadLocalMap的引用!

总结

  1. ThreadLocal中真正保存的值还是在线程的ThreadLocalMap中,ThreadLocal只是使用它的hashcode值充当中间计算变量。
  2. ThreadLocalMap内部使用一个Entry数组保存数据。
  3. ThreadLocal可能出现内存泄漏的情况,最好手动调用remove方法。

关注我,这里只有干货!

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

推荐阅读更多精彩内容