如何利用Redis分布式鎖實現(xiàn)控制并發(fā)操作

更新時間：2020年09月30日 10:29:16 作者：拉里·佩奇

這篇文章主要介紹了如何利用Redis分布式鎖實現(xiàn)控制并發(fā)操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

redis命令解釋

說道Redis的分布式鎖都是通過setNx命令結合getset來實現(xiàn)的，在講之前我們先了解下setNx和getset的意思，在redis官網(wǎng)是這樣解釋的

注：redis的命令都是原子操作

SETNX key value

將 key 的值設為 value ，當且僅當 key 不存在。

若給定的 key 已經(jīng)存在，則 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"

代碼示例

注意：為了讓分布式鎖的算法更穩(wěn)鍵些，持有鎖的客戶端在解鎖之前應該再檢查一次自己的鎖是否已經(jīng)超時，再去做DEL操作，因為可能客戶端因為某個耗時的操作而掛起，操作完的時候鎖因為超時已經(jīng)被別人獲得，這時就不必解鎖了。

我們看下代碼涉及以下幾個類，這里有關業(yè)務邏輯相關的只定義了方法沒有具體實現(xiàn)，關鍵是學習思路

OrderBiz.java

/**
 * 使用redis鎖來控制并發(fā)搶單
 * @author fuyuwei
 */
public class OrderBiz {
 public int createOrder(){
  // 下單之前的參數(shù)、合法性校驗這里就不在演示
  OrderLock<Boolean> orderLock = new RedisOrderLock<Boolean>("pro-12345678901");
  boolean isSyn = orderLock.isSyn(new OrderLockBiz<Boolean>(){
   @Override
   public Boolean createOrder() {
    // 省去創(chuàng)建訂單邏輯
    return null;
   }
  });
  if(!isSyn){
   BizLogger.info("創(chuàng)建訂單失敗");
  }
  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> {

 // 鎖等待超時，防止線程饑餓，永遠沒有入鎖執(zhí)行代碼的機會 
 public static final long timeout = 10000;//ms

 // 鎖持有超時，防止線程在入鎖以后，無限的執(zhí)行下去，讓鎖無法釋放 
 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();
 }

 /**
  * 線程安全的業(yè)務邏輯處理
  */
 @Override
 public boolean isSyn(OrderLockBiz<T> orderBiz) {
  jedis = this.getInstance();
  try {
   // 獲取到鎖
   if(acquire(jedis)){
    // 執(zhí)行創(chuàng)建訂單邏輯
    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;
    }
   }
   // 鎖沒有超時，繼續(xù)等待
   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分布式鎖實現(xiàn)控制并發(fā)操作就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: