一个本来应该线程安全的类,里面有一个线程不安全的变量,这样这个类也就线程不安全了,那应该怎么办呢?我们如果能够把这个变量和每个线程绑定,也就是每一个线程拥有这个变量的副本,那么整个对象就成为线程安全的了。一个解决方案就是使用一个Map,key对应于当前的线程,value对应于那个变量,这样我们就可以轻易的获取到当前线程的那个变量的副本了, ThreadLocal 就是这个东东。
我们不妨写写大致的代码:
- public
class ThreadLocal{ - private
Map values = Collections.synchronizedMap(new HashMap()); - public
Object get(){ - Thread
curThread = Thread.currentThread(); - Object
o = values.get(curThread); - if
(o == null && !values.containsKey(curThread)){ - o
= initialValue(); - values.put(curThread,
o); - }
- return
o; - }
-
- public
void set(Object newValue){ - values.put(Thread.currentThread(),
newValue); - }
-
- public
Object initialValue(){ - return
null; - }
- }
当然java的 ThreadLocal 实现的总体思路也大致如此。
我们看看jdk提供的api文档:
T get()
protected
void remove()
void set(T value)
我们看jdk文档提供了一个例子:
- import
java.util.concurrent.atomic.AtomicInteger; -
-
public class UniqueThreadIdGenerator { -
-
private static final AtomicInteger uniqueId = new AtomicInteger(0); -
-
private static final ThreadLocal < Integer > uniqueNum = -
new ThreadLocal < Integer > () { -
@Override protected Integer initialValue() { -
return uniqueId.getAndIncrement(); -
} -
}; -
-
public static int getCurrentThreadId() { -
return uniqueId.get();//应该是return uniqueNum.get(); -
} -
} // UniqueThreadIdGenerator
这个例子我看的时候没看懂,其实是有错误的 return uniqueId.get();,应该是return uniqueNum.get();我用的是中文翻译过来的jdk api 1.6.0,不知道
大家的jdk帮助文档有没有这个问题.
我们看看Hibernate官方文档提供的一个通过 ThreadLocal 维护Session的例子:
-
public class HibernateUtil { -
private static final SessionFactory sessionFactory; -
static { -
try { -
sessionFactory = new Configuration().configure() -
.buildSessionFactory(); -
-
} catch (Throwable ex) { -
ex.printStackTrace(); -
throw new ExceptionInInitializerEr ror(ex); -
-
} -
-
} -
public static final ThreadLocal tLocalsess = new ThreadLocal(); -
// 取得session -
public static Session currentSession() { -
Session session = (Session) tLocalsess.get(); -
// 打开一个新的session,如果当前的不可用. -
try { -
if (session == null || !session.isOpen()) { -
session = openSession(); -
tLocalsess.set(session); -
} -
-
} catch (HibernateException e) { -
// 抛出HibernateException异常 -
e.printStackTrace(); -
-
} -
return session; -
} -
public static void closeSession() { -
Session session = (Session) tLocalsess.get(); -
tLocalsess.set(null); -
try { -
if (session != null && session.isOpen()) { -
session.close(); -
} -
} catch (HibernateException e) { -
//抛出 InfrastructureException异常 -
} -
} - //other
code - }
- 这段代码借助threadLocal,每一个线程保存一个session实例,从而避免线程内的频繁创建和销毁session.
其它适合使用 ThreadLocal 但用池却不能成为很好的替代技术的应用程序包括存储或累积每线程 上下文信息以备稍后检索之用这样的应用程序。例如,假设您想创建一个用于管理多线程应用程序调试信息的工具 。您可以用 DebugLogger 类作为线程局部容器来累积调试信息。在一个工作单元的开头,您清空容器,而当一个错误出现时,您查询该容器以检索这个工作单元迄今为止生成的所有调试信息。
用 ThreadLocal 管理每线程调试日志
- public
class DebugLogger { -
private static class ThreadLocalList extends ThreadLocal { -
public Object initialValue() { -
return new ArrayList(); -
} -
public List getList() { -
return (List) super.get(); -
} -
} -
private ThreadLocalList list = new ThreadLocalList(); -
private static String[] stringArray = new String[0]; -
public void clear() { -
list.getList().clear(); -
} -
public void put(String text) { -
list.getList().add(text); -
} -
public String[] get() { -
return list.getList().toArray(stringArray); -
} - }
ThreadLocal 在基于 servlet 的应用程序或工作单元是一个整体请求的任何多线程应用程序服务器中也是很有用的,因为在处理请求的整个过程中将要用到单个线程。您可以通过每线程单子技术用 ThreadLocal 变量来存储各种每请求(per-request)上下文信息。
下面是对ThreadLocal 类的源码分析下:
void set(Object value):设置当前线程的线程局部变量的值。
public Object get() :该方法返回当前线程所对应的线程局部变量。 (同一个线程获取同一个变量的引用)
public void remove() :将当前线程局部变量的值删除,目的是为了减少内存的占用,该方法是JDK 5.0新增的方法。需要指出的是,当线程结束后,对应该线程的局部变量将自动被垃圾回收,所以显式调用该方法清除线程的局部变量并不是必须的操作,但它可以加快内存回收的速度。
protected Object initialValue():返回该线程局部变量的初始值,该方法是一个protected的方法,显然是为了让子类覆盖而设计的(在上面的例子也可以看到,其实就是一个默认值的设置)。这个方法是一个延迟调用方法,在线程第1次调用 get()或set(Object)时才执行,并且仅执行1次。ThreadLocal中的缺省实现直接返回一个null。
1、void set(Object value)方法分析
- public void set(T value) {
- Thread t = Thread.currentThread();
- //查看当前线程中的ThreadLocalMap 是否存在。
- ThreadLocalMap map = getMap(t);
- //存在的话就把当前线程设置进ThreadLocalMap中。
- if (map != null)
- map.set(this, value);
- else
- //否则生成一个ThreadLocalMap 对象,并把值赋到当前线程的ThreadLocalMap中
- createMap(t, value);
- }
- //-------------------------------解析的代码分割线-------------------------------
-
- //返回当前线程的ThreadLocalMap
- ThreadLocalMap getMap(Thread t) {
- return t.threadLocals;
- }
- //创建新的ThreadLocalMap赋值到当前线程
- void createMap(Thread t, T firstValue) {
- t.threadLocals = new ThreadLocalMap(this, firstValue);
- }
下面说下关于ThreadLocalMap 类,这个类有点类似于WeakHashMap(这个类对于解决某些问题提供了很好的性能上的方案,有空再继续深入了解。),虽然ThreadLocalMap没有继承Map的,但它具有Map的一些特性,有关这些特性就不说了,主要说说它们的一个很大的共同点就是,它们的key 都是弱引用的对象,因此在该键没有被引用时,将会被JVM收集。(有关弱引用的解析,请参考我的博客上的:http://zhxing.iteye.com/admin/blogs/571007)
ThreadLocalMap是ThreadLocal的一个嵌套类,它的一些源码如下:
- static class ThreadLocalMap {
-
- /**
- * The entries in this hash map extend WeakReference, using
- * its main ref field as the key (which is always a
- * ThreadLocal object). Note that null keys (i.e. entry.get()
- * == null) mean that the key is no longer referenced, so the
- * entry can be expunged from table. Such entries are referred to
- * as "stale entries" in the code that follows.
- */
- static class Entry extends WeakReference<ThreadLocal> {
- /** The value associated with this ThreadLocal. */
- Object value;
-
- Entry(ThreadLocal k, Object v) {
- super(k);
- value = v;
- }
- }
2、 public Object get() 方法分析
- public T get() {
- Thread t = Thread.currentThread();
- ThreadLocalMap map = getMap(t);
- if (map != null) {
- //这个是传入当前的ThreadLocal 对象引用获取所对应的值
- ThreadLocalMap.Entry e = map.getEntry(this);
- if (e != null)
- return (T)e.value;
- }
- //如果map 对象为空,说明是第一次调用这个方法,之前也没有调用set()方法
- return setInitialValue();
- }
- //-------------------------------解析的代码分割线-------------------------------
- //这个方法很简单就是调用initialValue()获取初始值(默认值),而且要赋值在当前线程的map中(相当于调用了set()方法)
- private T setInitialValue() {
- T value = initialValue();
- Thread t = Thread.currentThread();
- ThreadLocalMap map = getMap(t);
- if (map != null)
- map.set(this, value);
- else
- createMap(t, value);
- return value;
- }
- //ThreadLocal 的默认实现是返回null 的,所以如果在多个线程中使用时,如果需要用到默认值,那子类就应该覆盖该值
- protected T initialValue() {
- return null;
- }
3、public void remove() 方法
- //这个方法没什么可说的,就是删除掉当前线程中的变量(释放内存)
- public void remove() {
- ThreadLocalMap m = getMap(Thread.currentThread());
- if (m != null)
- m.remove(this);
- }
总结
ThreadLocal 可以实现在同一个线程内共享一个变量,不同的线程不能共享,只能得到该变量的一个副本(就是一个默认的初始化值)。
在论坛中也有人提到ThreadLocal和synchronized 的问题,我觉得它们都是解决线程的两个不同的方案。ThreadLocal解决的是同一个线程内的资源共享问题,而synchronized 解决的是多个线程间的资源共享问题。
有关的讨论在:http://www.iteye.com/topic/179040
有关这个类的应用,可以参考http://zhxing.iteye.com/admin/blogs/306070
3. ThreadLocal在JavaEE中的应用
ThreadLocal在Hibernate中得到充分的使用,特别是对于事物的处理上。比如,
getCurrentSession非常好,不需要我们自己写ThreadLocal只需要在hibernate.cfg的配置文件中声明一下便可获得ThreadLocal的好处是便于我们划分我们的程序的层次与封装,带也带来了一定的性能问题。
特别是“如果你使用的是getCurrentSession,那么就算你是一个简单的select语句,也必须包含在事务块中”。碰到这种情况,可以采取如下手段:
- 一个service方法中只有单个dao操作且此操作是一个select类的操作,请使用openSession,并且即时在finally块中关闭它;
- 如果一个service方法中涉及到多个dao操作,请一定使用getCurrentSession;
- 如果一个service方法中混合着select操作,delete, update, insert操作。请按照下述原则:
a. 将属于select的操作,单独做成一个dao方法,该dao使用openSession并且在finally块中及时关闭session,
该dao只需要返回一个java的object如:List<Student>即可,如果出错将exception抛回给调用它的service
方法。
b. 对于其它的delete, insert, update的dao操作,请使用getCurrentSession。
c. 忌讳,把select类的操作放在“事务”中;
4. openSession与getCurrentSession的区别
a. openSession一旦被调用,必须且一定要在finally块中close,要不然你就等着out of memory吧;
b. 如果你使用的是getCurrentSession,那么你不能在finally块中调用”session.close()”,不行你可以在finally块中用try-catch把session.close();包起来,然后在catch{}块中抛出这个exception,这个exception将会是:sessionhas been already closed。其原因是你用的是getCurrentSession,那么它在session.commit()或者是session.rollback()时就已经调用了一次session.close()了,因此你只要正确放置session.commit()与rollback()即可。另外,你可以在finally块中调用”ThreadLocalSessionContext.unbind(factory);”,以使得当前的事务结束时把session(即dbconnection)还回db connection pool中。