Java で Redis を使用して分散ロックを実装するには

2年 ago
蓮, 翼
1 minute

ネイティブなJavaでRedisを使用した分散ロックの実装は、以下の手順で行えます。

  1. JedisやLettuceなどのRedis関連の依存関係を導入します。
  2. Redis接続を取得するためのRedis接続プールまたは接続ファクトリを作成します。
  3. Redis接続インスタンスを利用し、setnxコマンド（あるいはそれに相当するRedisコマンド）を呼び出して、特定のキー（分散ロック）とそれに対応する値を設定する。
  4. setnx コマンドの戻り値を確認し、1が返された場合はロックの取得に成功したことを示し、それ以外の場合はロックが他のスレッドまたはプロセスによってすでに使用されていることを示しています。
  5. ロックの取得に成功したら、ロックが必要な業務ロジックを実行する。
  6. ビジネスロジックの実行後、del コマンド（または対応する Redis コマンド）を呼び出してロックを解放します。

分散ロックをJedisで実装する例を次に示します。

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
public class DistributedLockExample {
private static final String LOCK_KEY = "my_lock_key";
private static final int LOCK_EXPIRE_TIME = 30000; // 锁过期时间（毫秒）
private static final int ACQUIRE_TIMEOUT = 5000; // 获取锁超时时间（毫秒）
private JedisPool jedisPool;
public DistributedLockExample(JedisPool jedisPool) {
this.jedisPool = jedisPool;
}
public boolean acquireLock() {
try (Jedis jedis = jedisPool.getResource()) {
long startTime = System.currentTimeMillis();
while (true) {
// 尝试获取锁
String result = jedis.set(LOCK_KEY, "locked", "NX", "PX", LOCK_EXPIRE_TIME);
if ("OK".equals(result)) {
return true; // 成功获取到锁
}
// 获取锁超时
if (System.currentTimeMillis() - startTime > ACQUIRE_TIMEOUT) {
return false;
}
// 等待一段时间后重试
Thread.sleep(100);
}
} catch (Exception e) {
// 处理异常
}
return false;
}
public void releaseLock() {
try (Jedis jedis = jedisPool.getResource()) {
jedis.del(LOCK_KEY);
} catch (Exception e) {
// 处理异常
}
}
// 使用示例
public static void main(String[] args) {
JedisPool jedisPool = new JedisPool("localhost", 6379);
DistributedLockExample lockExample = new DistributedLockExample(jedisPool);
if (lockExample.acquireLock()) {
try {
// 执行需要加锁的业务逻辑
} finally {
lockExample.releaseLock();
}
} else {
// 获取锁失败
}
}
}

上に示した例のコードでは、Jedis の接続プールを使用して Redis 接続を取得し、jedis.set(LOCK_KEY, “locked”, “NX”, “PX”, LOCK_EXPIRE_TIME) メソッドを呼び出すことでロックを取得します。ロックの取得に成功した場合、ロックをかける必要があるビジネスロジックを実行し、そうでない場合は少し待ってから再試行します。最後に finally ブロックで、lockExample.releaseLock() メソッドを呼び出してロックを解放します。

bannerAds