ThreadLocal是Java中所提供的线程本地存储机制,可以利用该机制将数据缓存在某个线程内部,该线程可以在任意时刻、任意方法中获取缓存的数据。
原理
每个线程内都有一个 ThreadLocalMap 类型的成员变量,每个ThreadLocalMap里可以存很多ThreadLocal和其value。
public class Thread implements Runnable {
ThreadLocal.ThreadLocalMap inheritableThreadLocals = null;
}
- 调用 set 方法时,就会获取当前线程的ThreadLocalMap对象,以 ThreadLocal 自己作为 key,资源对象作为 value,放入当前线程的 ThreadLocalMap 集合中
- 调用 get 方法,就会获取当前线程的ThreadLocalMap对象,以 ThreadLocal 自己作为 key,到当前线程的ThreadLocalMap对象中获取值。
- 调用 remove 方法,就会获取当前线程的ThreadLocalMap对象,以 ThreadLocal 自己作为 key,当前线程的ThreadLocalMap对象中key对应的value。
public void set(T value) {
//获取当前线程
Thread t = Thread.currentThread();
//获取当前线程的ThreadLocalMap
ThreadLocalMap map = getMap(t);
if (map != null) {
map.set(this, value);
} else {
//当前线程还没有ThreadLocalMap,为当前线程new一个ThreadLocalMap
createMap(t, value);
}
}
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();
}
public void remove() {
ThreadLocalMap m = getMap(Thread.currentThread());
if (m != null) {
m.remove(this);
}
}
public class Main {
public static void main(String[] args) throws InterruptedException {
ThreadLocal<Object> tl = new ThreadLocal<>();
String s = "123123";
tl.set(s);
tl.get();
tl.remove();
ThreadLocal<Object> tl2 = new ThreadLocal<>();
String s1 = "2222222";
tl2.set(s);
tl2.get();
tl2.remove();
}
}
使用场景:实现登陆功能,使用登陆凭证token获取用户的信息,获取到的基本信息存入ThreadLocal中以供线程后续功能使用。