在Java中使用Redis實現分布式鎖可以通過以下步驟:
引入Redis相關的依賴,例如Jedis或Lettuce。
創建一個Redis連接池或連接工廠,用于獲取Redis連接。
使用Redis連接實例,調用setnx命令(或相應的Redis命令)來設置一個特定的鍵(作為分布式鎖)和對應的值。
檢查setnx命令的返回值,如果返回1表示成功獲取到鎖,否則表示鎖已被其他線程或進程占用。
如果成功獲取到鎖,執行需要加鎖的業務邏輯。
執行完業務邏輯后,調用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()
方法來釋放鎖。