Java并发编程：ReentrantLock替换ThreadLocal示例

原创拼搏现实的明天。 2024-12-06 12:33 6阅读 0赞

在Java并发编程中，`ReentrantLock`是取代`ThreadLocal`的常用同步工具。`ThreadLocal`主要用于线程间独立的变量存储，而`ReentrantLock`则提供了更强大的锁机制。

下面是一个使用`ReentrantLock`替换`ThreadLocal`的简单示例：

```java
import java.util.concurrent.locks.ReentrantLock;

class ThreadLocalExample {
    private static final ReentrantLock lock = new ReentrantLock();

// 使用ThreadLocal存储的数据
    private static volatile String threadData;

public void setData(String data) {
        try {
            // 获取锁，确保数据修改的原子性
            lock.lock();
            // 更新ThreadLocal存储的数据
            threadData = data;
        } finally {
            // 无论是否成功修改数据，都必须释放锁
            lock.unlock();
        }
    }

public static void main(String[] args) {
        ThreadLocalExample example = new ThreadLocalExample();

// 在线程A中设置数据
        Thread threadA = new Thread(() -> {
            String data = "Data from A";
            example.setData(data);
        }));

// 在线程B中获取并打印数据
        Thread threadB = new Thread(() -> {
            example.setData("Initial Data");
            String data = example.getData();
            System.out.println("Data from B: " + data);
        }));

// 启动两个线程
        threadA.start();
        threadB.start();

// 确保主线程能够完全运行和关闭应用程序
        try {
            threadA.join();  // 等待线程A结束
            threadB.join();  // 等待线程B结束
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}
```

这个例子中，我们创建了一个`ThreadLocalExample`类，它有一个`threadData`变量被存储在`ThreadLocal`中。然后，我们用一个`ReentrantLock`来控制对`threadData`的访问，确保了数据修改的原子性。

最后，在主线程中启动两个线程（A和B），分别进行数据设置和获取操作，展示了如何使用`ReentrantLock`替换`ThreadLocal`进行并发编程。