ThreadLocal 问题

先看源码

Thread 类

public class Thread implements Runnable {
	// 存放当前线程的ThreadLocal对象及其value
    ThreadLocal.ThreadLocalMap threadLocals = null;
}

ThreadLocal类

public class ThreadLocal<T> {
	// get 方法
    public T get() {
        Thread t = Thread.currentThread();
        // 获取当前线程的ThreadLocalMap
        ThreadLocalMap map = getMap(t);
        if (map != null) {
        	// 用当前ThreadLocal对象去当前线程的ThreadLocalMap里去找
            ThreadLocalMap.Entry e = map.getEntry(this);
            // 找到就返回
            if (e != null) {
                T result = (T)e.value;
                return result;
            }
        }
        // 找不到,就调用初始化方法
        return setInitialValue();
    }
    static class ThreadLocalMap {
        private Entry getEntry(ThreadLocal<?> key) {
        	// 通过hash计算获取下标
            int i = key.threadLocalHashCode & (table.length - 1);
            Entry e = table[i];
            // 如果该下标有值,且是当前key。则返回
            if (e != null && e.get() == key)
                return e;
            else
            	// 否则向后遍历数组查找
                return getEntryAfterMiss(key, i, e);
        }
        private Entry getEntryAfterMiss(ThreadLocal<?> key, int i, Entry e) {
            Entry[] tab = table;
            int len = tab.length;
			// 从该下标向后遍历查找。
            while (e != null) {
                ThreadLocal<?> k = e.get();
                // 是当前key则返回
                if (k == key)
                    return e;
				// 如果当前下表位置的key为null,则说明是被gc回收的
                if (k == null)
                	// 清理过期的Entry
                    expungeStaleEntry(i);
                else
                    i = nextIndex(i, len);
                e = tab[i];
            }
            return null;
        }
    }
}
public class ThreadLocal<T> {
	// set方法
    public void set(T value) {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null)
            // 以当前ThreadLoca对象作为key
            map.set(this, value);
        else
            createMap(t, value);
    }
    // 获取 当前线程 Thread 里的ThreadLocal.ThreadLocalMap 属性
    ThreadLocalMap getMap(Thread t) {
        return t.threadLocals;
    }
    // ThreadLocalMap 的set方法
    static class ThreadLocalMap {
        private void set(ThreadLocal<?> key, Object value) {

            Entry[] tab = table;
            int len = tab.length;
            // 通过hash计算数组下标
            int i = key.threadLocalHashCode & (len-1);
			// 从该下标向后遍历
            for (Entry e = tab[i];
                 e != null;  // 如果该处Entry 不为null
                 e = tab[i = nextIndex(i, len)]) {
                ThreadLocal<?> k = e.get();
				// 找到当前key,更新value
                if (k == key) {
                    e.value = value;
                    return;
                }
                // 如果当前key为null(被gc回收),则进行覆盖。
                if (k == null) {
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }
			// 放进 ThreadLocalMap 的元素里
            tab[i] = new Entry(key, value);
            int sz = ++size;
            // 清理空槽
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
            	// 如果没有清理掉,且已达到阈值,则进行扩充
                rehash();
        }
        // Entry 继承弱引用类
        static class Entry extends WeakReference<ThreadLocal<?>> {
            Object value;
			// 将ThreadLoca 转为弱引用
            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }
    }
}


为什么会被设置为弱引用

ThreadLocal是在ThreadLocalMap.Entry里被设置为弱引用的key,如果是强引用的话,当线程一直执行下去,且已没有外部强引用指向ThreadLocal,那么该 ThreadLocal 就无法被回收,导致内存泄露。

内存泄露问题

ThreadLocal 在ThreadLocal.ThreadLocalMap.Entry里作为key,且被设置为弱引用,所以当没有强引用指向时,就会被gc回收。
如果线程一直执行下去,那么在ThreadLocal.ThreadLocalMap.Entry里的强引用 value 就无法被回收。就会导致内存泄露。

设计者在很多地方都做了清理无用Entry,即key被回收的Entry。比如通过 key 查找 Entry 的时候,如果下标无法直接命中,那么就会向后遍历数组,此时遇到 key 为 null 的 Entry 就会清理掉;扩容时也会进行清理操作;set时,如果key为null,也会进行覆盖操作等。
这些被动的回收操作并不是做好的方法。最佳的实践方法是及时调用threadLocal.remove()方法进行清理。

当ThreadLocal 遇上线程池

线程池里的线程在执行完之后不会被销毁。
当线程再次被使用时,从ThreadLocal里获取的value可能是该线程上次被使用时创建使用的。所以可能会导致数据错误。
因为线程不被销毁,因此也会导致value内存泄露。

解决方案:当线程执行完成时,及时调用threadLocal.remove()方法来清除本次产生的value。

注:上述内存泄露的前提均是线程一直执行下去。当线程执行完,该线程的所有属性都会被回收。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值