使用 redistemplate 實現鎖的方案

springboot + redistemplate 實現鎖的方案

 

• 1、通過set命令設置鎖
• 2、判斷返回結果是否是OK
  • 1）Nil，獲取失敗，結束或重試（自旋鎖）
  • 2）OK，獲取鎖成功
    • 執行業務
    • 釋放鎖，DEL 刪除key即可
• 3、異常情況，服務宕機。超時時間EX結束，會自動釋放鎖

 

代碼清晰簡潔 廢話不多說直接上demo

這里使用的單元測試；歡迎研究討論，下一篇講解分布式鎖；

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.RedisTemplate;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.concurrent.TimeUnit;

/**
 * Description: demo <br>
 *
 * @author Liang lp
 * Date: 2019/12/13 12:27 <br>
 */
@SpringBootTest
@RunWith(SpringRunner.class)
public class demo {

    private static final Logger log = LoggerFactory.getLogger(demo.class);

    @Autowired
    RedisTemplate redisTemplate;

    @Test
    public void test() {
        String key = "test";
        //創建鎖
        boolean isLock = lock(key, 1, 50);
        // 判斷是否獲取鎖
        if (!isLock) {
            //獲取所失敗
            log.info("獲取所失敗 ");
            return;
        }
        try {
            log.info("獲取鎖成功，開始執行邏輯");
            //模擬程序執行
            Thread.sleep(2000);
        } catch (Exception e) {
            log.error("程序執行異常{}", e);
        } finally {
            // 釋放鎖
            deleteLock(key);
            log.info("執行完畢。釋放鎖完成；");
        }
    }

    /**
     * 創建鎖
     *
     * @param key         鎖的Key
     * @param value       值(隨便寫毫無意義)
     * @param releaseTime 鎖過期時間 防止死鎖
     * @return
     */
    public boolean lock(String key, int value, long releaseTime) {
        // 嘗試獲取鎖
        Boolean boo = redisTemplate.opsForValue().setIfAbsent(key, value, releaseTime, TimeUnit.SECONDS);
        // 判斷結果
        return boo != null && boo;
    }

    /**
     * 根據key'刪除鎖
     *
     * @param key
     */
    public void deleteLock(String key) {
        // 刪除key即可釋放鎖
        redisTemplate.delete(key);
    }
}