您的位置:首页 > 其它

彻底理解ThreadLocal一

2014-01-08 17:17 253 查看
synchronized这类线程同步的机制可以解决多线程并发问题,在这种解决方案下,多个线程访问到的,都是同一份变量的内容。为了防止在多线程访问的过程中,可能会出现的并发错误。不得不对多个线程的访问进行同步,这样也就意味着,多个线程必须先后对变量的值进行访问或者修改,这是一种以延长访问时间来换取线程安全性的策略。

而ThreadLocal类为每一个线程都维护了自己独有的变量拷贝(其实每个线程都new一个变量对象)。每个线程都拥有了自己独立的一个变量,竞争条件被彻底消除了,那就没有任何必要对这些线程进行同步,它们也能最大限度的由CPU调度,并发执行。并且由于每个线程在访问该变量时,读取和修改的,都是自己独有的那一份变量拷贝,变量被彻底封闭在每个访问的线程中,并发错误出现的可能也完全消除了。对比前一种方案,这是一种以空间来换取线程安全性的策略。

来看一个运用ThreadLocal来实现数据库连接Connection对象线程隔离的例子。

01
import
java.sql.Connection;
02
import
java.sql.DriverManager;
03
import
java.sql.SQLException;
04
05
public
class
ConnectionManager {
06
07
private
static
ThreadLocal<Connection> connectionHolder =
new
ThreadLocal<Connection>() {
08
@Override
09
protected
Connection initialValue() {
10
Connection conn =
null
;
11
try
{
12
conn = DriverManager.getConnection(
13
"jdbc:mysql://localhost:3306/test"
,
"username"
,
14
"password"
);
15
}
catch
(SQLException e) {
16
e.printStackTrace();
17
}
18
return
conn;
19
}
20
};
21
22
public
static
Connection getConnection() {
23
return
connectionHolder.get();
24
}
25
26
public
static
void
setConnection(Connection conn) {
27
connectionHolder.set(conn);
28
}
29
}
通过调用ConnectionManager.getConnection()方法,每个线程获取到的,都是和当前线程绑定的那个Connection对象,第一次获取时,是通过initialValue()方法的返回值来设置值的。通过ConnectionManager.setConnection(Connection conn)方法设置的Connection对象,也只会和当前线程绑定。这样就实现了Connection对象在多个线程中的完全隔离。在Spring容器中管理多线程环境下的Connection对象时,采用的思路和以上代码非常相似。

知其所以然

那么到底ThreadLocal类是如何实现这种“为每个线程提供不同的变量拷贝”的呢?先来看一下ThreadLocal的set()方法的源码是如何实现的:

01
/**
02
 
* Sets the current thread's copy of this thread-local variable
03
 
* to the specified value.  Most subclasses will have no need to
04
 
* override this method,relying solely on the {@link #initialValue}
05
 
* method to set the values of thread-locals.
06
 
*
07
 
* @param value the value to be stored in the current thread's copy of
08
 
*this thread-local.
09
 
*/
10
public
void
set(T value) {
11
Thread t = Thread.currentThread();
12
ThreadLocalMap map = getMap(t);
13
if
(map !=
null
)
14
map.set(
this
,value);
15
else
16
createMap(t,value);
17
}
没有什么魔法,在这个方法内部我们看到,首先通过getMap(Thread t)方法获取一个和当前线程相关的ThreadLocalMap,然后将变量的值设置到这个ThreadLocalMap对象中,当然如果获取到的ThreadLocalMap对象为空,就通过createMap方法创建。

线程隔离的秘密,就在于ThreadLocalMap这个类。ThreadLocalMap是ThreadLocal类的一个静态内部类,它实现了键值对的设置和获取(对比Map对象来理解),每个线程中都有一个独立的ThreadLocalMap副本,它所存储的值,只能被当前线程读取和修改。ThreadLocal类通过操作每一个线程特有的ThreadLocalMap副本,从而实现了变量访问在不同线程中的隔离。因为每个线程的变量都是自己特有的,完全不会有并发错误。还有一点就是,ThreadLocalMap存储的键值对中的键是this对象指向的ThreadLocal对象,而值就是你所设置的对象了。

