在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();
}
}
注意:上述代码仅为示例,实际使用时需要考虑异常处理、锁重入等情况。