快捷導(dǎo)航

scala+redis實(shí)現(xiàn)分布式鎖的示例代碼

更新時間：2019年06月19日 10:36:55 作者：張樂1993

這篇文章主要介紹了scala+redis實(shí)現(xiàn)分布式鎖的示例代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

1、redis的底層是單例模式，意思是同一個腳本同一時刻只能有一個線程來執(zhí)行，利用redis的這個特性來實(shí)現(xiàn)分布式鎖。

首先實(shí)現(xiàn)工具類

package utils
 
import CacheManager
 
/**
 * redis分布式鎖
 */
object RedisTool {
 
 //加鎖是否成功標(biāo)志
 val LOCK_SUCCESS:String = "OK"
 
 //即當(dāng)key不存在時，我們進(jìn)行set操作；若key已經(jīng)存在，則不做任何操作；
 val SET_IF_NOT_EXIST:String = "NX"
 
 //意思是我們要給這個key加一個過期的設(shè)置，具體時間由第五個參數(shù)決定。
 val SET_WITH_EXPIRE_TIME:String = "PX"
 
 val RELEASE_SUCCESS:String = "1"
 
 /**
  *
  * @param lockKey   鎖
  * @param requestId  請求標(biāo)識
  * @param expireTime  超期時間
  * @param isPersist  臨時緩存或者永久緩存
  */
 def tryGetDistributedLock(lockKey:String, requestId:String, expireTime:Int,isPersist:Boolean=false){
  CacheManager.redisClientPool.withClient(
   client => {
    //val redisKeyPrefix = CacheManager.getRedisKeyPrefix(isPersist)
    client.select(CacheManager.redisDBNum)
    val result = client.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime)
    var flag = false
    if(LOCK_SUCCESS == result){
     flag = true
    }
    flag
   }
  )
 }
 
 
 /**
  *釋放分布式鎖
  * @param lockKey   鎖
  * @param requestId  請求標(biāo)識
  * @param expireTime  超期時間
  * @param isPersist  臨時緩存或者永久緩存
  * @return
  */
 def releaseDistributedLock(lockKey:String, requestId:String,expireTime: Int = 10,isPersist:Boolean=false) ={
  CacheManager.redisClientPool.withClient(
   client => {
    val redisKeyPrefix = CacheManager.getRedisKeyPrefix(isPersist)
    client.select(CacheManager.redisDBNum)
    //lua腳本也是單例模式，同樣也可以保證同一時刻只有一個線程執(zhí)行腳本
    val lua =
     s"""
       |local current = redis.call('incrBy',KEYS[1],ARGV[1]);
       |if current == tonumber(ARGV[1]) then
       |  local t = redis.call('ttl',KEYS[1]);
       |  if t == -1 then
       |    redis.call('expire',KEYS[1],ARGV[2])
       |  end;
       |end;
       |return current;
      """.stripMargin
    val code = client.scriptLoad(lua).get
    val ret = client.evalSHA(code, List(redisKeyPrefix + lockKey),List(requestId,expireTime))
    val result = ret.get.asInstanceOf[Object].toString
    var flag = false
    if(result == RELEASE_SUCCESS){
     flag = true
    }
    flag
   }
  )
 }
 
}

2、實(shí)現(xiàn)CacheManager類

package utils
 
import com.redis.RedisClientPool
/**
 * 
 */
object CacheManager {
 
 val redisClientPool = "dev".equalsIgnoreCase(System.getenv("SCALA_ENV")) match {
  //開發(fā)環(huán)境
  case true => new RedisClientPool("127.0.0.1", 6379)
  //其他環(huán)境
  case false => new RedisClientPool("10.180.x.y", 6379, 8, 0, Some("root"))
 }
 
 val redisDBNum = 10
 
 def getRedisKeyPrefix(isPersist:Boolean) ={
  if(isPersist){
   //永久緩存前綴
   "persist_"
  }else{
   //臨時緩存前綴
   "tmp_"
  }
 }
 
}

3、調(diào)用鎖操作

def updateTableInfo(param:String) = {
  var resMap = Map[String,Any]()
  val lockKey = "mdms.MdmsUtils.updateTableInfo"
  //val requestId = UUID.randomUUID().toString().replace("-", "").toUpperCase()
  val flag = RedisTool.releaseDistributedLock(lockKey, "1")
  if(flag){
   try{
    
    //執(zhí)行你的操作
    resMap = Map("code" -> 200 ,"msg" -> "成功")
   }catch {
    case e:Exception => {
     
     e.printStackTrace()
     resMap = Map("code" -> 200101 ,"msg" -> "執(zhí)行失敗")
    }
   }
   
  }else{
   resMap = Map("code" -> 200102 ,"msg" -> "操作沖突，已經(jīng)被其他人捷足先登啦。")
  }
  resMap
 }

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: