ThreadLocal的那些事兒......

什麼是ThreadLocal?

顧名思義它是local variable(線程局部變量)。

它的功用非常簡單,就是爲每一個使用該變量的線程都提供一個變量值的副本,是每一個線程都可以獨立地改變自己的副本,而不會和其它線程的副本衝突。從線程的角度看,就好像每一個線程都完全擁有該變量。

使用場景:

To keep state with a thread (user-id, transaction-id, logging-id)
To cache objects which you need frequently


ThreadLocal類

主要由四個方法組成initialValue(),get(),set(T),remove(),其中值得注意的是initialValue(),該方法是一個protected的方法,顯然是爲了子類重寫而特意實現的。該方法返回當前線程在該線程局部變量的初始值,這個方法是一個延遲調用方法,在一個線程第1次調用get()或者set(Object)時才執行,並且僅執行1次。ThreadLocal中的確實現直接返回一個null。

ThreadLocal的原理

ThreadLocal是如何做到爲每一個線程維護變量的副本的呢?其實實現的思路很簡單,在ThreadLocal類中有一個Map,用於存儲每一個線程的變量的副本。比如下面的示例實現:

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;
}
}

ThreadLocal 的使用

使用方法一:

Hibernate中關於使用ThreadLocal管理多線程訪問的部分。具體代碼如下:

public static final ThreadLocal session = new ThreadLocal();
       public static Session currentSession() { 
            Session s = (Session)session.get(); 
            if(s == null){ 
                 s = sessionFactory.openSession(); 
                 session.set(s); 
            }  
           return s;
}

我們逐行分析
1.初始化一個ThreadLocal對象,ThreadLocal有三個成員方法 get()、set()、initialvalue()。如果不初始化initialvalue,則initialvalue返回null。
2.session的get根據當前線程返回其對應的線程內部變量,也就是我們需要的net.sf.hibernate.Session(相當於對應每個數據庫連接).多線程情況下共享數據庫鏈接是不安全的。ThreadLocal保證了每個線程都有自己的s(數據庫連接)。
3.如果是該線程初次訪問,自然,s(數據庫連接)會是null,接着創建一個Session。
4.創建一個數據庫連接實例 s
5.保存該數據庫連接s到ThreadLocal中。
6.如果當前線程已經訪問過數據庫了,則從session中get()就可以獲取該線程上次獲取過的連接實例。

使用方法二

當要給線程初始化一個特殊值時,需要自己實現ThreadLocal的子類並重寫該方法,通常使用一個內部匿名類對ThreadLocal進行子類化,EasyDBO中創建jdbc連接上下文就是這樣做的:

public class JDBCContext{
       private static Logger logger = Logger.getLogger(JDBCContext.class);
       private DataSource ds;
       protected Connection connection;
       private boolean isValid = true;
       private static ThreadLocal jdbcContext;
       private JDBCContext(DataSource ds){
             this.ds = ds;
             createConnection(); 
       }
      public static JDBCContext getJdbcContext(javax.sql.DataSource ds){ 
             if(jdbcContext==null)jdbcContext=new JDBCContextThreadLocal(ds);
             JDBCContext context = (JDBCContext) jdbcContext.get();
             if (context == null) {
                   context = new JDBCContext(ds);
             }
             return context;
       }

       private static class JDBCContextThreadLocal extends ThreadLocal {
            public javax.sql.DataSource ds;
            public JDBCContextThreadLocal(javax.sql.DataSource ds){
                   this.ds=ds;
            }
            protected synchronized Object initialValue() {
                  return new JDBCContext(ds);
            }
        }
  }

使用單例模式,不同的線程調用getJdbcContext()獲得自己的jdbcContext,都是通過JDBCContextThreadLocal 內置子類來獲得JDBCContext對象的線程局部變量。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章