浅析ThreadLocal

一般有多个孩子的家庭,买玩具都得买多个。如果就买一个,嘿嘿就比较刺激了。这就是避免共享,给孩子每人一个玩具对应到我们Java中也就是每个线程都有自己的本地变量,咱们自己玩自己的,避免争抢,和谐相处使得线程安全。

Java就是通过ThreadLocal来实现线程本地存储的。

这思路也很清晰,就是每个线程要有自己的本地变量呗,那就Thread里面搞一个私有属性呗ThreadLocal.ThreadLocalMap threadLocals = null;
就是如下图所示的这个关系

image

ThreadLocal

简单的应用如下

    public class Demo {
        private static final ThreadLocal<Foo> fooLocal = new ThreadLocal<Foo>();
 
        public static Foo getFoo() {
            return fooLocal.get();
        }
 
        public static void setFoo(Foo foo) {
            fooLocal.set(foo);
        }
    }

再深入了解一下内部情况,ThreadLocalMapThreadLocal的内部静态类,它虽然叫Map但是和java.util.Map没有啥亲戚关系,只是它实现的功能像Map

    static class ThreadLocalMap {
        static class Entry extends WeakReference<ThreadLocal<?>> {
            Object value;
            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }
        private static final int INITIAL_CAPACITY = 16;
        private Entry[] table;

可以看到ThreadLocalMap里面有个Entry数组,只有数组没有像HashMap那样有链表,因此当hash冲突的之后,ThreadLocalMap采用线性探测的方式解决hash冲突。

线性探测,就是先根据初始keyhashcode值确定元素在table数组中的位置,如果这个位置上已经有其他key值的元素被占用,则利用固定的算法寻找一定步长的下个位置,依次直至找到能够存放的位置。在ThreadLocalMap步长是1。

用这种方式解决hash冲突的效率很低,因此要注意ThreadLocal的数量

        /**
         * 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);
        }

而且可以看到这个EntryThreadLocal的弱引用作为key。那为什么要搞成弱引用(只要发生了GC弱引用对象就会被回收)呢?

首先ThreadLocal内部没有存储任何的值,它的作用只是当我们的ThreadLocalMap的key,让线程可以拿到对应的value。当我们不需要用这个key的时候我们,我们把fooLocal=null这样强引用就没了。假设Entry里面也是强引用的话,那等于这个ThreadLocal实例还有个强引用在,那么我们想让GC回收fooLocal就回收不了了。那可能有人想,你弄成弱引用不是很危险啊,万一GC一下不是没了?别怕只要fooLocal这个强引用在这个ThreadLocal实例就不会回收的。(关于强软弱虚引用可以看我之前的文章四种引用方式的区别)

因此弄成弱引用,主要是让没用的ThreadLocal得以GC清除。

这里可能还有人问那key清除掉了,value咋办,这个Entry还在的呀。是的,当在使用线程池的情况下,由于线程的生命周期很长,某些大对象的key被移除了之后,value一直存在的就可能会导致内存泄漏。

不过java考虑到这点了。当调用get()、set()方法时会去找到那个key被干掉的entry然后干掉它。并且提供了remove()方法。虽然get()、set()会清理keynull的Entry,但是不是每次调用就会清理的,只有当get时候直接hash没中,或者set时候也是直接hash没中,开始线性探测时候,碰到key为null的才会清理。

        //get 方法
        private Entry getEntry(ThreadLocal<?> key) {
            int i = key.threadLocalHashCode & (table.length - 1);
            Entry e = table[i];
            if (e != null && e.get() == key)
                return e;                           //命中就直接返回
            else
                return getEntryAfterMiss(key, i, e); //直接没命中
        }
        private Entry getEntryAfterMiss(ThreadLocal<?> key, int i, Entry e) {
            Entry[] tab = table;
            int len = tab.length;
            while (e != null) { //开始探测了
                ThreadLocal<?> k = e.get();
                if (k == key)  //命中了就返回
                    return e;
                if (k == null)  //探测到key是null的就清理
                    expungeStaleEntry(i);
                else
                    i = nextIndex(i, len); //否则继续
                e = tab[i];
            }
            return null;
        }
        //set 方法
        private void set(ThreadLocal<?> key, Object value) {
            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)]) {
                ThreadLocal<?> k = e.get();
                if (k == key) {
                    e.value = value;    //如果已经有就替换原有的value
                    return;
                }
                if (k == null) {
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }
            tab[i] = new Entry(key, value);
            int sz = ++size;
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                rehash();
        }

因此,当不需要threadlocal的时候还是显示调用remove()方法较好。

结语

线程本地存储本质就是避免共享,在使用中注意内存泄露问题和hash碰撞问题即可。使用还是很广泛的像spring中事务就用到threadlocal


如有错误欢迎指正!

个人公众号:yes的练级攻略

有相关面试进阶(分布式、性能调优、经典书籍pdf)资料等待领取

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

推荐阅读更多精彩内容