ThreadLocal详解

65 阅读2分钟

ThreadLocal简介

ThreadLocal又叫线程变量,是用来解决多线程并发问题的方案。通过将每个线程所需的信息独立存储来达到数据隔离的目的。
简单点说就是ThreadLocal中存的信息都属于当前线程

image.png

ThreadLocal源码理解

  1. set方法:
public void set(T value) {
    //获取当前线程
    Thread t = Thread.currentThread();
    //获取线程中的ThreadLocalMap,找到后对map赋值,如果没有创建新map
    ThreadLocalMap map = getMap(t);
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
}
void createMap(Thread t, T firstValue) {
        t.threadLocals = new ThreadLocalMap(this, firstValue);
    }
  1. get方法:
public T get() {
    //获取当前线程
    Thread t = Thread.currentThread();
    //获取线程中的ThreadLocalMap,找到后返回value值,如果没有初始化并赋值
    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();
}
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;
    }
  1. remove方法:
public void remove() {
    ThreadLocalMap m = getMap(Thread.currentThread());
    if (m != null)
        m.remove(this);
}

ThreadLocal简单使用

  1. 跨层传递数据;例如在请求进入时将用户信息存到ThreadLocal中,交由服务层使用,最终请求结束时清理用户信息

class Main{
    private static final ThreadLocal<String> threadLocal = new ThreadLocal<>();

    public static void main(String[] args) {
        Main main = new Main();
        main.controller();
        main.service();
        main.over();
    }

    /**
     * 模拟请求进入controller层
     */
    public void controller(){
        System.out.println("请求进入controller");
        threadLocal.set(Thread.currentThread().getName()+"线程的用户数据");
    }

    /**
     * 模拟业务使用线程数据
     */
    public void service(){
        System.out.println("请求进入service");
        System.out.println("service使用"+ threadLocal.get());
    }

    /**
     * 模拟请求结束动作
     */
    public void over(){
        System.out.println("请求结束");
        threadLocal.remove();
    }

}

注意事项

  1. ThreadLocal数据使用完毕后需要执行remove方法,否则可能会出现key为空value有值的问题;
    ThreadLocalMap 中使用的 key 为 ThreadLocal 的弱引用,弱引用的特点是,如果这个对象只存在弱引用,那么在下一次垃圾回收的时候必然会被清理掉。
    所以如果 ThreadLocal 没有被外部强引用的情况下,在垃圾回收的时候会被清理掉的,这样一来 ThreadLocalMap中使用这个 ThreadLocal 的 key 也会被清理掉。但是,value 是强引用,不会被清理