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