Java并发之ThreadLocal使用和源码分析

本篇文章首先给出了ThreadLocal常用的方法,并举例来说明ThreadLocal的使用。然后分析了ThreadLocal的源代码中的get()方法,初始化方法,ThreadLocalMap的实现等来理解ThreadLocal的原理。

ThreadLocal为每个线程提供一个独立的对象副本,每个线程可以单独访问自己独有的对象,不存在多线程同时访问一个对象时的共享问题。

一.使用ThreadLocal

ThreadLocal是一个泛型类,使用时要指定持有的对象的类型。一般会重写initialValue()方法,来设置线程第一次通过ThreadLocal获取对象时的初始值。

常用方法:

\\1.get()方法 获取对象引用,每个线程获取到的不同
public T get(); 
\\2.set()方法,修改当前线程在ThreadLocal中存储的值
public void set(T value);
\\3.initialValue()方法,设置初始值,默认返回null,一般被重写来设置初始值
protected T initialValue();

具体例子

public class ThreadLocalDemo {
    static ThreadLocal<String> threadLocal = new ThreadLocal<String>(){
        @Override
        protected String initialValue() {
            return Thread.currentThread().getName();
        }
    };

    public static void main(String[] args) throws Exception{
        Thread thread1 = new Thread(new Runnable() {
            public void run() {
                System.out.println(threadLocal.get());
            }
        });

        thread1.start();
        thread1.join();
        System.out.println(threadLocal.get());
    }
}

输出结果:

Thread-0
main

可以看到,主线程和thread1通过threadLocal变量分别持有了各自的String对象。

二.源码分析

get方法的主要流程
我们从get方法入手,看看ThreadLocal的实现 原理,先po上源代码:

public T get() {
    //先获取当前线程
    Thread t = Thread.currentThread();   
    //获取当前线程的ThreadLocalMap
    ThreadLocalMap map = getMap(t);   
    if (map != null) {
        //从ThreadLocalMap中获取到要的值,注意这里使用的是当前的ThreadLocal对象作为key的
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
    }
    //当前线程的ThreadLocalMap为null,或者map里没有保存和当前ThreadLocal相关的值时
    return setInitialValue();                   
}

从代码中可以总结出TheadLocal.get的流程:

  • 先获取到当前的线程t,
  • 获取到t中保存的ThreaaLocalMap,如果没有则转到最后一步
  • 以当前ThreadLocal作为键值,从上述map中获取到要的变量
  • 如果获取到变量为null,则转到最后一步进行初始化
  • 初始化thread的map或保存的变量值

查看getMap(t):从线程中获取ThreadLocalMap的代码发现:Thread类中有一个名为threadLocals的ThreadLocalMap类型的变量,初始值为null,在第一次ThreadLocal.get时初始化。getMap会获取这个变量。

setInitialValue方法分析
下面再分析 一下setInitialValue()

private T setInitialValue() {
    //获取初始值,这里调用的方法常常被重写,来设置初始值。默认返回Null
    T value = initialValue();
    //获取当前线程
    Thread t = Thread.currentThread();
    //先尝试获取map,因为也不知道是因为没有map还是因为map中没有值才进来这个方法的
    ThreadLocalMap map = getMap(t);
    //map存在,放入value值,注意这里的key是当前ThreadLocal对象;和get中的map.getEntry相呼应
    if (map != null)
        map.set(this, value);
    else
        //map不存在,创建map
        createMap(t, value);
    return value;
}

看一下createMap的代码:

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

直接调用了ThreadLocalMap的构造方法,将其初始化并返回。这个构造方法不再细究了,只要记住,传入的两个参数是用来作为初始化后的第一个键值对存入该map的即可。
总结一下setInitialValue()方法:

  • 获取value的初始化值,从initialValue方法,这里就看出initialValue的作用了
  • 如果Thread有ThreadLocalMap,则直接使用,存入当前ThreadLocal->value的键值对
  • 如果没有map,则创建map,并同时存入键值对
  • 返回value

ThreadLocalMap简析

ThreadLocalMap是ThreadLocal的一个内部类。在每一个Thread都持有一个ThreadLocalMap变量(如果该线程没用的ThreadLocal则为null)。它是一个map,key是ThreadLocal类型,value是保存的值。

这样设计就保证了一个Thread可以存储多个ThreadLocal。

ThreadLocalMap的Entry设计:
一个map中肯定有一个Entry来存储键值对,ThreadLocal的Entry的设计的较为特殊:

static class Entry extends WeakReference<ThreadLocal<?>> {
    /** The value associated with this ThreadLocal. */
    Object value;

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

可以看到它继承了WeakReference。我们知道WeakReference是四大引用的一种叫做弱引用,它的含义是当对象只被弱引用引用时,下次GC时该对象就会被回收。

Q1:ThreadLocalMap的Entry为什么继承WeekReference?
(个人理解)这样当ThreadLocal变量生命周期结束后,对应的线程的ThreadLocalMap中存储的相应键值对也会被回收,不会造成内存泄露。

参考资料

本文是Java并发专题(欢迎大家关注)的一篇。
以下是完整的目录:
Java并发之基础知识
Java并发之volatile关键字
Java并发之synchronized关键字
Java并发之原子类
Java并发之线程池
Java并发之并发工具类
Java并发之AQS原理
Java并发之ThreadLocal使用和源码分析


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

推荐阅读更多精彩内容