并发 -- ThreadLocal

1. 介绍

ThreadLocal类主要解决的就是让每个线程绑定自己的值,可以将ThreadLocal类形象的比喻成存放数据的盒子,盒子中可以存储每个线程的私有数据。

数据结构

public class Thread implements Runnable {
    //与此线程有关的ThreadLocal值。由ThreadLocal类维护
    ThreadLocal.ThreadLocalMap threadLocals = null;

    //与此线程有关的InheritableThreadLocal值。由InheritableThreadLocal类维护
    ThreadLocal.ThreadLocalMap inheritableThreadLocals = null;
}

每个线程有一个自己的ThreadLocalMap。将它的key视作ThreadLocal(非本身,而是它的一个弱引用),value为代码中放入的值。ThreadLocalMap 为 ThreadLocal的静态内部类。
在这里插入图片描述

内存泄漏
如果把ThreadLocal置为null,只有弱引用,GC是可以回收这部分空间的,但是value却存在一条从Current Thread过来的强引用链。只有当Current Thread销毁时,value才能得到释放。在threadLocal 为 null 和线程结束这段时间内,就发生了内存泄露。

解决:
ThreadLocalMap实现中考虑了这种情况,在调用 set()、get()、remove() 方法的时候,会清理掉 key 为 null 的记录。使用完 ThreadLocal方法后 最好手动调用remove()方法。

部分源码:
ThreadLocal类的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 getMap(Thread t) {
        return t.threadLocals;
    }

ThreadLocalMap 中存的是[ThreadLocal对象引用, 存放的值],即最终变量是放在当前线程的成员变量ThreadLocals中,而ThreadLocals是ThreadLocalMap 类型,并不是存在 ThreadLocal 上,ThreadLocal 可以理解为ThreadLocalMap的封装,传递了变量值。

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)
                 return (T)e.value;
         }
         return setInitialValue();
   } 

ThreadLocal类的initialValue()方法

protected T initialValue() {
        return null;
 }

该方法实现只返回 null,并且修饰符为protected,如果用户想返回初始值不为null,则需要定义线程变量时重写该方法;最多调用此方法一次,即线程第一次使用 get() 方法访问变量的时候。如果线程先调用 set(T) 方法,再调 get 方法,则不会在线程中调用 initialValue 方法。

ThreadLocalMap Hash算法
既然是Map结构,那么ThreadLocalMap也要实现自己的hash算法来解决散列表数组冲突问题。
int i = key.threadLocalHashCode & (len-1);

public class ThreadLocal<T> {
    private final int threadLocalHashCode = nextHashCode();
    private static AtomicInteger nextHashCode = new AtomicInteger();
    private static final int HASH_INCREMENT = 0x61c88647;
    
    //每当创建一个ThreadLocal对象,这个ThreadLocal.nextHashCode这个值就会增长 `0x61c88647`
    private static int nextHashCode() {
        return nextHashCode.getAndAdd(HASH_INCREMENT);
    }
    
    static class ThreadLocalMap{
        ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
            table = new Entry[INITIAL_CAPACITY];
            int i = firstKey.threadLocalHashCode & (INITIAL_CAPACITY - 1);
            table[i] = new Entry(firstKey, firstValue);
            size = 1;
            setThreshold(INITIAL_CAPACITY);
        }
    }
}

ThreadLocalMap.Entry:

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

            Entry(ThreadLocal<?> k, Object v) {
                super(k);
                value = v;
            }
        }
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值