c# 分布式锁
SZMD.ls.nct 人气:0声明：
这里首先使用的是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);
}
}这里就是对于共享锁的一点简单实现，多了挺多与本次的命令无关的代码，海涵海涵
加载全部内容