redis分佈式加鎖解鎖

public class RedisUtil {
    private static JedisPool jedisPool;
    private static final String LOCK_SUCCESS = "OK";
    private static final String SET_IF_NOT_EXIST = "NX";
    private static final String SET_WITH_EXPIRE_TIME = "PX";
    private static final Long RELEASE_SUCCESS = 1L;

    public RedisUtil(JedisPoolConfig config) {
        if (jedisPool == null) {
            jedisPool = new JedisPool(config, "localhost", 6379, 2000);
        }
    }

    public RedisUtil(JedisPoolConfig config, String host, int port) {
        if (jedisPool == null) {
            jedisPool = new JedisPool(config, host, port, 2000);
        }
    }

    public RedisUtil(JedisPoolConfig config, String host, int port, int timeout) {
        if (jedisPool == null) {
            jedisPool = new JedisPool(config, host, port, timeout);
        }
    }

    public RedisUtil(JedisPoolConfig config, String host, int port,
                     int timeout, String password) {
        if (jedisPool == null) {
            jedisPool = new JedisPool(config, host, port, timeout, password);
        }
    }

    public RedisUtil(JedisPool pool) {
        if (RedisUtil.jedisPool == null) {
            RedisUtil.jedisPool = pool;
        }
    }
/**
 * 獲取分佈式鎖
 * @param lockKey
 * @param requestId
 * @param expireTime
 * @return
 */
public static boolean getDistributedLock( String lockKey, String requestId, int expireTime) {
    Jedis jedis = null;
    String result="";
    boolean success=false;
    try {
        jedis = jedisPool.getResource();
        result = jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
        if (LOCK_SUCCESS.equals(result)) {
           success=true;
        }
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        jedis.close();
    }
    return success;
}

/**
 * 釋放分佈式鎖
 * @param lockKey
 * @param requestId
 * @return
 */
public static boolean releaseDistributedLock(String lockKey, String requestId) {
    Jedis jedis = null;
    Object result=null;
    boolean success=false;
    String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
    try {
        jedis = jedisPool.getResource();
        result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));
        if (RELEASE_SUCCESS.equals(result)) {
            success=true;
        }
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        jedis.close();
    }
    return success;
}
}

加鎖操做解析:redis

  • 第一個爲key,咱們使用key來當鎖,由於key是惟一的。dom

  • 第二個爲value,咱們傳的是requestId,不少童鞋可能不明白,有key做爲鎖不就夠了嗎,爲何還要用到value?緣由就是咱們在上面講到可靠性時,分佈式鎖要知足第四個條件解鈴還須繫鈴人,經過給value賦值爲requestId,咱們就知道這把鎖是哪一個請求加的了,在解鎖的時候就能夠有依據。requestId可使用UUID.randomUUID().toString()方法生成。分佈式

  • 第三個爲nxxx,這個參數咱們填的是NX,意思是SET IF NOT EXIST,即當key不存在時,咱們進行set操做;若key已經存在,則不作任何操做;.net

  • 第四個爲expx,這個參數咱們傳的是PX,意思是咱們要給這個key加一個過時的設置,具體時間由第五個參數決定。code

  • 第五個爲time,與第四個參數相呼應,表明key的過時時間。blog

總的來講,執行上面的set()方法就只會致使兩種結果:1. 當前沒有鎖(key不存在),那麼就進行加鎖操做,並對鎖設置個有效期,同時value表示加鎖的客戶端。2. 已有鎖存在,不作任何操做。ip

解鎖操做解析:get

能夠看到,咱們解鎖只須要兩行代碼就搞定了!第一行代碼,咱們寫了一個簡單的Lua腳本代碼,第二行代碼,咱們將Lua代碼傳到jedis.eval()方法裏,並使參數KEYS[1]賦值爲lockKey,ARGV[1]賦值爲requestId。eval()方法是將Lua代碼交給Redis服務端執行。io

那麼這段Lua代碼的功能是什麼呢?其實很簡單,首先獲取鎖對應的value值,檢查是否與requestId相等,若是相等則刪除鎖(解鎖)。那麼爲何要使用Lua語言來實現呢?由於要確保上述操做是原子性的。那麼爲何執行eval()方法能夠確保原子性,源於Redis的特性,下面是官網對eval命令的部分解釋:class

簡單來講,就是在eval命令執行Lua代碼的時候,Lua代碼將被當成一個命令去執行,而且直到eval命令執行完成,Redis纔會執行其餘命令。

參考:https://blog.csdn.net/zht741322694/article/details/79290822

相關文章
相關標籤/搜索