ThreadLocal浅析

ThreadLocal用来存储线程隔离的数据。

Thread类中有一个ThreadLocalMap成员变量threadLocals,ThreadLocalMap中有一个ThreadLocal.ThreadLocalMap.Entry[16]数组table,用来存储该线程的ThreadLocal的存储的值。多个ThreadLocal在设置时通过ThreadLocal的成员变量threadLocalHashCode和数组大小-1与操作,得到存储的下标值。

ThreadLocal的常规用法:

ThreadLocal<String> stringThreadLocal = new ThreadLocal<>(){
  @Override
  // 设置初始值
  protected String initialValue() {
    return "stringThreadLocal initialValue";
  }
};

ThreadLocal<Integer> intThreadLocal = new ThreadLocal<>(){
  @Override
  protected Integer initialValue() {
    return 1;
  }
};

Thread t = new Thread(new Runnable() {
  @Override
  public void run() {
    stringThreadLocal.get();
    stringThreadLocal.set("set new value");
    // 使用完后最好进行移除,防止内存泄漏
    stringThreadLocal.remove();
  }
});

一个线程可以拥有多个ThreadLocal,用来存储多个不同的值,可以使用initialValue()方法进行初始化值。

ThreadLocal是如何保证线程隔离,存储数据的?

ThreaLocal类中有一个静态内部类ThreadLocal.ThreadLocalMap,而ThreadLocalMap中有一个静态内部类ThreadLocal.ThreadLocalMap.Entry,Entry中的成员变量Object value 用来存储ThreadLocal设置的值。

ThreadLocal的get方法

public T get() {
    // 获取当前线程
    Thread t = Thread.currentThread();
    // 当前线程作为key值,获取ThreadLocalMap
    ThreadLocal.ThreadLocalMap map = this.getMap(t);
    if (map != null) {
        // 通过getEntry方法传入当前的ThreadLocal变量,经过运算获取Entry的值
        ThreadLocal.ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            T result = e.value;
            return result;
        }
    }
    // 返回设置的初始值
    return this.setInitialValue();
}

ThreadLocal.ThreadLocalMap getMap(Thread t) {
  // 通过传入当前线程,获取线程的成员变量threadLocals
  return t.threadLocals;
}

// ThreadLocalMap中的getEntry方法
private ThreadLocal.ThreadLocalMap.Entry getEntry(ThreadLocal<?> key) {
  // 计算Entry存储的下标
  int i = key.threadLocalHashCode & this.table.length - 1;
  // 通过ThreadLocal.ThreadLocalMap.Entry[] table 数组获取对应的Entry
  ThreadLocal.ThreadLocalMap.Entry e = this.table[i];
  return e != null && e.get() == key ? e : this.getEntryAfterMiss(key, i, e);
}

可用看到,通过Thread.currentThread()获取当前线程,通过当前线程作为键值,获取到ThreadLocalMap。而这个ThreadLocalMap就是线程中的成员变量threadLocals,这就与Thread对应起来了。再通过获取到的ThreadLocalMap通过getEntry()方法,传入当前的ThreadLocal获取ThreadLocal存储的值。

ThreadLocal的set()方法

public void set(T value) {
    Thread t = Thread.currentThread();
    ThreadLocal.ThreadLocalMap map = this.getMap(t);
    if (map != null) {
        map.set(this, value);
    } else {
        this.createMap(t, value);
    }

}

// ThreadLocalMap中的set方法
private void set(ThreadLocal<?> key, Object value) {
  // 数组,用来存储对应的Entry
  ThreadLocal.ThreadLocalMap.Entry[] tab = this.table;
  int len = tab.length;
  // 通过threadLocalHashCode加上与操作,获取下标
  int i = key.threadLocalHashCode & len - 1;

  // 获取对应的ThreadLocal进行值的设置
  for(ThreadLocal.ThreadLocalMap.Entry e = tab[i]; e != null; e = tab[i = nextIndex(i, len)]) {
    ThreadLocal<?> k = (ThreadLocal)e.get();
    if (k == key) {
      e.value = value;
      return;
    }

    if (k == null) {
      this.replaceStaleEntry(key, value, i);
      return;
    }
  }

  tab[i] = new ThreadLocal.ThreadLocalMap.Entry(key, value);
  int sz = ++this.size;
  if (!this.cleanSomeSlots(i, sz) && sz >= this.threshold) {
    this.rehash();
  }

}

