c# 理解csredis库实现分布式锁的详细流程
作者:SZMD.ls.nct
声明:
这里首先使用的是csredis,地址是https://github.com/2881099/csredis
该库本身已经足够完善,这里我画蛇添足一下,为了方便自己的使用。
本身csredis库已经实现了完整的加锁和去锁的逻辑,这里实现的与库本身所实现的有以下几点区别(csredis实现代码位置为:https://github.com/2881099/csredis/blob/bb6d947695770333027f3936f80052041db41b64/src/CSRedisCore/CSRedisClient.cs#L4344,有兴趣可以去了解看下)
1. 去掉了csredis的锁续租部分的功能,尽量简化
2. 将锁的token的设定交给外部,使用guid也罢,使用id也行。通过已知的token,保证了你可以在任意地方以观察者的身份释放锁。
3. 尽量不修改其key的原本值,不添加前缀,防止在观测时出现不必要的麻烦。
逻辑:
加锁就是set 一个 key ,如果key 存在的情况下则返回失败。那么典型的命令就是setnx.
一个锁显然是需要一个过期时间的,那么我们可能要用到 expire命令。
释放锁则是一个del命令
查看锁的值是需要get命令
比较常见的加锁使用的是setnx,不过由于redis支持了SETkeytokenNXEX/PXmax-lock-time(sec/millsec) (设置key token 是否不存在才set 秒数模式/毫秒数模式 秒数或毫秒数) 这种传参模式,由此,这里更加推荐使用set 命令。
如果在我们的代码端执行del 则小概率发生以下情况:
A 申请锁set x,过期时间为t。
经过时间t后,A恰好忙完了,A通过get命令看看token是否一致,得到结果发现一致的。
A决定发送del到redis服务器,此时A恰好网络拥堵。
redis服务器由于锁x超时,进而释放了锁x。
此时B恰好也申请了锁x,无过期时间。
A网络恢复,del命令发送成功。
结果 B的锁被A释放了。
幸好redis支持了lua脚本。让我们得以简单的实现过期,加锁,去锁功能,而不需要自己手动timer过期。
这里要使用到eval命令执行脚本。
代码
using CSRedis; using System; using System.Collections.Concurrent; using System.Collections.Generic; namespace CsRedis.Helper { /// <summary> /// 基于csredis的简单封装 /// </summary> public class CsRedisManager { private ConcurrentDictionary<string, CSRedisClient> _serviceNameWithClient; /// <summary> /// 初始化 /// </summary> public void Init() { _serviceNameWithClient = new ConcurrentDictionary<string, CSRedisClient>(); } /// 获取业务redis服务 /// <param name="serviceName"></param> /// <returns></returns> public CSRedisClient GetRedisClient(string serviceName) CSRedisClient result = null; _serviceNameWithClient.TryGetValue(serviceName,out result); return result; /// 添加redis服务 /// <param name="connectStr"></param> public bool AddRedisClient(string serviceName,string connectStr) CSRedisClient cSRedisClient = new CSRedisClient(connectStr); return _serviceNameWithClient.TryAdd(serviceName, cSRedisClient); /// 设置字符串型kv /// <param name="key">key</param> /// <param name="value">value</param> /// <param name="expireSecond">过期时间(秒)</param> /// <returns>是否成功</returns> public bool Set(string serviceName,string key,string value,int expireSecond=-1) var redisClient = GetRedisClient(serviceName); GetExceptionOfClient(redisClient); return redisClient.Set(key, value, expireSecond); /// 获取相应key的值 /// <param name="key"></param> public string Get(string serviceName, string key) return redisClient.Get(key); /// 如果不存在则执行,存在则忽略 /// <param name="value"></param> public bool SetNx(string serviceName, string key, string value) var res = redisClient.SetNx(key, value); return res; /// 带过期时间的setNx /// <param name="seconds"></param> public bool SetNx(string serviceName, string key, string value, int millSeconds = -1) var res = Set(serviceName, key, value, RedisExistence.Nx, millSeconds); /// 带过期时间的SetXx public bool SetXx(string serviceName, string key, string value, int millSeconds = -1) var res = Set(serviceName, key, value, RedisExistence.Xx, millSeconds); /// 带参数set /// <param name="existence"></param> public bool Set(string serviceName, string key, string value, RedisExistence existence, int millSeconds = -1) var res = redisClient.Set(key, value, millSeconds, existence); /// 设置生存时间 public bool Expire(string serviceName, string key, int seconds) return redisClient.Expire(key, seconds); /// 获取剩余的生存时间(秒) public long Ttl(string serviceName, string key) return redisClient.Ttl(key); /// 删除del public long Del(string serviceName,params string[] keys) return redisClient.Del(keys); /// 执行脚本 /// <param name="script"></param> /// <param name="args"></param> public object Eval(string serviceName, string script,string key,params object[] args) var res = redisClient.Eval(script, key,args); /// 添加共享锁 public bool AddLock(string serviceName, string key,string token, int millSeconds = -1) var valRes = SetNx(serviceName, key, token, millSeconds); return valRes; /// 删除共享锁 public bool ReleaseLock(string serviceName, string key,string token) var script = GetReleaseLockScript(); var res = redisClient.Eval(script, key, token); if (0== (long)res) { return false; } return true; /// 获取键值 /// <param name="pattern"></param> public string[] Keys(string serviceName, string pattern) var res = redisClient.Keys(pattern); /// 获取client发生异常 /// <param name="client"></param> private void GetExceptionOfClient(CSRedisClient client) if (client == null) throw new Exception("无有效的redis服务"); /// lua脚本删除共享锁 /// 解决在A申请锁 xxkey 过期的瞬间,B 申请锁xxkey, /// 此时恰好A执行到释放xxkey从而引起的异常释放 private static string GetReleaseLockScript() return "if redis.call(\"get\",KEYS[1]) == ARGV[1] \nthen\nreturn redis.call(\"del\", KEYS[1])\nelse\nreturn 0\nend"; } }
这里我把要单独执行的lua脚本单独提出来
if redis.call("get",KEYS[1]) == ARGV[1] then return redis.call("del",KEYS[1]) else return 0 end
这段脚本对应的是c# 中的GetReleaseLockScript()方法中的文字。
这里我个人偷了个懒,按照道理,这里应该有个LoadScriptPath,加载脚本所在位置,调用的时候先检查脚本是否在内存中,不在则去LoadScriptPath找对应的脚本,方便不同的人协同合作。不过那个就是脚本管理器了,还要设计interface,有点偏离主题了。
下面是测试代码
using CsRedis.Helper; using NUnit.Framework; namespace TestProject { public class Tests { [SetUp] public void Setup() { } [Test] public void Test1() CsRedisManager csRedisManager = new CsRedisManager(); csRedisManager.Init(); csRedisManager.AddRedisClient("TEST", "127.0.0.1:6379,password=123456, connectTimeout =1000,connectRetry=1,syncTimeout=10000,defaultDatabase=0"); //csRedisManager.AddRedisClient("PRODUCT", "127.0.0.1:6379,password=123456, connectTimeout =1000,connectRetry=1,syncTimeout=10000,defaultDatabase=1"); var token = "123"; var lockKey = "LOCKKEY1"; csRedisManager.AddLock("TEST", lockKey,token,20 * 1000); csRedisManager.ReleaseLock("TEST", lockKey, token); } }
这里就是对于共享锁的一点简单实现,多了挺多与本次的命令无关的代码,海涵海涵
到此这篇关于c# 理解csredis实现分布式锁的文章就介绍到这了,更多相关c# 分布式锁内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!