基於redis的setnx()、get()、getset()方法 分布式鎖

一.redis命令講解：
setnx()命令：
setnx的含義就是SET if Not Exists，其主要有兩個參數 setnx(key, value)。

該方法是原子的，如果key不存在，則設定當前key成功，返回1；如果當前key已經存在，則設定當前key失敗，返回0。

get()命令：
get(key) 擷取key的值，如果存在，則返回；如果不存在，則返回nil；
getset()命令：
這個命令主要有兩個參數 getset(key, newValue)。該方法是原子的，對key設定newValue這個值，並且返回key原來的舊值。
假設key原來是不存在的，那麼多次執行這個命令，會出現下邊的效果：
1. getset(key, “value1”) 返回nil 此時key的值會被設定為value1
2. getset(key, “value2”) 返回value1 此時key的值會被設定為value2
3. 依次類推。
二.具體的使用步驟如下：
1. setnx(lockkey, 目前時間+到期逾時時間) ，如果返回1，則擷取鎖成功；如果返回0則沒有擷取到鎖，轉向2。
2. get(lockkey)擷取值oldExpireTime ，並將這個value值與當前的系統時間進行比較，如果小於當前系統時間，則認為這個鎖已經逾時，可以允許別的請求重新擷取，轉向3。
3. 計算newExpireTime=目前時間+到期逾時時間，然後getset(lockkey, newExpireTime) 會返回當前lockkey的值currentExpireTime。
4. 判斷currentExpireTime與oldExpireTime 是否相等，如果相等，說明當前getset設定成功，擷取到了鎖。如果不相等，說明這個鎖又被別的請求擷取走了，那麼當前請求可以直接返回失敗，或者繼續重試。
5. 在擷取到鎖之後，當前線程可以開始自己的業務處理，當處理完畢後，比較自己的處理時間和對於鎖設定的逾時時間，如果小於鎖設定的逾時時間，則直接執行delete釋放鎖；如果大於鎖設定的逾時時間，則不需要再鎖進行處理。

具體代碼如下：

實現分布式鎖DistributedLockHandler類：

[java] view plain copy
package tk.mybatis.springboot.distributedLock;

import org.springframework.stereotype.Service;

import redis.clients.jedis.Jedis;

@Service(“distributedLockHandler”)
public class DistributedLockHandler {

private static final Integer Lock_Timeout = 3;  private Jedis jedis;  /**  * 外部調用加鎖的方法  * @param lockKey 鎖的名字  * @param timeout 逾時時間（放置時間長度，如：5L）  * @return  */  public boolean tryLock(String lockKey, Long timeout) {      try {          Long currentTime = System.currentTimeMillis();//開始加鎖的時間          boolean result = false;          while (true) {              if ((System.currentTimeMillis() - currentTime) / 1000 > timeout) {//目前時間超過了設定的逾時時間                  System.out.println("Execute DistributedLockHandler.tryLock method, Time out.");                  break;              } else {                  result = innerTryLock(lockKey);                  if (result) {                      break;                  } else {                      System.out.println("Try to get the Lock,and wait 100 millisecond....");                      Thread.sleep(100);                  }              }          }          return result;      } catch (Exception e) {          System.out.println("Failed to run DistributedLockHandler.getLock method."+ e);          return false;      }  }  /**  * 釋放鎖  * @param lockKey 鎖的名字  */  public void realseLock(String lockKey) {      if(!checkIfLockTimeout(System.currentTimeMillis(), lockKey)){          jedis.del(lockKey);      }  }  /**  * 內部擷取鎖的實現方法  * @param lockKey 鎖的名字  * @return  */  private boolean innerTryLock(String lockKey) {      long currentTime = System.currentTimeMillis();//目前時間      String lockTimeDuration = String.valueOf(currentTime + Lock_Timeout + 1);//鎖的期間      Long result = jedis.setnx(lockKey, lockTimeDuration);      if (result == 1) {          return true;      } else {          if (checkIfLockTimeout(currentTime, lockKey)) {              String preLockTimeDuration = jedis.getSet(lockKey, lockTimeDuration);              if (currentTime > Long.valueOf(preLockTimeDuration)) {                  return true;              }          }          return false;      }  }  /**  * 判斷加鎖是否逾時  * @param currentTime 目前時間  * @param lockKey 鎖的名字  * @return  */  private boolean checkIfLockTimeout(Long currentTime, String lockKey) {      if (currentTime > Long.valueOf(jedis.get(lockKey))) {//目前時間超過鎖的期間          return true;      } else {          return false;      }  }  public DistributedLockHandler setJedis(Jedis jedis) {      this.jedis = jedis;      return this;  }  

}

調用Demo類：

[java] view plain copy
package tk.mybatis.springboot.distributedLock;

import redis.clients.jedis.Jedis;

/**
* 基於redis的setnx()、get()、getset()方法 分布式鎖
* @author KF01
*
*/
public class Demo {
private static final String lockKey = “Lock.TecentIm_Interface_Counter”;

public static void main(String[] args) {      Jedis jedis = new Jedis("127.0.0.1", 6379);      DistributedLockHandler distributedLockHandler = new DistributedLockHandler().setJedis(jedis);      try{          boolean getLock = distributedLockHandler.tryLock(lockKey, Long.valueOf(5));          if(getLock){              // Do your job              System.out.println("Do your job........");          }      }catch(Exception e){          System.out.println(e);      }finally {          distributedLockHandler.realseLock(lockKey);      }  }  

}