快捷導(dǎo)航

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

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

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

前言

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

ThreadLocal源碼分析

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

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

ThreadLocal value內(nèi)存溢出

前面講到ThreadLocal的key是threadlocals是弱引用不會(huì)存在內(nèi)存溢出，那么容易存在內(nèi)存溢出的一定是它的value，它與current thread 存在一個(gè)強(qiáng)引用的關(guān)系，導(dǎo)致value無法進(jìn)行回收，如果線程的對(duì)象一直不去銷毀這個(gè)強(qiáng)引用的對(duì)象，那么導(dǎo)致這個(gè)關(guān)系一直存在就會(huì)出現(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個(gè)核心線程、5個(gè)最大線程、隊(duì)列長度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對(duì)象");
                Mytask mytask =new Mytask();
                threadLocal.set(mytask);
            }
        });
    }

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

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

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

來看看remove的代碼，可以看到獲取當(dāng)前線程的threadLocals，然后調(diào)用remove方法獲取到全部的Entry數(shù)組，判斷不為空，key也是當(dāng)前的key則調(diào)用clear方法將數(shù)組清除，這樣數(shù)組空間得到了釋放自然就不會(huì)出現(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;
        }
    }
}