package com.realize;
import redis.clients.jedis.Jedis;
import java.util.UUID;
import redis.clients.jedis.params.SetParams;
public class DistributedLock {
private Jedis jedis;
private String lockKey;
private String lockValue;
private int expireTime;
public DistributedLock(Jedis jedis, String lockKey, int expireTime) {
this.jedis = jedis;
this.lockKey = lockKey;
this.expireTime = expireTime;
}
// 获取锁
public boolean acquireLock() {
this.lockValue = UUID.randomUUID().toString();
//String result = jedis.set(lockKey, lockValue, "NX", "EX", expireTime);
String result = jedis.set(lockKey, lockValue,SetParams.setParams().nx().ex(10L));
return "OK".equals(result);
}
// 释放锁
public boolean releaseLock() {
String luaScript =
"if redis.call('get', KEYS[1]) == ARGV[1] then " +
"return redis.call('del', KEYS[1]) " +
"else " +
"return 0 " +
"end";
Object result = jedis.eval(luaScript, 1, lockKey, lockValue);
return result.equals(1L);
}
public static void main(String[] args) {
Jedis jedis = new Jedis("localhost", 6379);
DistributedLock lock = new DistributedLock(jedis, "resource_lock", 10);
if (lock.acquireLock()) {
try {
System.out.println("Lock acquired, doing some work...");
// 执行业务逻辑
} finally {
lock.releaseLock();
System.out.println("Lock released");
}
} else {
System.out.println("Failed to acquire lock");
}
jedis.close();
}
}
-
注意事项
- 唯一标识符:使用唯一标识符来标识每个持有锁的客户端,防止锁被错误释放。
- 过期时间:设置合理的过期时间,防止死锁。如果客户端在持有锁期间崩溃,过期时间可以自动释放锁。
- 重试机制:获取锁时可以设置重试机制,防止短暂的争用导致获取锁失败。
-
使用场景
SET NX
命令通常用于实现分布式锁,以确保在分布式环境中,同一时间只有一个客户端可以持有锁,从而避免并发访问导致的数据不一致问题。 -
获取锁:
lockValue:生成一个唯一标识符,作为锁的值。
jedis.set(lockKey, lockValue, "NX", "EX", expireTime):尝试使用 SET NX 命令设置锁。只有在 lockKey 不存在时,才能成功设置该键,并指定过期时间 expireTime。