RedisTemplate用SETNX命令實現分佈式鎖

使用SETNX命令獲取分佈式鎖的步驟：

• C1和C2線程同時檢查時間戳獲取鎖，執行SETNX命令並都返回0，此時鎖仍被C3持有，而且C3已經崩潰
• C1 DEL鎖
• C1 使用SETNX命令獲取鎖，而且成功
• C2 DEL鎖
• C2 使用SETNX命令獲取鎖，而且成功
• ERROR : 因爲競態條件，C1和C2都獲取到了鎖

幸運的是，如下面的步驟徹底能夠避免這種狀況發生，看看C4線程如何操做

• C4使用SETNX命令獲取鎖
• C3已經崩潰可是仍然持有鎖，因此Redis返回0給C4
• C4使用GET命令獲取鎖並檢查鎖是否已通過期，若是沒有過時，則繼續等待一段時間並從新重試
• 若是鎖已通過期，C4嘗試 GETSET lock.foo <current Unix timestamp + lock timeout + 1>
• 利用GETSET語法，C4能夠檢查舊時間是否仍然是過時時間，若是是，則獲取鎖
• 若是另外一個客戶端C5率先獲取到鎖，C4執行GETSET命令後將返回非過時時間，而後C4繼續從頭開始從新嘗試獲取鎖。此操做C4將延長一點C5獲取到的鎖的過時時間，不過這不是什麼大問題。

本次demo使用的是 Spring Boot 進行的單元測試；歡迎討論；

package com.example.demo.controller; import org.junit.Test; import org.junit.runner.RunWith; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.data.redis.core.RedisCallback; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.test.context.junit4.SpringRunner; import java.util.Objects; /** * Description: demo1 <br> * * @author Liang lp * Date: 2019/12/13 16:41 <br> */ @SpringBootTest @RunWith(SpringRunner.class) public class demo1 { private static final Logger log = LoggerFactory.getLogger(demo1.class); //鎖名稱前綴
    public static final String LOCK_PREFIX = "redis_lock"; //加鎖失效時間，毫秒
    public static final int LOCK_EXPIRE = 600; // ms
 @Autowired RedisTemplate redisTemplate; /** * 使用進行測試 */ @Test public void demo() { //設置個Key
        String key = "order_id"; try { boolean b = lock(LOCK_PREFIX + key); if (b) { log.info("開始執行業務邏輯"); Thread.sleep(2000); } else { log.info("獲取鎖錯誤{}", b); return; } } catch (Exception e) { log.info("獲取鎖異常{}", e); } finally { //刪除鎖；
 deleteLock(LOCK_PREFIX + key); } } /** * 得到鎖 * * @param lock * @return
     */
    public boolean lock(String lock) { return (boolean) redisTemplate.execute((RedisCallback) connection -> { //獲取時間毫秒值
            long expireAt = System.currentTimeMillis() + LOCK_EXPIRE + 1; //獲取鎖
            Boolean acquire = connection.setNX(lock.getBytes(), String.valueOf(expireAt).getBytes()); if (acquire) { return true; } else { byte[] bytes = connection.get(lock.getBytes()); //非空判斷
                if (Objects.nonNull(bytes) && bytes.length > 0) { long expireTime = Long.parseLong(new String(bytes)); // 若是鎖已通過期
                    if (expireTime < System.currentTimeMillis()) { // 從新加鎖，防止死鎖
                        byte[] set = connection.getSet(lock.getBytes(), String.valueOf(System.currentTimeMillis() + LOCK_EXPIRE + 1).getBytes()); return Long.parseLong(new String(set)) < System.currentTimeMillis(); } } } return false; }); } /** * 刪除鎖 * * @param key */
    public void deleteLock(String key) { redisTemplate.delete(key); } }