java 不同传入id不同的锁

原创

mob64ca12e83232 2024-07-29 06:25:44 ©著作权

文章标签 java Java 锁机制 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e83232的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java实现不同ID的锁机制

作为一名刚入行的开发者，你可能会对如何在Java中实现不同ID的锁机制感到困惑。别担心，接下来我将通过一个简单的示例，向你展示如何实现这一功能。

1. 理解需求

首先，我们需要明确需求：实现一个系统，当多个线程尝试访问同一个ID时，它们需要等待，直到当前线程完成操作。这就需要我们为每个ID创建一个锁。

2. 准备环境

在开始编码之前，确保你的开发环境已经安装了Java开发工具包（JDK）。

3. 编写代码

我们将使用Java的ReentrantLock来实现锁机制。下面是实现的步骤和代码：

3.1 创建锁的容器

首先，我们需要一个容器来存储每个ID对应的锁。

import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class LockManager {
    private Map<Integer, Lock> locks = new ConcurrentHashMap<>();
}

这里的ConcurrentHashMap是线程安全的，可以用于存储锁。

3.2 获取锁

接下来，我们需要一个方法来获取对应ID的锁。

public Lock getLock(int id) {
    return locks.computeIfAbsent(id, k -> new ReentrantLock());
}

computeIfAbsent方法会检查容器中是否已经存在对应ID的锁，如果不存在，则使用lambda表达式创建一个新的ReentrantLock。

3.3 使用锁

现在我们可以在需要同步的代码块中使用这个锁了。

public void performAction(int id) {
    Lock lock = getLock(id);
    lock.lock();
    try {
        // 同步代码块
        System.out.println("Performing action for ID: " + id);
    } finally {
        lock.unlock();
    }
}

在lock.lock()和lock.unlock()之间，我们放置需要同步执行的代码。使用finally块确保即使发生异常，锁也能被释放。

4. 测试代码

创建一个测试类，模拟多线程环境下访问不同ID的场景。

public class Main {
    public static void main(String[] args) {
        LockManager lockManager = new LockManager();

        for (int i = 0; i < 5; i++) {
            new Thread(() -> {
                lockManager.performAction(i % 3);
            }).start();
        }
    }
}

这里我们创建了5个线程，每个线程尝试访问ID为0、1或2的资源。