java实现分布式锁的代码怎么写

42次阅读

共计 1582 个字符，预计需要花费 4 分钟才能阅读完成。

在 Java 中实现分布式锁可以使用以下几种方式：
1. 基于数据库的实现：
– 创建一个数据库表，表中有一个唯一索引字段用于标识锁的名称。
– 在获取锁时，向数据库中插入一条记录，如果插入成功则表示获取锁成功。
– 在释放锁时，删除对应的记录。
2. 基于 Redis 的实现：
– 使用 Redis 的 setnx 命令（set if not exist）尝试获取锁，如果返回 1 则表示获取锁成功。
– 获取锁成功后设置一个过期时间，防止锁一直被占用。
– 在释放锁时，删除对应的 key。
下面是一个基于 Redis 的分布式锁的示例代码：
“`java
import redis.clients.jedis.Jedis;
public class DistributedLock {
private static final String LOCK_KEY = “distributed_lock”;
private static final int LOCK_EXPIRE_TIME = 30000;
private Jedis jedis;
private boolean locked = false;
public DistributedLock(Jedis jedis) {
this.jedis = jedis;
}
public boolean tryLock() {
long now = System.currentTimeMillis();
long expireTime = now + LOCK_EXPIRE_TIME + 1;
String result = jedis.set(LOCK_KEY, String.valueOf(expireTime), “NX”, “PX”, LOCK_EXPIRE_TIME);
if (“OK”.equals(result)) {
locked = true;
return true;
}
return false;
}
public void unlock() {
if (locked) {
jedis.del(LOCK_KEY);
locked = false;
}
}
}
“`
使用示例：
“`java
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
public class Main {
public static void main(String[] args) {
JedisPoolConfig poolConfig = new JedisPoolConfig();
JedisPool jedisPool = new JedisPool(poolConfig, “localhost”, 6379);
Jedis jedis = jedisPool.getResource();
DistributedLock lock = new DistributedLock(jedis);
try {
if (lock.tryLock()) {
// 获取锁成功，执行业务逻辑
System.out.println(“ 获取锁成功 ”);
} else {
// 获取锁失败，执行其他逻辑
System.out.println(“ 获取锁失败 ”);
}
} finally {
lock.unlock();
}
jedis.close();
jedisPool.close();
}
}
“`
注意：上述代码仅为示例，实际使用时需要考虑异常处理、锁重入等情况。