ThreadLocal源码解读

一.threadlocal概述

顾名思义线程本地存储,如果定义了一个threadlocal对象,每个线程往这个threadlocal对象中的读写是隔离的,可以为相同对象在不同线程中都创建不同的存储,因此如果多个线程都使用变量x,那线程本地存储会生成多个用于x的不同存储块。

二.threadlocal实现的大致思路

Thread类有一个ThreadLocal.threadLocalMap类型的实例变量threadlocals,也就是说每个线程都有一个threadLocalMap,threadLocalMap为定义在threadlocal内的静态内部类,虽是一个map但没有实现map接口,有自己独立的实现。threadLocalMap的是使用开放寻址线性探测的哈希表,节点Entry继承自弱引用WeakReference,本身为指向threadlocal的弱引用,有一个value属性用来存放放到threadlocal内的值。可以简单的认为key为threadlocal对象,值为放入threadlocal的具体值。每个线程往threadlocal中塞值的时候,都会以该threadlocal对象为key,往自己的threadlocalmap中存,读也是以某个threadlocal为引用,在自己的threadlocalmap中找对应的key,从而实现线程隔离。
图为Thread类中threadlocalmap实例变量:

ThreadLocal.ThreadLocalMap threadLocals = null;

三.API

image.png

threadLocalMap的api:


image.png

四.threadlocal API

get方法:

    public T get() {
        //获取当前线程的threadlocalmap
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        //如果线程中的threadlocalmap非空,则以这个threadlocal为key查询对应的value
        if (map != null) {
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        //否则设置初始化值
        return 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;
    }
    //可以客户端自己继承threadlocal重写该类,否则如果没有set就get的话会抛出空指针错误
    protected T initialValue() {
        return null;
    }

set,remove:

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

     public void remove() {
         ThreadLocalMap m = getMap(Thread.currentThread());
         if (m != null)
             m.remove(this);
     }

可以看出get set remove方法都是获得当前线程的threadlocalmap,然后调用他的get set remove方法。

五.ThreadLocalMap详解

entry:

        static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;
            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }

entry本身为一个弱引用,定义了一个类型为object的value

1.1为什么使用弱引用:
如果使用普通的key-value形式存储,那么就造成threadlocal对象的生命周期与线程一样长,因为总会存在thread->threadlocalmap->entry->threadlocal这样一条引用链存在,只要线程没被销毁,只要线程没有被销毁,那么节点在gc可达性分析中一直处于可达状态,无法被gc回收。而使用弱引用,如果threadlocal没有强引用可达,那么它不会活过下次gc,而该threadlocal对应的entry会失效,这位threadlocalmap垃圾清理提供了便利。

1.2类的成员变量和相应方法:

/**
         * 初始容量16,容量必须为2的幂
         */
        private static final int INITIAL_CAPACITY = 16;

        /**
         * 内部其实是entry类型的数组
         */
        private Entry[] table;

        /**
         * 表中entry的个数
         */
        private int size = 0;

        /**
         * 装载因子
         */
        private int threshold; // Default to 0

        /**
         * 最差的装载因子为2/3
         */
        private void setThreshold(int len) {
            threshold = len * 2 / 3;
        }

threadlocalmap为使用线性探测的哈希表,nextIndex和prevIndex为获取表前一个元素和后一个元素的方法:

        /**
         * Increment i modulo len.
         */
        private static int nextIndex(int i, int len) {
            return ((i + 1 < len) ? i + 1 : 0);
        }

        /**
         * Decrement i modulo len.
         */
        private static int prevIndex(int i, int len) {
            return ((i - 1 >= 0) ? i - 1 : len - 1);
        }

由此可见threadlocalmap逻辑上来说为一个环形。

set方法:

 private void set(ThreadLocal<?> key, Object value) {

            Entry[] tab = table;
            int len = tab.length;
            //key所在的槽位,使用该key的哈希值与上tab表的长度减一,相当于用len对threadLocalHashCode取模
            int i = key.threadLocalHashCode & (len-1);
            //若槽不为空,线性探测
            for (Entry e = tab[i];
                 e != null;
                 e = tab[i = nextIndex(i, len)]) {
                ThreadLocal<?> k = e.get();
                //如果槽中entry对应的threadlocal对象与参数所传入对象相同,则直接替换
                if (k == key) {
                    e.value = value;
                    return;
                }
                //若槽中entry弱引用指向的threadlocal对象已经被gc回收,则该entry失效,替换
                if (k == null) {
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }
            //否则new一个entry
            tab[i] = new Entry(key, value);
            int sz = ++size;
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();
        }

未完待续

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

推荐阅读更多精彩内容