下面是一个使用Java实现的简单分布式锁:
import redis.clients.jedis.Jedis;
import redis.clients.jedis.params.SetParams;
public class DistributedLock {
private static final String LOCK_KEY = "my_lock";
private static final int LOCK_EXPIRE_TIME = 60; // 锁过期时间,单位秒
private Jedis jedis;
public DistributedLock(Jedis jedis) {
this.jedis = jedis;
}
public boolean tryLock() {
SetParams params = SetParams.setParams().nx().ex(LOCK_EXPIRE_TIME);
String result = jedis.set(LOCK_KEY, "locked", params);
return result != null;
}
public void unlock() {
jedis.del(LOCK_KEY);
}
public static void main(String[] args) {
Jedis jedis = new Jedis("localhost");
DistributedLock lock = new DistributedLock(jedis);
// 尝试获取锁
if (lock.tryLock()) {
try {
// 执行需要加锁的代码
System.out.println("Locked");
// 模拟业务代码执行时间
Thread.sleep(5000);
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
// 释放锁
lock.unlock();
System.out.println("Unlocked");
}
} else {
System.out.println("Failed to acquire lock");
}
jedis.close();
}
}
上面的代码实现了一个简单的分布式锁,使用的是Redis作为存储介质。我们通过尝试设置一个指定过期时间的键值对来实现获取锁的功能。
-
首先,我们定义了一个常量
LOCK_KEY
作为锁的键名,以及LOCK_EXPIRE_TIME
作为锁的过期时间。 -
然后,我们创建了一个
DistributedLock
类,用于封装获取和释放锁的操作。 -
在
tryLock
方法中,我们使用 Redis 的SET
命令来尝试设置一个具有过期时间的键值对,并且通过nx()
参数来保证只有当键不存在时才能设置成功。 -
在
unlock
方法中,我们使用 Redis 的DEL
命令来删除锁的键。 -
在
main
方法中,我们创建了一个Jedis
对象来连接 Redis,然后创建一个DistributedLock
对象。 -
接下来,我们尝试获取锁并执行需要加锁的代码。如果获取锁成功,我们就输出 “Locked” 并模拟业务代码的执行时间。
-
最后,无论是否获取到锁,我们都需要释放锁,并输出 “Unlocked”。
需要注意的是,上面的代码只是一个简单的例子,实际的分布式锁实现可能会更复杂。可以根据具体的需求进行相应的扩展和优化。此外,除了 Redis,还可以使用其他的分布式存储系统,如 ZooKeeper 等来实现分布式锁。