scala+redis实现分布式锁的示例代码
1、redis的底层是单例模式，意思是同一个脚本同一时刻只能有一个线程来执行，利用redis的这个特性来实现分布式锁。
首先实现工具类
packageutils
importCacheManager
/**
*redis分布式锁
*/
objectRedisTool{
//加锁是否成功标志
valLOCK_SUCCESS:String="OK"
//即当key不存在时，我们进行set操作；若key已经存在，则不做任何操作；
valSET_IF_NOT_EXIST:String="NX"
//意思是我们要给这个key加一个过期的设置，具体时间由第五个参数决定。
valSET_WITH_EXPIRE_TIME:String="PX"
valRELEASE_SUCCESS:String="1"
/**
*
*@paramlockKey锁
*@paramrequestId请求标识
*@paramexpireTime超期时间
*@paramisPersist临时缓存或者永久缓存
*/
deftryGetDistributedLock(lockKey:String,requestId:String,expireTime:Int,isPersist:Boolean=false){
CacheManager.redisClientPool.withClient(
client=>{
//valredisKeyPrefix=CacheManager.getRedisKeyPrefix(isPersist)
client.select(CacheManager.redisDBNum)
valresult=client.set(lockKey,requestId,SET_IF_NOT_EXIST,SET_WITH_EXPIRE_TIME,expireTime)
varflag=false
if(LOCK_SUCCESS==result){
flag=true
}
flag
}
)
}
/**
*释放分布式锁
*@paramlockKey锁
*@paramrequestId请求标识
*@paramexpireTime超期时间
*@paramisPersist临时缓存或者永久缓存
*@return
*/
defreleaseDistributedLock(lockKey:String,requestId:String,expireTime:Int=10,isPersist:Boolean=false)={
CacheManager.redisClientPool.withClient(
client=>{
valredisKeyPrefix=CacheManager.getRedisKeyPrefix(isPersist)
client.select(CacheManager.redisDBNum)
//lua脚本也是单例模式，同样也可以保证同一时刻只有一个线程执行脚本
vallua=
s"""
|localcurrent=redis.call('incrBy',KEYS[1],ARGV[1]);
|ifcurrent==tonumber(ARGV[1])then
|localt=redis.call('ttl',KEYS[1]);
|ift==-1then
|redis.call('expire',KEYS[1],ARGV[2])
|end;
|end;
|returncurrent;
""".stripMargin
valcode=client.scriptLoad(lua).get
valret=client.evalSHA(code,List(redisKeyPrefix+lockKey),List(requestId,expireTime))
valresult=ret.get.asInstanceOf[Object].toString
varflag=false
if(result==RELEASE_SUCCESS){
flag=true
}
flag
}
)
}
}
2、实现CacheManager类
packageutils
importcom.redis.RedisClientPool
/**
*
*/
objectCacheManager{
valredisClientPool="dev".equalsIgnoreCase(System.getenv("SCALA_ENV"))match{
//开发环境
casetrue=>newRedisClientPool("127.0.0.1",6379)
//其他环境
casefalse=>newRedisClientPool("10.180.x.y",6379,8,0,Some("root"))
}
valredisDBNum=10
defgetRedisKeyPrefix(isPersist:Boolean)={
if(isPersist){
//永久缓存前缀
"persist_"
}else{
//临时缓存前缀
"tmp_"
}
}
}
3、调用锁操作
defupdateTableInfo(param:String)={
varresMap=Map[String,Any]()
vallockKey="mdms.MdmsUtils.updateTableInfo"
//valrequestId=UUID.randomUUID().toString().replace("-","").toUpperCase()
valflag=RedisTool.releaseDistributedLock(lockKey,"1")
if(flag){
try{
//执行你的操作
resMap=Map("code"->200,"msg"->"成功")
}catch{
casee:Exception=>{
e.printStackTrace()
resMap=Map("code"->200101,"msg"->"执行失败")
}
}
}else{
resMap=Map("code"->200102,"msg"->"操作冲突，已经被其他人捷足先登啦。")
}
resMap
}
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。