ThreadLocal简介
ThreadLocal又叫线程变量,是用来解决多线程并发问题的方案。通过将每个线程所需的信息独立存储来达到数据隔离的目的。
简单点说就是ThreadLocal中存的信息都属于当前线程
ThreadLocal源码理解
- 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);
}
- 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;
}
- remove方法:
public void remove() {
ThreadLocalMap m = getMap(Thread.currentThread());
if (m != null)
m.remove(this);
}
ThreadLocal简单使用
- 跨层传递数据;例如在请求进入时将用户信息存到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();
}
}
注意事项
- ThreadLocal数据使用完毕后需要执行remove方法,否则可能会出现key为空value有值的问题;
ThreadLocalMap 中使用的 key 为 ThreadLocal 的弱引用,弱引用的特点是,如果这个对象只存在弱引用,那么在下一次垃圾回收的时候必然会被清理掉。
所以如果 ThreadLocal 没有被外部强引用的情况下,在垃圾回收的时候会被清理掉的,这样一来 ThreadLocalMap中使用这个 ThreadLocal 的 key 也会被清理掉。但是,value 是强引用,不会被清理