线程学习记录07-ThreadLocal源码解析及实例

ThreadLocal ：每个线程独立的局部变量

jdk1.8主要有三个方法：

set(); 为变量赋值
get();获取线程对应的变量
remove();移除该线程的变量

部分源码：
get（）方法:

   public T get() {
    //获取当前线程(这作为key值)
    Thread t = Thread.currentThread();
    //获取这个线程对应的MAP
    ThreadLocalMap map = getMap(t);
     //如果该线程的map存在，则去获取该map的value值(当前ThreadLocal对象就是这个map的key)
    if (map != null) {
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
    }
    //如果不存在map，则进行初始化
    return setInitialValue();
}

 /**
  * 初始化ThreadLocalMap 的方法
  */
private T setInitialValue() {
    //这个方法固定返回的就是null
    T value = initialValue();
    //如果ThreadLocalMap 不存在，则就先初始化一个null的ThreadLocalMap 
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
    return value;
}

set()方法:

public void set(T value) {
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null)
        //重点：this是map的key值
        map.set(this, value);
    else
        createMap(t, value);
}

void createMap(Thread t, T firstValue) {
    //产生一个map容器，this固定的作为key值。创建map之后将其赋予给线程，threadLocals 是Thread里面的一个变量
    t.threadLocals = new ThreadLocalMap(this, firstValue);
}

remove()方法:

public void remove() {
    //很简单，直接利用内部存在且已知的两个key删除对应的数据
     ThreadLocalMap m = getMap(Thread.currentThread());
     if (m != null)
         m.remove(this);
 }

JDK1.8的ThreadLocal 主要的就是以上三个方法。通过上面三个方法，我们可以知道的就是ThreadLocal为每个线程创建了一个ThreadLocalMap ，而ThreadLocal对象本身又去作为ThreadLocalMap 的key值，这样可以保证每个线程都能用有自己独立的变量。
ThreadLocal 与synchronized等加锁的方式不同，ThreadLocal 完全不提供锁，而使用空间换时间的手段，为每个线程提供独立的副本，保证线程的安全性。如果在并发不是很高的时候加锁的性能更好，但是并发量很高的时候使用ThreadLocal 可以一定程度的减少锁竞争。

实例代码:

public class ConnThreadLocal {

public static ThreadLocal th = new ThreadLocal();

public void setTh(String value){
    th.set(value);
}
public void getTh(){
    System.out.println(Thread.currentThread().getName() + ":" + this.th.get());
}

public static void main(String[] args) throws InterruptedException {

    final ConnThreadLocal ct = new ConnThreadLocal();
    Thread t1 = new Thread(new Runnable() {
        @Override
        public void run() {
            ct.setTh("张三");
            ct.getTh();
        }
    }, "t1");

    Thread t2 = new Thread(new Runnable() {
        @Override
        public void run() {
            try {
                Thread.sleep(1000);
                ct.getTh();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }, "t2");

    t1.start();
    t2.start();
}
}

运行结果:

image.png

t1线程已经为ThreadLocal赋值，可以获取到正确的值。t2获取的依旧是null，因为t2在ThreadLocal里面的Map并没被赋值