如何利用Redis分散式鎖實現控制併發操作
阿新 • • 發佈：2020-10-13
redis命令解釋
說道Redis的分散式鎖都是通過setNx命令結合getset來實現的，在講之前我們先了解下setNx和getset的意思，在redis官網是這樣解釋的
注：redis的命令都是原子操作
SETNX key value
將 key 的值設為 value ，當且僅當 key 不存在。
若給定的 key 已經存在，則 SETNX 不做任何動作。
SETNX 是『SET if Not eXists』(如果不存在，則 SET)的簡寫。
可用版本：
1.0.0+
時間複雜度：
O(1)
返回值：
設定成功，返回 1 。
設定失敗，返回 0 。
redis> EXISTS job # job 不存在 (integer) 0 redis> SETNX job "programmer" # job 設定成功 (integer) 1 redis> SETNX job "code-farmer" # 嘗試覆蓋 job ，失敗 (integer) 0 redis> GET job # 沒有被覆蓋 "programmer"
GETSET key value
將給定 key 的值設為 value ，並返回 key 的舊值(old value)。
當 key 存在但不是字串型別時，返回一個錯誤。
可用版本：
1.0.0+
時間複雜度：
O(1)
返回值：
返回給定 key 的舊值。
當 key 沒有舊值時，也即是， key 不存在時，返回 nil 。
redis> GETSET db mongodb # 沒有舊值，返回 nil (nil) redis> GET db "mongodb" redis> GETSET db redis # 返回舊值 mongodb "mongodb" redis> GET db "redis"
程式碼示例
注意：為了讓分散式鎖的演算法更穩鍵些，持有鎖的客戶端在解鎖之前應該再檢查一次自己的鎖是否已經超時，再去做DEL操作，因為可能客戶端因為某個耗時的操作而掛起，操作完的時候鎖因為超時已經被別人獲得，這時就不必解鎖了。
我們看下程式碼涉及以下幾個類，這裡有關業務邏輯相關的只定義了方法沒有具體實現，關鍵是學習思路
OrderBiz.java
/**
* 使用redis鎖來控制併發搶單
* @author fuyuwei
*/
public class OrderBiz {
public int createOrder(){
// 下單之前的引數、合法性校驗這裡就不在演示
OrderLock<Boolean> orderLock = new RedisOrderLock<Boolean>("pro-12345678901");
boolean isSyn = orderLock.isSyn(new OrderLockBiz<Boolean>(){
@Override
public Boolean createOrder() {
// 省去建立訂單邏輯
return null;
}
});
if(!isSyn){
BizLogger.info("建立訂單失敗");
}
return 0;
}
}
OrderLock.java
public interface OrderLock<T> {
public boolean isSyn(OrderLockBiz<T> orderBiz);
}
OrderLockBiz.java
public interface OrderLockBiz<T> {
public T createOrder();
}
RedisOrderLock.java
public class RedisOrderLock<T> implements OrderLock<T> {
// 鎖等待超時，防止執行緒飢餓，永遠沒有入鎖執行程式碼的機會
public static final long timeout = 10000;//ms
// 鎖持有超時，防止執行緒在入鎖以後，無限的執行下去，讓鎖無法釋放
public static final long expireMsecs = 10000;// ms
public String lockKey = "orderLockKey";
public Jedis jedis;
private static volatile JedisPool jedisPool;
public RedisOrderLock(String lockKey) {
this.lockKey = lockKey;
}
/**
* 初始化redis
* @return
*/
public Jedis getInstance() {
if(jedisPool == null) {
synchronized(RedisOrderLock.class) {
if(jedisPool == null) {
JedisPoolConfig config = new JedisPoolConfig();
config.setMaxIdle(100);
jedisPool = new JedisPool(config,"localhost",6379,3000,"test");
}
}
}
return jedisPool.getResource();
}
/**
* 執行緒安全的業務邏輯處理
*/
@Override
public boolean isSyn(OrderLockBiz<T> orderBiz) {
jedis = this.getInstance();
try {
// 獲取到鎖
if(acquire(jedis)){
// 執行建立訂單邏輯
orderBiz.createOrder();
}else{
BizLogger.info("waiting other thread creating");
}
} catch (Exception e) {
BizLogger.error(e,"acquire lock failre");
}finally{
// 解鎖
this.releaseLock(jedis);
}
return false;
}
/**
* accqure lock
* @param jedis
* @return
* @throws InterruptedException
*/
public synchronized boolean acquire(Jedis jedis){
boolean locked = false;
while(timeout > 0){
long expires = System.currentTimeMillis() + expireMsecs + 1;
// 10秒之後鎖到期
String expiresStr = String.valueOf(expires);
// 獲取到鎖
if(jedis.setnx(lockKey,expiresStr) == 1){
locked = true;
return locked;
}
// 沒有獲取到鎖
String oldValue = jedis.get(lockKey);
// expireMsecs（10秒）鎖的有效期內無法進入if判斷，如果鎖超時了
if(oldValue != null
&& Long.parseLong(oldValue) < System.currentTimeMillis()){
// 如果鎖超時重新設定
String oldValue_ = jedis.getSet(lockKey,expiresStr);
// 值相同說明是同一個執行緒的操作，獲取鎖成功
if(Long.valueOf(oldValue_) == Long.valueOf(oldValue)){
locked = true;
}else{
// 被其他執行緒搶先獲取鎖
locked = false;
}
}
// 鎖沒有超時，繼續等待
return false;
}
}
/**
* 釋放鎖
* @param jedis
*/
public synchronized void releaseLock(Jedis jedis){
try {
long current = System.currentTimeMillis();
// 避免刪除非自己獲取得到的鎖
if (current < Long.valueOf(jedis.get(lockKey)))
jedis.del(lockKey);
} catch (Exception e) {
e.printStackTrace();
}finally{
// 把用完的連線放到連線池匯中供其他執行緒呼叫
jedisPool.returnResource(jedis);
}
}
}
以上這篇如何利用Redis分散式鎖實現控制併發操作就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。