一个故事讲明白线程的私家领地:ThreadLocal

张大胖上午遇到了一个棘手的问题,他在一个AccountService中写了一段类似这样的代码:

 

Context ctx = new Context();
ctx.setTrackerID(.....)

 

然后这个AccountService 调用了其他Java类,不知道经过了多少层调用以后,最终来到了一个叫做AccountUtil的地方,在这个类中需要使用Context中的trackerID来做点儿事情:

 

 

很明显,这个AccountUtil没有办法拿到Context对象, 怎么办?

 

张大胖想到,要不把Context对象一层层地传递下去,这样AccountUtil不就可以得到了吗?

 

 

可是这么做改动量太大!涉及到的每一层函数调用都得改动,有很多类都不属于自己的小组管理,还得和别人协调。 

 

更要命的是有些类根本就没有源码,想改都改不了。

 

这也难不住我,张大胖想:可以把那个set/get TrackerID的方法改成静态(static)的,这样不管跨多少层调用都没有问题!

 

public class Context{
   public static String getTrackerID(){
       ......
   }
   public static void setTrackerID(String id){
       ......
   }
}

 

 

这样就不用一层层地传递了,Perfect!

 

张大胖得意洋洋地把代码提交给Bill做Review。 

 

Bill看了一眼就指出了致命的问题: 多线程并发的时候出错!

 

张大胖恨不得找个地缝钻进去:又栽在多线程上面了,这次犯的还是低级错误!

 

线程1调用了Context.setTrackerID(), 线程2 也调用了Context.setTrackerID(),数据互相覆盖,不出乱子才怪。

 

张大胖感慨地说:“像我这样中情况,需要在某处设置一个值,然后经过重重方法调用,到了另外一处把这个值取出来,又要线程安全,实在是不好办啊, 对了,我能不能把这个值就放到线程中? 让线程携带着这个值到处跑,这样我无论在任何地方都可以轻松获得了!”

 

Bill说:“有啊,每个线程都有一个私家领地! 在Thread这个类中有个专门的数据结构,你可以放入你的TrackerID,然后到任何地方都可以把这个TrackerID给取出来。”

 

“这么好? ” 

 

张大胖打开JDK中的Thread类,仔细查看,果然在其中有个叫做threadLocals的变量,还是个Map类型 , 但是在Thread类中却没有对这个变量操作的方法。 

 

看到张大胖的疑惑,Bill说:“也许你注意到了,这个变量不是通过Thread的访问的,对他的访问委托给了ThreadLocal这个类。

 

“那我怎么使用它?”

 

“非常简单, 你可以轻松创建一个ThreadLocal类的实例:

 

ThreadLocal<String> threadLocalA= new ThreadLocal<String>();

线程1: threadLocalA.set("1234");
线程2: threadLocalA.set("5678");

 

像‘1234’, ‘5678’这些值都会放到自己所属的线程对象中。”

 

 

“等你使用的时候,可以这么办:”

 

线程1: threadLocalA.get()  --> "1234"
线程2: threadLocalA.get() --> "5678"

 

“明白了,相当于把各自的数据放入到了各自Thread这个对象中去了,每个线程的值自然就区分开了。 可是我不明白的是为什么那个数据结构是个map 呢?”

 

“你想想,假设你创建了另外一个threadLocalB:”

 

ThreadLocal<Integer> threadLocalB = new ThreadLocal<Integer>();

线程1: threadLocalB.set(30);
线程2: threadLocalB.set(40);

 

那线程对象的Map就起到作用了:

 

 

“明白了,这个私家领地还真是好用,我现在就把我那个Context给改了,让它使用ThreadLocal:”

public class Context {
   private static final ThreadLocal<String> mThreadLocal
       = new ThreadLocal<String>();

   public static void setTrackerID(String id) {
       mThreadLocal.set(id);
   }  
   public static String getTrackerID() {
       return mThreadLocal.get();
   }  

}

 

小结:

 

ThreadLocal这个名字起得有点让人误解, 很容易让人认为是“本地线程”, 其实是用来维护本线程的变量。 对照着上面的原理讲解,我想大家可以自行去看ThreadLocal的源码,轻松理解。

 

ThreadLocal 并不仅仅是Java中的概念,其他语言例如Python,C#中也有,作用类似。

 

ThreadLocal在日常工作中用得不多,但是在框架(如Spring)中是个基础性的技术,在事务管理,AOP等领域都能找到

posted on 2020-01-07 11:14  二云  阅读(214)  评论(0编辑  收藏  举报

导航