ThreadLocal

最新推荐文章于 2025-07-11 12:20:25 发布

原创最新推荐文章于 2025-07-11 12:20:25 发布 · 319 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #jvm #开发语言

ThreadLocal（线程局部变量）用于在线程中保存数据，在ThreadLocal中保存的数据仅属于当前线程，该变量是当前线程独有的变量。ThreadLocal使用ThreadLocalMap进行数据存储。

常用方法

ThreadLocal中的get()方法,set()方法，remove()方法，都是在操作ThreadLocalMap中的数据。

1.set():将数据存储至当前线程的ThreadLocalMap中。ThreadLocal对象做key,将数据保存在value中。

public void set(T value) {
    // 获取当前线程
    Thread t = Thread.currentThread();

	// 获取当前线程的ThreadLocalMap
    ThreadLocalMap map = getMap(t);

	// 存储数据至当前线程的ThreadLocalMap中
	// 使用ThreadLocal对象做key，保存数据value
    if (map != null)
        map.set(this, value);
    else
        createMap(t, value);
}

2. get()从当前现成的ThreadLocalMap中获取数据

public T get() {
    // 获取当前线程的ThreadLocalMap
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);

    // 使用ThreadLocal对象做key，获取数据(Entry类型)
    if (map != null) {
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            @SuppressWarnings("unchecked")
            T result = (T)e.value;
            return result;
        }
    }
    return setInitialValue();
}

ThreadLocal中的数据只属于当前线程，该数据对别的线程是不可见的.

public class Demo01 {
	//定义全局的公共区域
	public static ThreadLocal<String> roleThreadLocal=new ThreadLocal<String>();
	
	public static void main(String[] args) throws InterruptedException {
		//创建线程1
		Thread t1=new Thread(new Runnable() {		
			@Override
			//当前ThreadLocal对象做key
			public void run() {
				roleThreadLocal.set("苏妲己");
				show();
				Sample.dosth();
				
			}
		},"线程1");
		
	Thread t2=new Thread(new Runnable() {
			
			@Override
			public void run() {
				roleThreadLocal.set("商纣王");
				show();
				Sample.dosth();
				
			}
		},"线程2");
		
		t1.start();
		t2.start();
		
		t1.join();
		t2.join();
		System.out.println("THE END!");
	}
	
	// 线程1或线程2
	public static void show() {
		System.out.println("show:"+Thread.currentThread().getName()+"分配角色:"+roleThreadLocal.get());
	}
}

class Sample {
	// 线程1或线程2
	public static void dosth() {
		System.out.println("dosth:"+Thread.currentThread().getName()+"分配角色:"+Demo01.roleThreadLocal.get());
	}
}

3.remove()从当前线程的ThreadLocalMap中删除数据

线程执行完毕后也要调用remove()，避免在线程被重新放入线程池中时被本地变量的旧状态仍然被保存。在finally代码块中，调用remove()方法清理没用的数据。如果业务代码出现异常，也能及时清理没用的数据。

remove()方法中会把Entry中的key和value都设置成null，这样就能被GC及时回收，无需触发额外的清理机制，所以它能解决内存泄露问题。

public class Demo02 {
public static void main(String[] args) {
	ThreadPoolExecutor threadPoolExecutor=new ThreadPoolExecutor(
			1, 1, 0, TimeUnit.SECONDS, new LinkedBlockingDeque<Runnable>());
	String[] roleArrary= {"闻仲","杨戬","商纣王","苏妲己"}; 
	//循环提交4个线程任务，分配不同的角色
	for(String role:roleArrary) {
		threadPoolExecutor.submit(new Runnable() {
			
			@Override
			public void run() {
				try {
					Demo01.show();
					Demo01.roleThreadLocal.set(role);
					Demo01.show();
					Sample.dosth();		
				}finally {
					Demo01.roleThreadLocal.remove();//清空,以免数据留存
				}
						
			}
		});
	}
	
	//关闭线程池
	threadPoolExecutor.shutdown();
}
}

ThreadLocalMap内部结构

ThreadLocalMap内部数据结构是Entry类型数组。每个Entry对象的key为ThreadLocal对象，value为存储的数据。

不使用Thread做key的原因：一个线程中可能不止有一个ThreadLocal存值，用Thread，会导致key混乱。

static class ThreadLocalMap {
    static class Entry extends WeakReference<ThreadLocal<?>> {
        Object value;

        Entry(ThreadLocal<?> k, Object v) {
            super(k);
            value = v;
        }
   }
   ...
   private Entry[] table;
   ...
}

父子线程如何共享数据

在父线程中在ThreadLocal设置了值，在子线程中能够获取到，就是父子线程共享数据。

父子线程都是独立线程，ThreadLocal在这种情况下是无法做到的。main方法是在主线程中执行的，相当于父线程。在main方法中开启了另外一个线程，相当于子线程。两个线程对象，各自拥有不同的ThreadLocalMap。

想要实现父子线程共享数据，应该使用InheritableThreadLocal，它是JDK自带的类，继承了ThreadLocal类。

public class Demo03 {
	public static void main(String[] args) {

		//指向到子类后子线程和父线程共用一个threadLocal
		ThreadLocal<String> threadLocal=new InheritableThreadLocal();
		threadLocal.set("天王盖地虎");
		System.out.println("main主线程"+threadLocal.get());
		//创建并启动子线程
		Thread t1=new Thread(new Runnable() {
			
			@Override
			public void run() {
				System.out.println("子线程"+threadLocal.get());
				
			}
		});
		t1.start();
	}

}