分布式锁实现方式(基于Redis的分布式锁+基于ZooKeeper的分布式锁)
一. 基于Redis的分布式锁
1.名词解释
Redis 是一种内存数据库,其特点之一就是支持原子性操作,这使得它成为实现分布式锁的理想选择。基于 Redis 实现的分布式锁主要依赖于 Redis 的 SETNX(SET if Not eXists)命令,该命令用于设置键的值,但仅在键不存在时设置成功。锁的释放则可以通过 DEL 命令或设置过期时间来实现。
实现步骤:
- 客户端请求获取锁时,调用 SETNX 命令尝试将某个唯一标识符作为键名存入 Redis 中,如果 SETNX 执行成功,则获取到锁。
- 如果 SETNX 执行失败,则表示锁已被其他客户端占用,客户端可以选择重试或放弃。
- 获取到锁的客户端在完成操作后,通过 DEL 命令删除键来释放锁,或者可以为键设置过期时间,确保即使客户端崩溃或网络中断,锁也会自动释放。
优点:
- 实现简单,易于理解和部署。
- Redis 的高性能和可靠性保证了分布式锁的性能和稳定性。
缺点:
- 锁的获取和释放不是原子操作,可能会导致死锁或锁过期等问题。
- 锁的粒度较粗,无法支持细粒度的锁。
2.代码实现
下面是一个基于 Redis 的分布式锁的 Java 代码示例:
import redis.clients.jedis.Jedis;
public class RedisDistributedLock {
private final Jedis jedis;
private final String lockKey;
private final int expireTimeMs;
private boolean locked = false;
public RedisDistributedLock(Jedis jedis, String lockKey, int expireTimeMs) {
this.jedis = jedis;
this.lockKey = lockKey;
this.expireTimeMs = expireTimeMs;
}
public boolean acquireLock() {
// 使用 SETNX 命令尝试获取锁
String result = jedis.set(lockKey, "1", "NX", "PX", expireTimeMs);
if ("OK".equals(result)) {
locked = true;
return true;
}
return false;
}
public void releaseLock() {
// 释放锁
if (locked) {
jedis.del(lockKey);
locked = false;
}
}
// 示例用法
public static void main(String[] args) {
Jedis jedis = new Jedis("localhost");
RedisDistributedLock lock = new RedisDistributedLock(jedis, "mylock", 10000); // 锁10秒钟
try {
if (lock.acquireLock()) {
System.out.println("获取锁成功,执行业务逻辑...");
// 执行业务逻辑
} else {
System.out.println("获取锁失败,其他线程正在操作...");
}
} finally {
lock.releaseLock(); // 释放锁
}
jedis.close();
}
}
这个示例代码创建了一个 `RedisDistributedLock` 类,它封装了 Redis 的连接和操作方法,并提供了获取锁和释放锁的功能。在 `main` 方法中,首先创建了一个 `RedisDistributedLock` 实例,并尝试获取锁。如果获取锁成功,则执行业务逻辑;否则输出获取锁失败的消息。最后,无论是否获取到锁,都会释放锁。
需要注意的是,该示例中使用的是 Redis 的单机模式,如果要在生产环境中使用 Redis 集群,需要使用相应的连接池和集群模式进行连接。
二. 基于ZooKeeper的分布式锁
1.名称解释
ZooKeeper 是一个分布式协调服务,提供了一套完善的分布式原语,可以用来实现分布式锁。基于 ZooKeeper 实现的分布式锁主要依赖于其顺序节点和监听机制。
实现步骤:
- 客户端请求获取锁时,创建一个顺序临时节点。
- 客户端获取父节点下所有的子节点,并监视比自己小的那个节点。
- 如果发现自己是最小的节点,则表示获取到了锁,可以执行业务逻辑。
- 如果发现自己不是最小的节点,则等待监听的节点被删除,然后重复步骤 2。
优点:
- 锁的获取和释放是原子操作,能够确保强一致性。
- 支持细粒度的锁。
缺点:
- 实现相对复杂,对 ZooKeeper 的依赖性较强。
- 性能可能不如基于 Redis 的实现方式。
2.代码实现
下面是一个基于 ZooKeeper 的分布式锁的 Java 代码示例:
import org.apache.zookeeper.*;
import java.io.IOException;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.CountDownLatch;
public class ZooKeeperDistributedLock implements Watcher {
private ZooKeeper zooKeeper;
private String rootNode = "/locks";
private String lockName;
private String lockPath;
private CountDownLatch countDownLatch;
public ZooKeeperDistributedLock(String host, int sessionTimeout, String lockName) throws IOException {
this.lockName = lockName;
countDownLatch = new CountDownLatch(1);
zooKeeper = new ZooKeeper(host, sessionTimeout, this);
try {
countDownLatch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
public void lock() {
try {
String lockNode = zooKeeper.create(rootNode + "/" + lockName, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
List<String> nodes = zooKeeper.getChildren(rootNode, false);
Collections.sort(nodes);
if (lockNode.equals(rootNode + "/" + nodes.get(0))) {
// 如果创建的节点是最小的节点,则表示获取到了锁
lockPath = lockNode;
return;
}
String currentNode = lockNode.substring(lockNode.lastIndexOf("/") + 1);
// 获取比当前节点小的最大节点,并设置监听
String preNode = nodes.get(Collections.binarySearch(nodes, currentNode) - 1);
zooKeeper.exists(rootNode + "/" + preNode, new LockWatcher());
} catch (Exception e) {
e.printStackTrace();
}
}
public void unlock() {
try {
zooKeeper.delete(lockPath, -1);
} catch (Exception e) {
e.printStackTrace();
}
}
@Override
public void process(WatchedEvent event) {
if (event.getState() == Event.KeeperState.SyncConnected) {
countDownLatch.countDown();
}
}
private class LockWatcher implements Watcher {
@Override
public void process(WatchedEvent event) {
if (event.getType() == Event.EventType.NodeDeleted) {
lock();
}
}
}
// 示例用法
public static void main(String[] args) {
String host = "localhost:2181";
int sessionTimeout = 5000;
String lockName = "mylock";
try {
ZooKeeperDistributedLock lock = new ZooKeeperDistributedLock(host, sessionTimeout, lockName);
lock.lock();
System.out.println("获取锁成功,执行业务逻辑...");
// 执行业务逻辑
lock.unlock();
} catch (IOException e) {
e.printStackTrace();
}
}
}
这个示例代码创建了一个 ZooKeeperDistributedLock 类,它封装了 ZooKeeper 的连接和操作方法,并提供了获取锁和释放锁的功能。在 lock 方法中,创建了一个临时顺序节点,并获取了父节点下的所有子节点,并对这些子节点进行排序。如果当前创建的节点是最小的节点,则表示获取到了锁;否则,设置对比当前节点小的最大节点的监听器,当该节点被删除时重新尝试获取锁。
需要注意的是,该示例中创建的节点是临时顺序节点,当客户端与 ZooKeeper 断开连接时,该节点会自动删除,从而释放锁。
更多推荐


所有评论(0)