ThreadLocal源码解析

本文详细介绍了ThreadLocal的作用,数据存储机制(ThreadLocalMap),以及如何避免因线程池导致的内存泄露问题。特别强调了使用后清理ThreadLocal以防止内存泄漏的重要性。
摘要由CSDN通过智能技术生成

提供了线程独享局部变量存储的能量,主要是可以在整个线程存活的时间内,在任意地方可以随用随取。简化了参数的传递,数据的获取

注意:

  • 使用完成后,ThreadLocal的值需要清除,不然可能会造成内存泄漏的风险
  • 不具备线程安全以及线程间数据隔离的能力,仅仅简化了参数的传递,方便线程的任意地方去获取数据

ThreadLocal的数据存储是存到了ThreadLocalMap中,只封装了get、set读写的方法

ThreadLocalMap

数据模型:

  • 每个线程对应一个ThreadLocalMap对象。本质是一个数组实现的散列map
  • 每个元素都是Entry对象,key为threadLocal,value为任意值

set

    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进行存储
    • 如果获取的ThreadLocalMap不为空,就调用这个map的set方法,取更新或者插入这个值
    • 如果为空,就调用createMap去创建一个ThreadLocalMap对象,并把初始值存入,更新线程的threadLocals的引用
    void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }

注意

        这里是以当前线程作为key,value值只能是一种类型的数据

如何存储多个数据:

  • 封装成一个对象
  • 创建多个ThreadLocal进行存储

get

    public T get() {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        return setInitialValue();
    }

首先也是拿到当前线程的ThreadLocalMap对象,然后通过map.getEntey去获取值

如果为空就会调用setInitialValue()方法

    private T setInitialValue() {
        T value = initialValue();
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
        return value;
    }

创建ThreadLocalMap,然后给ThreadLocalMap设置一个初始数据、

初始数据是从被调用ThreadLocal的初始值中获取到的,

比如在主线程中创建了一个ThreadLocal,复写了它的initialValue方法,在子线程去调用ThreadLocal时,就会为子线程创建一个ThreadLocalMap,它的初始值就为initialValue

ThreadLocalMap

    static class ThreadLocalMap {

        static class Entry extends WeakReference<ThreadLocal<?>> {
            /** The value associated with this ThreadLocal. */
            Object value;

            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }
        ……

        private void set(ThreadLocal<?> key, Object value) {
            ……
            tab[i] = new Entry(key, value);
            ……
        }

}

将key-value存储到Entry数组中,这个Entry是继承自WeakReference,也就是ThreadLocal的key是被这个弱引用所持有的,而value是进行了强引用的

这个弱引用的对象在什么时候被回收?

        当这个弱引用所持有的对象的强引用,就是new出来的对象被置为null时,GC扫描出来这个弱引用才会被回收

ThreadLocal内存泄露发生的场景:

        线程池的使用,池子中的线程,任务执行完成之后并不会被立刻回收,如果手动将ThreadLocal的强引用置为null,在GC扫描时就会把ThreadLocal给回收调,而ThreadLocal的value是一个强引用,如果说任务执行完成之后没有手动的把value给移除掉,页面关闭了,但是这个线程里所持有的value对象是不会被回收的,它是一个强引用,这个value超过了它的生命周期,就发生了内存泄露

        在调用get、set、replace时都会去检查是否有连续的key为null的数据,如果有的话就把这个value给移除掉,这个时候下一次GC扫描时就能把value给回收掉,但不能保证能移除所有key为null的情况

        线程池的解决方案:要主动的把它remove掉

  • 22
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值