void createMap(Thread t, T firstValue) {
  t.threadLocals = new ThreadLocal.ThreadLocalMap(this, firstValue);
}

ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
  this.table = new ThreadLocal.ThreadLocalMap.Entry[16];
  int i = firstKey.threadLocalHashCode & 15;
  this.table[i] = new ThreadLocal.ThreadLocalMap.Entry(firstKey, firstValue);
  this.size = 1;
  this.setThreshold(16);
}

可以看到,ThreadLocalMap中通过table来保存一个大小为16的ThreadLocal.ThreadLocalMap.Entry数组,而这个数组就是用来存储ThreadLocal.ThreadLocalMap.Entry的,也就是说多个ThreadLocal设置的值,通过这个数组来进行保存。

ThreadLocal的线程不安全

ThreadLocal在设置一个同一个对象的值时,可能存在线程不安全的问题。因为ThreadLocal存储的是栈中一个对象的引用,而对象是存储在堆中的,ThreadLocal对这个对象的引用指向的其实是堆中的同一个对象。所以一个线程对这个对象的值进行更改时,其它ThreadLocal在取值时也会被修改。

解决方法:1. 不要存储同一个对象。2. 使用initialValue()对每个TreadLocal进行赋值初始值。

Integer的valueOf方法:

public static Integer valueOf(int i) {
    return i >= -128 && i <= Integer.IntegerCache.high ? Integer.IntegerCache.cache[i + 128] : new Integer(i);
}

在这可以看到,如果对Integer的值进行加减操作,并且存储的值不是在-128~127之间的时候,会创建一个新的Integer对象。

ThreadLocal导致内存泄漏

ThreadLocal内存泄漏图解

使用ThreadLocal进行存取值时,传入的ThreadLocal作为键值时用弱引用包裹的,当发生GC时,如果ThreadLocal被GC回收掉了,那么这个key值就不存在了,而Thread中存储的ThreadLocal.ThreadLocalMap 是一个强引用,不会被GC回收,这种情况下,就会发生内存泄漏。只有当前thread结束以后, current thread就不会存在栈中,强引用断开, Current Thread, Map, value将全部被GC回收。

所以当ThreadLocal使用完成的时候,需要调用remove方法将value移除掉,这样就不会存在内存泄漏了。

key值为什么使用弱引用而不是强引用?

下面我们分两种情况讨论:

key 使用强引用:对 ThreadLocal 对象实例的引用被置为 null 了,但是 ThreadLocalMap 还持有这个 ThreadLocal 对象实例的强引用,如果没有手动删除, ThreadLocal 的对象实例不会被回收,导致 Entry 内存泄漏。

key 使用弱引用:对 ThreadLocal 对象实例的引用被被置为 null 了,由于 ThreadLocalMap 持有 ThreadLocal 的弱引用,即使没有手动删除,ThreadLocal 的对象实例也会被回收。value 在下一次 ThreadLocalMap 调用 set,get,remove 都 有机会被回收。

比较两种情况,我们可以发现:由于 ThreadLocalMap 的生命周期跟 Thread 一样长,如果都没有手动删除对应 key,都会导致内存泄漏,但是使用弱引用可 以多一层保障。

因此,ThreadLocal 内存泄漏的根源是:由于 ThreadLocalMap 的生命周期跟Thread 一样长,如果没有手动删除对应 key 就会导致内存泄漏,而不是因为弱引用。

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

推荐阅读更多精彩内容

  • 前言 最近在肝原神,略微有些玩物丧志的赶脚。但鲁迅先生说过任何人都逃不过真香定律,即使他再喜欢学习也不行。----...
    小艾咪阅读 82评论 0 0
  • 1、背景 在使用我们的ToolBox(我们公司的数据分析平台) 在往greenplum使用goload 入数据的...
    早点起床晒太阳阅读 116评论 0 0
  • 什么是ThreadLocal: ThreadLocal 提供了线程本地的实例。 它与普通变量的区别在于,每个使用该...
    YocnZhao阅读 137评论 0 1
  • 概述 ThreadLocal如果单纯从名字上来看像是“本地线程"这么个意思,只能说这个名字起的确实不太好,很容易让...
    eliter0609阅读 487评论 0 0
  • 一、ThreadLocal简介及误区 ThreadLocal一般称为线程本地变量,它是一种特殊的线程绑定机制,将变...
    无量散人阅读 278评论 0 2