ThreadLocal 的開發(fā)應(yīng)用
作者：KerryWu
來源：SegmentFault 思否社區(qū)
ThreadLocal是線程私有的局部變量存儲容器，可以理解成每個線程都有自己專屬的存儲容器，用來存儲線程私有變量。ThreadLocal 在日常開發(fā)框架中應(yīng)用廣泛，但用不好也會出現(xiàn)各種問題，本文就此講解一下。
1. 應(yīng)用場景
ThreadLocal 的常見應(yīng)用場景有兩種：
多線程并發(fā)場景中，用來保障線程安全。 處理較為復(fù)雜的業(yè)務(wù)時，使用ThreadLocal代替參數(shù)的顯示傳遞。
1.1. 保障線程安全
1.2. 顯示傳遞參數(shù)
public class UserProfileThread {
private static ThreadLocal<UserProfile> USER_PROFILE_TL =new ThreadLocal<>();
public static void setUserProfile(UserProfile userProfile){
USER_PROFILE_TL.set(userProfile);
}
public static UserProfile getUserProfile() {
return USER_PROFILE_TL.get();
}
public static String getCurrentUser() {
return Optional.ofNullable(USER_PROFILE_TL.get())
.map(UserProfile::getUid)
.orElse(UserProfile.ANONYMOUS_USER);
}
}
@Override
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
UserProfile userProfile = null;
// ... 驗證和獲取用戶信息 userProfile
UserProfileThread.setUserProfile(userProfile);
filterChain.doFilter(servletRequest, servletResponse);
}
//獲取當前用戶
String uid=UserProfileThread.getCurrentUser();
//獲取當前用戶對象
UserProfile user=UserProfileThread.getUserProfile();
2. 實現(xiàn)原理
public class ThreadLocal<T> {
public void set(T value) {
Thread t = Thread.currentThread();
ThreadLocalMap map = getMap(t);
if (map != null)
map.set(this, value);
else
createMap(t, value);
}
ThreadLocalMap getMap(Thread t) {
return t.threadLocals;
}
void createMap(Thread t, T firstValue) {
t.threadLocals = new ThreadLocalMap(this, firstValue);
}
// 省略其他方法
}
public class Thread implements Runnable {
ThreadLocal.ThreadLocalMap threadLocals = null;
//省略其他
}
每個線程是一個Thread實例，其內(nèi)部維護一個threadLocals的實例成員，其類型是ThreadLocal.ThreadLocalMap。 ThreadLocal本身并不是一個容器，我們存取的value實際上存儲在ThreadLocalMap中，ThreadLocal只是作為TheadLocalMap的key。
3. 注意事項
線程復(fù)用：在“獲取接口的當前請求用戶”的例子中，Tomcat中是通過線程池來處理用戶請求的，而線程池中線程是復(fù)用的。肯定會出現(xiàn)一個線程前后被不同用戶的接口請求復(fù)用的情況，因此需要對用過的ThreaLocal變量進行覆蓋或清除。 內(nèi)存溢出：由于ThreadLocalMap的生命周期跟Thread一樣長，如果創(chuàng)建的ThreadLocal變量很多，即對應(yīng)的key占用的內(nèi)存很大，但卻沒有手動刪除，到了一定程度就會導(dǎo)致內(nèi)存泄漏。
評論
圖片
表情