快捷導(dǎo)航

詳解ThreadLocal為什么會內(nèi)存溢出原理

更新時間：2023年01月04日 14:56:36 作者：Lxlxxx

這篇文章主要為大家介紹了ThreadLocal為什么會內(nèi)存溢出原理詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

前言

關(guān)于ThreadLocal （線程本地存儲），從字面意思上看主要是存儲一些本地變量，使它們能在一個線程內(nèi)共用，與其他的線程進(jìn)行數(shù)據(jù)隔離，保證了數(shù)據(jù)在一個線程內(nèi)的安全性，日常的開發(fā)中，ThreadLocal的使用場景還是比較常見的，包括登陸信息的token的存儲、連接管理一個線程持有一個鏈接，該連接可以在不同的方法之間進(jìn)行傳遞，一個線程內(nèi)數(shù)據(jù)共享，通過key，value的形式存儲數(shù)據(jù)。

ThreadLocal源碼分析

ThreadLocal類有一個靜態(tài)內(nèi)部類，ThreadLocalMap可以看到內(nèi)部有個Entry 數(shù)組k就是ThreadLocal的引用，Entry 繼承了WeakReference 說明Entry 的k是個弱引用，從這看來如果是弱引用那么就不會存在內(nèi)存溢出，GC運行的時候，這個對象就會被回收掉，value則是存儲的對象，而ThreadLocalMap則是由threadLocals來創(chuàng)建的，可以看到這兩個變量的默認(rèn)都是NULL。

p>只有當(dāng)線程第一次調(diào)用的時候才會創(chuàng)建它。

ThreadLocal value內(nèi)存溢出

前面講到ThreadLocal的key是threadlocals是弱引用不會存在內(nèi)存溢出，那么容易存在內(nèi)存溢出的一定是它的value，它與current thread 存在一個強(qiáng)引用的關(guān)系，導(dǎo)致value無法進(jìn)行回收，如果線程的對象一直不去銷毀這個強(qiáng)引用的對象，那么導(dǎo)致這個關(guān)系一直存在就會出現(xiàn)內(nèi)存溢出，

/**
 * 內(nèi)存溢出例子
 */
public class ThreadLocalTest {
    static class Mytask{
        //定義10m的Byte數(shù)組
       private Byte[] bytes =new Byte[10 *1024 * 1024];
    }
    private static ThreadLocal<Mytask> threadLocal = new ThreadLocal();
    public static void main(String[] args) throws InterruptedException {
        // 5個核心線程、5個最大線程、隊列長度100
        ThreadPoolExecutor threadPoolExecutor =new ThreadPoolExecutor(5, 5, 60,
                TimeUnit.SECONDS, new LinkedBlockingQueue<>(100));
        for (int i = 0; i < 10; i++) {
            //執(zhí)行任務(wù)
            executeTask(threadPoolExecutor);
            Thread.sleep(1000);
        }
    }
    private static void executeTask(ThreadPoolExecutor threadPoolExecutor){
        threadPoolExecutor.execute(new Runnable() {
            @Override
            public void run() {
                System.out.println("創(chuàng)建Mytask對象");
                Mytask mytask =new Mytask();
                threadLocal.set(mytask);
            }
        });
    }

堆內(nèi)存設(shè)置50m

通過上面代碼，創(chuàng)建線程池對創(chuàng)建的任務(wù)，放入threadlocal里面，可以看到出現(xiàn)了堆內(nèi)存的溢出，存放的任務(wù)一直在引用沒有得到釋放導(dǎo)致堆內(nèi)存空間不足。

p>我們在set值到threadLocal后面加入finally，調(diào)用它的remove方法來清除它的內(nèi)存那么就不會發(fā)生內(nèi)存溢出。

來看看remove的代碼，可以看到獲取當(dāng)前線程的threadLocals，然后調(diào)用remove方法獲取到全部的Entry數(shù)組，判斷不為空，key也是當(dāng)前的key則調(diào)用clear方法將數(shù)組清除，這樣數(shù)組空間得到了釋放自然就不會出現(xiàn)內(nèi)存溢出。

public void remove() {
    ThreadLocalMap m = getMap(Thread./currentThread/());
    if (m != null)
        m.remove(this);
}
private void remove(ThreadLocal<?> key) {
    Entry[] tab = table;
    int len = tab.length;
    int i = key.threadLocalHashCode & (len-1);
    for (Entry e = tab[i];
         e != null;
         e = tab[i = nextIndex(i, len)]) {
        if (e.get() == key) {
            e.clear();
            expungeStaleEntry(i);
            return;
        }
    }
}