JAVA 并发 ThreadLocal
ThreadLocal,即线程变量,是一个以ThreadLocal对象为键、任意对象为值的存储结构。在 Java 里,ThreadLocal
类能够让每个线程都拥有属于自己的独立变量副本。每个线程对这个变量的操作,都只会影响自身的副本,不会对其他线程的副本造成影响。
核心概念
ThreadLocal
的主要作用是为每个使用该变量的线程都创建一个独立的副本。每个线程都可以独立地改变自己的副本,而不会影响其他线程所对应的副本。从本质上来说,ThreadLocal
是通过将数据与线程进行绑定,以此来实现线程间数据的隔离。
ThreadLocal的数据结构
Thread
类有一个类型为ThreadLocal.ThreadLocalMap
的实例变量threadLocals
,也就是说每个线程有一个自己的ThreadLocalMap
。
ThreadLocalMap
有自己的独立实现,可以简单地将它的key
视作ThreadLocal
,value
为代码中放入的值(实际上key
并不是ThreadLocal
本身,而是它的一个弱引用)。
弱引用引发的问题:
当外部对ThreadLocal
对象的强引用被释放后,ThreadLocal
对象会被垃圾回收。然而,此时Entry
中的值(value)是强引用,它不会被自动回收。如果线程一直存活(比如在线程池中),这些值就会一直存在,从而导致内存泄漏。
为了避免内存泄漏,在使用完ThreadLocal
后,要及时调用remove()
方法来移除对应的条目。
基本用法
使用ThreadLocal
主要涉及以下几个方法:
get()
:获取当前线程的变量副本。set(T value)
:为当前线程设置变量的副本值。remove()
:移除当前线程的变量副本。initialValue()
:提供初始值,该方法为 protected 类型,通常需要通过匿名内部类进行重写。
示例:
public class Haohaoxuexi {// 每个实例包含一个消息列表private List<String> messages = new ArrayList<>();// 静态ThreadLocal,为每个线程提供独立的Haohaoxuexi实例public static final ThreadLocal<Haohaoxuexi> holder = ThreadLocal.withInitial(Haohaoxuexi::new);// 向当前线程的消息列表添加消息public static void add(String msg) {holder.get().messages.add(msg);}// 清除当前线程的消息列表并返回public static List<String> clear() {List<String> messages = holder.get().messages;holder.remove();System.out.println("size is" + messages.size());return messages;}// 主方法:创建10个线程,每个线程添加一条消息并打印public static void main(String[] args) {Thread[] threads = new Thread[10];for (int i = 0; i < 10; i++) {int j = i;threads[i] = new Thread(() -> {Haohaoxuexi.add("msg" + j);List<String> messages = holder.get().messages;System.out.println(messages);});threads[i].start();}}
}
程序流程:
1.主方法创建 10 个工作线程
Thread[] threads = new Thread[10];
2.核心结构与初始化
// 每个实例包含一个消息列表private List<String> messages = new ArrayList<>();// 静态ThreadLocal,为每个线程提供独立的Haohaoxuexi实例public static final ThreadLocal<Haohaoxuexi> holder = ThreadLocal.withInitial(Haohaoxuexi::new);
ThreadLocal<Haohaoxuexi> holder
:当创建线程时,为每个线程创建独立的key为holder,value为Haohaoxuexi类的实例(其中包含了messages对象,messages是引用,指向独立的ArrayList实例)withInitial(Haohaoxuexi::new)
:首次调用get()
时初始化实例
3.每个工作线程执行流程
for (int i = 0; i < 10; i++) {int j = i;threads[i] = new Thread(() -> {// 1. 调用静态方法add()Haohaoxuexi.add("msg" + j);// 2. add()方法内部:// a. holder.get()为线程T1生成第一个Haohaoxuexi实例// b. 调用该实例的messages.add("msg0")// 3. 获取当前线程的消息列表并打印List<String> messages = holder.get().messages;System.out.println(messages); // 输出: [msg0]});threads[i].start();}
(1)调用Haohaoxuexi
类中的add方法 -->通过holder.get().messages.add(msg) 为messages动态数组对象加入“msg”+【j】元素
为什么需要int i = j 而不是直接使用变量i :
在 Java 中,Lambda 表达式和匿名内部类只能捕获外部的最终变量。所谓最终变量,是指在初始化后就不会再被修改的变量。如果你直接在 Lambda 表达式中使用循环变量i
,会出现编译错误,因为循环变量i
在每次迭代中都会被修改,不满足最终变量的要求。
为了满足这一要求,需要创建一个局部变量j
,并将当前迭代的i
值赋给它。这个局部变量j
在 Lambda 表达式捕获时是不会被修改的,因此它是最终变量。这样,每个线程捕获的是不同的j
实例,从而保证了每个线程使用的是正确的索引值。
(2)List<String> messages = holder.get().messages;
通过key holder使用get方法返回messages实例
(3)输出
常见应用场景
ThreadLocal
在实际开发中有多种常见的应用场景:
- 数据库连接管理:可以为每个线程分配独立的数据库连接,避免在多线程环境下使用同一个连接而引发的问题。
- 会话管理:在 Web 应用中,为每个请求线程分配独立的会话对象,方便在整个请求处理过程中使用。
- 事务管理:确保同一个事务中的操作都在同一个数据库连接上执行。
- 线程安全的单例模式:实现每个线程都有自己的单例实例。
总结
ThreadLocal
是 Java 中实现线程隔离的重要工具,合理使用它可以简化多线程编程,但同时也需要注意其潜在的内存泄漏风险和适用场景。