为了加深理解,我们接着看上面代码中出现的getMap和createMap方法的实现:

1
ThreadLocalMap getMap(Thread t) {
2
return
t.threadLocals;
3
}
1
void
createMap(Thread t,T firstValue) {
2
t.threadLocals =
new
ThreadLocalMap(
this
,firstValue);
3
}
代码已经说的非常直白,就是获取和设置Thread内的一个叫threadLocals的变量,而这个变量的类型就是ThreadLocalMap,这样进一步验证了上文中的观点:每个线程都有自己独立的ThreadLocalMap对象。打开java.lang.Thread类的源代码,我们能得到更直观的证明:

1
/* ThreadLocal values pertaining to this thread. This map is maintained
2
 
* by the ThreadLocal class. */
3
ThreadLocal.ThreadLocalMap threadLocals =
null
;
那么接下来再看一下ThreadLocal类中的get()方法,代码是这么说的:

01
/**
02
 
* Returns the value in the current thread's copy of this
03
 
* thread-local variable.  If the variable has no value for the
04
 
* current thread,it is first initialized to the value returned
05
 
* by an invocation of the {@link #initialValue}method.
06
 
*
07
 
* @return the current thread's value of this thread-local
08
 
*/
09
public
T get() {
10
Thread t = Thread.currentThread();
11
ThreadLocalMap map = getMap(t);
12
if
(map !=
null
) {
13
ThreadLocalMap.Entry e = map.getEntry(
this
);
14
if
(e !=
null
)
15
return
(T)e.value;
16
}
17
return
setInitialValue();
18
}
19
20
/**
21
 
* Variant of set() to establish initialValue. Used instead
22
 
* of set() in case user has overridden the set() method.
23
 
*
24
 
* @return the initial value
25
 
*/
26
private
T setInitialValue() {
27
T value = initialValue();
28
Thread t = Thread.currentThread();
29
ThreadLocalMap map = getMap(t);
30
if
(map !=
null
)
31
map.set(
this
,value);
32
else
33
createMap(t,value);
34
return
value;
35
}
这两个方法的代码告诉我们,在获取和当前线程绑定的值时,ThreadLocalMap对象是以this指向的ThreadLocal对象为键进行查找的,这当然和前面set()方法的代码是相呼应的。

进一步地,我们可以创建不同的ThreadLocal实例来实现多个变量在不同线程间的访问隔离,为什么可以这么做?因为不同的ThreadLocal对象作为不同键,当然也可以在线程的ThreadLocalMap对象中设置不同的值了。通过ThreadLocal对象,在多线程中共享一个值和多个值的区别,就像你在一个HashMap对象中存储一个键值对和多个键值对一样,仅此而已。

设置到这些线程中的隔离变量,会不会导致内存泄漏呢?ThreadLocalMap对象保存在Thread对象中,当某个线程终止后,存储在其中的线程隔离的变量,也将作为Thread实例的垃圾被回收掉,所以完全不用担心内存泄漏的问题。在多个线程中隔离的变量,光荣的生,合理的死,真是圆满,不是么?

最后再提一句,ThreadLocal变量的这种隔离策略,也不是任何情况下都能使用的。如果多个线程并发访问的对象实例只允许,也只能创建那么一个,那就没有别的办法了,老老实实的使用同步机制来访问吧。

ThreadLocal为每一个线程提供一份拷贝,变量线程隔离后,成了互不相干的东西。但是这不正是实现了这个变量的线程安全么?不管你每个变量如何折腾,都不会有线程安全问题,也无需任何额外的同步。回到文章最后,还是要看看这个变量“是否允许这样做”。如果这个变量一定是要求全局唯一,那么你就只能通过同步机制,在多个线程间共享它了

ThreadLocalMap 是Thread 对象的属性,会随着Thread 对象的回收而回收,但是很多容器都有线程池,我们很多时候还需要手动移除!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: