写一个关于redis的分布式锁的工具类
时间: 2023-07-25 11:09:34 浏览: 104
Redis分布式锁.zip
好的,下面是一个简单的Java代码实现:
```java
import redis.clients.jedis.Jedis;
import redis.clients.jedis.params.SetParams;
public class RedisDistributedLock {
private Jedis jedis;
public RedisDistributedLock(Jedis jedis) {
this.jedis = jedis;
}
public boolean acquireLock(String lockKey, String lockValue, int expireTime) {
SetParams params = new SetParams().nx().ex(expireTime);
String result = jedis.set(lockKey, lockValue, params);
return "OK".equals(result);
}
public boolean releaseLock(String lockKey, String lockValue) {
String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
Object result = jedis.eval(script, 1, lockKey, lockValue);
return Long.valueOf(1).equals(result);
}
}
```
这个工具类基于 Redis 实现了一个简单的分布式锁。它有两个方法:`acquireLock` 和 `releaseLock`。使用 `acquireLock` 方法可以获取锁,使用 `releaseLock` 方法可以释放锁。
在 `acquireLock` 方法中,我们使用了 Redis 的 SET NX EX 命令来获取锁。如果这个 key 不存在,那么就会设置这个 key 的值为 lockValue,并且设置过期时间为 expireTime 秒。这里的 NX 表示只有当这个 key 不存在时才能设置成功,防止多个线程同时获取锁。
在 `releaseLock` 方法中,我们使用了 Redis 的 EVAL 命令来删除锁。这个命令可以执行一段 Lua 脚本。脚本中首先判断这个 key 的值是否等于 lockValue,如果相等,那么就删除这个 key。这里的 ARGV[1] 表示第一个参数,也就是 lockValue。
使用这个工具类的方法如下:
```java
Jedis jedis = new Jedis("localhost", 6379);
RedisDistributedLock lock = new RedisDistributedLock(jedis);
String lockKey = "myLock";
String lockValue = UUID.randomUUID().toString();
int expireTime = 60;
boolean success = lock.acquireLock(lockKey, lockValue, expireTime);
try {
if (success) {
// 获取锁成功,执行业务逻辑
} else {
// 获取锁失败,执行其他逻辑
}
} finally {
lock.releaseLock(lockKey, lockValue);
}
```
在这个例子中,我们首先创建了一个 Jedis 实例,然后使用这个实例创建了一个 RedisDistributedLock 实例。接着,我们定义了锁的 key、锁的值和过期时间。然后,我们调用了 `acquireLock` 方法来获取锁,并且在 try-finally 块中执行了业务逻辑和释放锁的操作。如果获取锁失败,那么就不会执行业务逻辑。
阅读全文