Java

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

小亿
92
2023-10-23 16:17:15
栏目: 编程语言

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

0
看了该问题的人还看了