ThreadLocal缺点及解决方法

在前面的一篇文章曾经介绍过ThreadLocal,大体知道了ThreadLocal的应用场景和实现原理。

简单的一句话总结是每个Thread上都有一个threadLocals属性,它是一个ThreadLocalMap,里面存放着一个Entry数组,key是ThreadLocal类型的弱引用,value是对用的值。所有的操作都是基于这个ThreadLocalMap操作的。

但是它有一个局限性,就是不能在父子线程之间传递。 即在子线程中无法访问在父线程中设置的本地线程变量。 后来为了解决这个问题,引入了一个新的类InheritableThreadLocal

使用该方法后,子线程可以访问在创建子线程时父线程当时的本地线程变量,其实现原理就是在父线程创建子线程时将父线程当前存在的本地线程变量拷贝到子线程的本地线程变量中。

public class InheritableThreadLocal<T> extends ThreadLocal<T> {
    protected T childValue(T parentValue) {
        return parentValue;
    }
    ThreadLocalMap getMap(Thread t) {
       return t.inheritableThreadLocals;
    }
    void createMap(Thread t, T firstValue) {
        t.inheritableThreadLocals = new ThreadLocalMap(this, firstValue);
    }
}

从上面的结构中可以看出,它主要是重写了getMapcreateMap方法。

Thread类中有两个重要的变量

    ThreadLocal.ThreadLocalMap threadLocals = null;
    ThreadLocal.ThreadLocalMap inheritableThreadLocals = null;
# init()方法片段
Thread parent = currentThread(); 
.....
if (inheritThreadLocals && parent.inheritableThreadLocals != null)
            this.inheritableThreadLocals =                ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);

子线程时通过在父线程通过调用new Thread()方法来创建子线程,Thread#init方法在Thread的构造方法中被调用。

主要是先获取当前线程对象,即待创建的线程的父线程
如果父线程的inheritableThreadLocals不为空,并且inheritThreadLocals为true(默认为true),则使用父线程的ingerit本地变量的值来创建子线程的inheritableThreadLocals结构,即将父线程中的本地变量复制到子线程中。

private Entry[] table;

private ThreadLocalMap(ThreadLocalMap parentMap) {
            Entry[] parentTable = parentMap.table;
            int len = parentTable.length;
            setThreshold(len);
            table = new Entry[len];

            for (int j = 0; j < len; j++) {
                Entry e = parentTable[j];
                if (e != null) {
                    @SuppressWarnings("unchecked")
                    ThreadLocal<Object> key = (ThreadLocal<Object>) e.get();
                    if (key != null) {
                        Object value = key.childValue(e.value);
                        Entry c = new Entry(key, value);
                        int h = key.threadLocalHashCode & (len - 1);
                        while (table[h] != null)
                            h = nextIndex(h, len);
                        table[h] = c;
                        size++;
                    }
                }
            }
        }

子线程默认拷贝父线程的方式是浅拷贝,如果需要使用深拷贝,如果需要使用深拷贝,需要使用自定义ThreadLocal,继承InheritThreadLocal并重写childValue方法。

而它的实现原理主要是在创建子线程时将父类线程中的本地变量值parent.inheritableThreadLocals复制到子线程,即复制的机会在创建子线程时。

但是它也有一种缺陷,就是在使用线程池的情况下,因为线程池是复用线程,不会重复创建,而上述的inheritableThreadLocals是在创建子线程时才会将父线程的值复制到子线程,但是在线程池中不会重复创建,所以多次使用后,仍然记录的是第一次提交任务时的外部线程的值,造成了数据的错误。

那如何解决这种现象呢?

只需在用户线程向线程池提交任务时复制父线程的上下文环境,就可以实现本地变量在线程池调用的透传。 基于这个思想,阿里提出了TransmittableThreadLocal类。
【导入图片】
在提交任务时的Runable或者Callable必须封装成TtlRunable或者TtlCallableTransmittableThreadLocal覆盖实现了ThreadLocal的set、get、remove,实际存储inheritableThreadLocal值的工作还是inheritableThreadLocal父类完成,TransmittableThreadLocal只是为每个使用它的Thread单独记录一份存储了哪些TransmittableThreadLocal对象。

public final void set(T value) {
  super.set(value);
  if (null == value) removeValue();
  else addValue();
}

首先它会调用父类的InheritableThreadLocal的set方法,将value加入到Thread对象的inheritableThreadLocals变量中。
如果value为null,则调用removeValue()方法,否则调用addValue方法。

private void addValue() {
    if (!holder.get().containsKey(this)) {    // @1
        holder.get().put(this, null); // WeakHashMap supports null value.
    }
}
private void removeValue() {
    holder.get().remove(this);
}

调用addValue方法,会将当前ThreadLocal存储到TransmittableThreadLocal的全局静态变量hodler。所有和Thread绑定的所有TransmittableThreadLocal对象都保存在这个holder中,holder只是为了记录当前Thread绑定了哪些TransmittableThreadLocal对象。

下面具体讲一下TtlRunable的原理。

private TtlRunnable(@Nonnull Runnable runnable, boolean releaseTtlValueReferenceAfterRun) {
    this.capturedRef = new AtomicReference<Object>(capture());   
    this.runnable = runnable;
    this.releaseTtlValueReferenceAfterRun = releaseTtlValueReferenceAfterRun;
}

先创建Map容器,用来存储父线程的本地线程变量,键为在父线程执行过程中的TransmittableLocal线程;将线程中的值存放在里面。默认是浅拷贝,需要深拷贝的话,要重写copyValue方法。

public void run() {
     Object captured = capturedRef.get();             
     if (captured == null || releaseTtlValueReferenceAfterRun && !capturedRef.compareAndSet(captured, null)) {
         throw new IllegalStateException("TTL value reference is released after run!");
   }
     Object backup = replay(captured);            
     try {
         runnable.run();                                           
    } finally {
        restore(backup);                                       
    }
}

TtlRunable是实现于Runable,所以线程池执行的是TtlRunable,但是在TtlRunnable run方法中国会执行Runable run方法。

TtlRunable构造方法中,调用了capture()获取当前线程中所有的上下文,并存储在AtomicReference中。

当线程执行时,调用TtlRunable run方法, TtlRunable会从AtomicReference中获取出调用线程中的上下文,并把上下文利用replay方法把上下文复制到当前线程,并把上下文备份。

当线程执行完,调用restore把备份的上下文传入,恢复备份的上下文传入,把后面新增的上下文删除,并重新把上下文复制到当前线程。本次执行并不会污染线程池中线程原先的上下文环境。

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

推荐阅读更多精彩内容