ThreadLocal其实在很早以前就看到过,也看过一些介绍,但还是不甚了解。今天跟踪了一下源码,才觉得ThreadLocal非常巧妙。
现在网上对ThreadLocal的介绍也有很多了,我主要想聊聊ThreadLocal的代码设计。
作用
为什么要引入ThreadLocal呢?具体应用我在项目中并没有遇到过,只是觉得当我们创建一个线程的时候,我的主线程中的所有数据都是对子线程可见的。也就是如果子线程想要获取、更改什么数据,是会对主线程产生影响的。这些问题带来了一系列的问题,数据竞争(由此导致要数据同步)。那我们总希望有这样一种情景:主线程和子线程都有各自的隐私空间。即这部门数据只对自己可见。
OK,那我们想想如何实现这个需求呢?
仔细想一下,我们似乎没有一种有效的方法,除非新建一个子线程类,然后只将必要的数据传送给子线程。那如果使用内部类呢?我们很多时候在创建线程的时候可都采用的是内部类,特别是匿名内部类。这样的话是无论如何都不能阻止子线程访问主线程的数据。
设计
threadLocal的设计思路是,并不阻止子线程访问主线程的对象(也是无法阻止的),但是子线程和主线程存放数据的位置并不在一起。听起来是不是很神奇,同一个对象,但是存放数据的位置并不相同!!这样就可以产生如下结果:当主线程使用这个对象引用设置数据或者添加数据时,子线程完全感知不到,同理对子线程也是如此。尽管他们拿的是同一个引用。
这也是我觉得设计最优美的地方!
实现
OK,谈完思想了,还是得把思想落到实地。那它是如何实现的呢?,我们看看ThreadLocal的源码。先看set,再看get。
/**
* Sets the current thread's copy of this thread-local variable
* to the specified value. Most subclasses will have no need to
* override this method, relying solely on the {@link #initialValue}
* method to set the values of thread-locals.
*
* @param value the value to be stored in the current thread's copy of
* this thread-local.
*/
public void set(T value) {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
}
注释我看的不是很懂,还是直接看代码吧,首先,在set的时候先获取ThreadLocalMap,然后将当前对象和值存储在threadlocalmap中。这里,当前对象是指ThreadLocal对象。ThreadLocalMap中的entry实际为
/**
* Get the map associated with a ThreadLocal. Overridden in
* InheritableThreadLocal.
*
* @param t the current thread
* @return the map
*/
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
getMap的函数是返回当前线程下的threadLocals。
/* ThreadLocal values pertaining to this thread. This map is maintained
* by the ThreadLocal class. */
ThreadLocal.ThreadLocalMap threadLocals = null;
//这是Thread对象中对ThreadLocalMap的定义
现在我们应该大概清楚了,数据是存储在Thread对象的空间里,在使用的时候,实际上是通过当前线程的引用查找到当前线程的ThreadLocalMap,然后在用map来获取到ThreadLocal对象对应的value值。
put方法与此相似。
ThreadLocalMap
下面谈谈ThreadLocalMap的实现,这个数据结构的定义倒没有特别花哨的东西,它并没有实现Map接口,因为它并不需要,它也有一个内部的Entry类来存储<K,V>值。具体而言,K为ThreadLocal,V为Object
同时在ThreadLocalMap中使用一个数组来存储Entry引用。存储时使用ThreadLocal对象的哈希值(不是hashcode,但不重要)选择一个合适的数组项。然后将Entry引用添加进去。
我们注意到Entry类中并没有像链表一样的next属性,这就意味着当发生碰撞时,无法使用像HashMap一样的链表的形式来存储碰撞值。那么当碰撞时,ThreadLocalMap将依次从碰撞的Index项开始继续向前搜索(线性探测法),知道找到一个空项为止。当然了,在get时,也需要这样的一个循环。
总得来说,这种实现方式很美妙。