Thread、Thread Local、ThreadLocalMap学习

Thread、ThreadLocal、ThreadLocalMap学习

一、概念理解

java中，为了在线程内部实现数据的传输共享，以及保证数据或者变量的线程隔离，引入了ThreadLocal类。
ThreadLocal类中，有一个静态内部类ThreadLocalMap，这个map并没有继承HashMap，而是自己实现了map的功能以及map中的键值对单元。

• Thread

java中的线程类，每个对象代表一个线程。

• ThreadLocal

按照名称翻译应为：本地线程，但实际上应该理解为本地线程变量的处理类。

• ThreadLocalMap

很明显，这是一个容器，用来存储线程内的变量。

• 以上三者之间的关系：

  • JAVA8之前
    ThreadLocalMap作为ThreadLocal中的属性存在，Thread对象作为ThreadLocalMap中的key存在，也就是说在ThreadLocal对象的ThreadLocalMap对象中，有很多的Entry，每个Entry的key就是一个Thread线程对象，Entry的value就是线程变量的值。

  • JAVA8之后
    ThreadLocalMap作为Thread的属性存在，ThreadLocal对象作为ThreadLocalMap中的key存在，也就是说，在一个Thread线程对象中，有很多的Entry，每个Entry的key就是一个ThreadLocal对象，而Entry的value也是线程变量的值。

  • 总结
    JAVA8之前和JAVA8之后，都是使用ThreadLocalMap中的value来存储实际的变量值，区别是JAVA8之前map是存放在ThreadLocal对象中，key是采用的Thread线程对象；
    而JAVA8之后map是归属在Thread中，key采用的是ThreadLocal对象。仿佛是做了一次行列转换。

二、源码分析（JAVA8）

• ThreadLocal
  ThreadLocal类中，主要提供了若干方法，具体如下
  • getMap

  ThreadLocalMap getMap(Thread t) {
      return t.threadLocals;
  }
  

  这个方法是根据线程来获取ThreadLocalMap，从这里可以看出，ThreadLocalMap是作为Thread的属性存在的，代码中的threadLocals属性就是ThreadLocal.ThreadLocalMap对象。
  • 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);
  }
  

  set方法的流程如下：首先获取当前线程；然后获取当前线程中的ThreadLocalMap；最后在map中存储相应的变量值，而key用的就是this，此处代表的为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();
  }
  

  get方法的流程也是首先获取当前线程，再根据当前线程获取ThreadLocalMap对象，然后在map中根据this，也就是ThreadLocal对象作为key，来获取相应的Entry中的value。最后，如果map不存在，或者Entry不存在的话，就初始化map或者Entry，此时Entry中的value为null。
  • remove

  public void remove() {
       ThreadLocalMap m = getMap(Thread.currentThread());
       if (m != null)
           m.remove(this);
   }
  

  remove方法比较简单，就是根据当前线程获取map，如果map不为null，则从map中删除以本ThreadLoca对象作为key的Entry。
  • 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;
  }
  

setInitialValue这个方法就是个初始化，从上面可以看到，在ThreadLoca对象的get方法没有获取value的时候，如果map为null，或者Entry为null，则调用此方法进行初始化，没有map就创建map，没有Entry，就以null作为value，以ThreadLocal作为key创建一个Entry，存储到map中。

• ThreadLocalMap
  ThreadLocalMap类是ThreadLocal类的一个静态内部类，可以理解为功能如HashMap的类，具体的代码如下：

public class ThreadLocal<T> {
	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;
            }
        }
	}
}

从以上的代码可以看出，ThreadLocal、ThreadLocalMap以及Entry的层级关系，其中值得注意的是Entry是继承自WeakReference弱引用，java中弱引用的意义为：gc的时候，如果没有其它强引用指向这个对象的话，这个对象就会被回收。