天天看點

什麼是ThreadLocal

顧名思義它是local variable(線程局部變量)。它的功用非常簡單,就是為每一個使用該變量的線程都提供一個變量值的副本,使每一個線程都可以獨立地改變自己的副本,而不會和其它線程的副本沖突。從線程的角度看,就好像每一個線程都完全擁有該變量。

使用場景

  1. To keep state with a thread (user-id, transaction-id, logging-id)
  2. 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(); 
     //open a new session,if this session has none 
  if(s == null){ 
     s = sessionFactory.openSession(); 
     session.set(s); 
  } 
  return s; 
}       

我們逐行分析 

1。 初始化一個ThreadLocal對象,ThreadLocal有三個成員方法 get()、set()、initialvalue()。 

    如果不初始化initialvalue,則initialvalue傳回null。 

3。session的get根據目前線程傳回其對應的線程内部變量,也就是我們需要的net.sf.hibernate.Session(相當于對應每個資料庫連接配接).多線程情況下共享資料庫連結是不安全的。ThreadLocal保證了每個線程都有自己的s(資料庫連接配接)。 

5。如果是該線程初次通路,自然,s(資料庫連接配接)會是null,接着建立一個Session,具體就是行6。 

6。建立一個資料庫連接配接執行個體 s 

7。儲存該資料庫連接配接s到ThreadLocal中。 

8。如果目前線程已經通路過資料庫了,則從session中get()就可以擷取該線程上次擷取過的連接配接執行個體。

使用方法二

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

1 public class JDBCContext{
 2  private static Logger logger = Logger.getLogger(JDBCContext.class);
 3  private DataSource ds;
 4  protected Connection connection;
 5  private boolean isValid = true;
 6  private static ThreadLocal jdbcContext;
 7  
 8  private JDBCContext(DataSource ds){
 9   this.ds = ds;
10   createConnection();  
11  }
12  public static JDBCContext getJdbcContext(javax.sql.DataSource ds)
13  {  
14   if(jdbcContext==null)jdbcContext=new JDBCContextThreadLocal(ds);
15   JDBCContext context = (JDBCContext) jdbcContext.get();
16   if (context == null) {
17    context = new JDBCContext(ds);
18   }
19   return context;
20  }
21 
22  private static class JDBCContextThreadLocal extends ThreadLocal {
23   public javax.sql.DataSource ds;
24   public JDBCContextThreadLocal(javax.sql.DataSource ds)
25   {
26    this.ds=ds;
27   }
28   protected synchronized Object initialValue() {
29    return new JDBCContext(ds);
30   }
31  }
32 }      

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