ThreadLocal源码剖析

简述

  • ThreadLocal是用来解决共享对象的多线程访问问题的,一般情况下,通过ThreadLocal.set(Object obj) 到线程中的对象obj是该线程自己使用的对象,其他线程是不需要访问的,也访问不到的。各个线程中访问的是不同的对象。
  • 线程局部变量功用非常简单,就是为每一个使用该变量的线程都提供一个变量值的副本,是每一个线程都可以独立地改变自己的副本,而不会和其它线程的副本冲突。从线程的角度看,就好像每一个线程都完全拥有该变量。
  • ThreadLocal使得各线程能够保持各自独立的一个对象,并不是通过ThreadLocal.set()来实现的,而是通过每个线程中的new对象的操作来创建的对象,每个线程创建一个,不是什么对象的拷贝或副本。
  • ThreadLocal适用于资源共享但不需要维护状态的情况,也就是一个线程对资源的修改,不影响另一个线程的运行;这种设计是‘空间换时间’,synchronized顺序执行是‘时间换取空间’。

工作原理

在某类中定义ThreaLocal对象threadLocal。如下:

private static ThreadLocal<String> threadLocal = new ThreadLocal<String>();
  • 上面我们说每个线程有一个ThreadLocal的拷贝。具体到底是怎么样的呢??
    1、首先每个线程都有一个私有的ThreadLocalMap的引用map
    2、当我们第一次调用threadLocal.set()/threadLocal.get()方法的时候会对这个map进行初始化。
    3、这个ThreadLocalMap就是用来保存(threadLocal,value)这样的键值对的,即每个使用threadLocal的线程都是将其作为键而指定的值作为value保存在这个map中,而map是每个线程私有的,因此是独立的,可以随意操作。
  • 当我们调用threadLocal.get()方法时,他首先会拿到当前线程的ThreadLocalMap对象map,由于这个ThreadLocalMap对象map中保存了(threadLocal,value)的键值对,因此根据map.get(threadLocal)来拿到相应的value值,这样我们就可以随意来操作这个value,不会影响其他线程中的value。

源码剖析

构造方法

 public ThreadLocal() {
 }

get方法

  public T get() {
     Thread t = Thread.currentThread();
     ThreadLocalMap map = getMap(t); //得到当前线程的ThreadLocalMap对象 成员变量
     if (map != null) {
        ThreadLocalMap.Entry e = map.getEntry(this);  //根据ThreadLocal得到键值对对象Entry。
        //检查是否为空,如果不为空,则取出value即可,如果为空,则说明此ThreadLocal随关联的键
    //值对被垃圾回收了,这里就再一次的调用了setInitialValue()方法进行初始化
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
     }
        return setInitialValue();//利用默认值来为当前线程创建一个ThreadLocalMap对象,如果没有
                             //重写inititalValue(),则默认值为null.
    }
  • get()方法的功能:返回当前线程的ThreadLocal副本中的值value,如果当前线程的这个变量没有值,则首先调用inititalValue这个方法来初始化value。
  • 第一次使用ThreadLocal.get()时,这个getMap(t)的返回值肯定是null。

setInitialValue方法

  private T setInitialValue() {
        T value = initialValue();
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
        return value;
    }
  • 方法执行流程
    1.调用initialValue()来获取初始值,ThreadLocal的默认实现是返回null,如果程序员期望这个ThreadLocal变量有一个初始值(不是null),则我们需要在子类(内部类)中重写这个方法。一般情况下,这个方法至多被每个线程调用一次。但是,在调用remove方法之后紧跟着调用了get方法则会又一次的调用此方法。
    2.和ThreadLocal的get()方法一样,调用了getMap(t)方法来获取当前线程的ThreadLocalMap对象。如果getMap(t)返回为null,则调用createMap(t,value)方法来实例化当前线程的ThreadLocalMap对象。如果不为空,则将(threadLocal,value)作为键值对保存在这个ThreadLocalMap中。
    3.createMap(t,value)方法来实例化当前线程的ThreadLocalMap对象,方法的代码如下:
    void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

ThreadLocalMap.Entry e = map.getEntry(this)方法;

  • 根据ThreadLocal(即this)在ThreadLocalMap中找到与ThreadLocal关联的键值对对象Entry。
  • 如果这个Entry对象不为null,则直接取出value值返回即可。如果这个Entry为null,则说明此ThreadLocal关键的键值对被垃圾回收了,因此再一次的调用setInitialValue()方法进行初始化。

总结

1、第一步:得到当前线程的ThreadLocalMap对象map.
2、第二步:如果这个map对象为null,则进行实例化,如果不为null,则根据当前的ThreadLocal作为key取出我们想要的结果value。

内部类ThreadLocalMap

  • ThreadLocalMap是ThreadLocal的一个内部类, ThreadLocalMap底层借用了一个数组table来保存键值对
    (ThreadLocal,value).ThreadLocal的hashCode&(table.length-1)作为数组下标.
  • 构造方法
 ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
     table = new Entry[INITIAL_CAPACITY];
     int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1); //求得此ThreadLocal在数组中保存的位置
     table[i] = new Entry(firstKey, firstValue);
     size = 1;
     setThreshold(INITIAL_CAPACITY);
 }
  • 内部类Entry
  static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;

            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
  }
  • ThreadLocalMap.Entry类继承了WeakReference(弱引用),如果entry.get()==null,意味着key不在引用,因此在table中的键值对就会被去除。
  • 关于ThreadLocalMap我们需要了解的一点是:它是用来保存(threadLocal,value)键值对。当我们需要使用(threadLocal,value)键值对中的value时,只需要使用entry.get(threadLocal)即可获得。

set(T value)方法

  public void set(T value) {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
    }
  • 函数功能:设置当前线程对ThreadLocal变量副本的值为指定值
  • 思路如下:
  1. 首先也是获取当前线程的ThreadLocalMap对象map。
  2. 如果map为空,则实例化这个ThreadLocalMap对象并将(threadLocal,value)作为一个Entry对象保存在这个map中。
  3. 如果map不为空,则直接(threadLocal,value)作为一个Entry对象保存在这个map中(如果ThreadLocal已经作为key在这个map中存在,则修改对应的value即可)。
    如果map不为空,则会调用ThreadLocalMap类的set方法。
  private void set(ThreadLocal<?> key, Object value) {
            Entry[] tab = table;
            int len = tab.length;
            //首先也是根据key的hash值得到其在数组中的存储位置
            int i = key.threadLocalHashCode & (len-1);
            /*
                对于第hashcode对应的槽有存储元素,则说明发生了hash碰撞。
                发生hash碰撞的解决方法是:以加一的形式逐渐遍历整个数组,直到找到key或者是找到一个空位。
            */
            for (Entry e = tab[i];
                 e != null;
                 e = tab[i = nextIndex(i, len)]) {
                ThreadLocal<?> k = e.get();
                if (k == key) { //此key本来存在,则更新其值即可。
                    e.value = value;
                    return;
                }
                if (k == null) { //找到一个空位
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }
            //对于此key的hashcode对应的槽没有存储元素,则会直接新建一个对象并存储在这个位置上。
            tab[i] = new Entry(key, value);
            int sz = ++size;
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();
        }

remove()方法

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

推荐阅读更多精彩内容