1、背景
我们在开发很多业务场景会使用到锁,例如库存控制,抽奖等。一般我们会使用内存锁的方式来保证线性的执行。但现在大多站点都会使用分布式部署,那多台服务器间的就必须使用同一个目标来判断锁。分布式与单机情况下最大的不同在于其不是多线程而是多进程。
2、演变
[分布式站点使用内存锁方式如下图]
假设有3个用户同时购买一件商品,商品库存只剩下1,如果3个用户同时购买,负载均衡把3个用户分别指向站点1、2、3,那结果将会是3个用户都购买成功。下面我们使用分布式锁解决这个问题。
[分布式站点使用分布式锁如下图]
单台服务器由于可以共享堆内存,因此可以简单的采取内存作为标记存储位置。而多服务器之间都不在同一台物理机上,因此需要将标记存储在一个所有进程都能看到的地方。
3、实现
选型
想想我们实现分布式锁要满足哪些条件?
1、在分布式系统环境下,一个锁在同一时间只能被一个服务器获取;(这是所有分布式锁的基础)
2、高性能的获取锁和释放锁;(锁用完了,要及时释放,以供别人继续使用)
3、具备锁失效机制,防止死锁;(防止因为某些意外,锁没有得到释放,那别人将永远无法使用)
4、具备非阻塞锁特性,即没有获取到锁将直接返回获取锁失败。(满足等待锁的同时,也要满足非阻塞锁特性,便于多样性的业务场景使用)
分布式锁有很多的实现方式:数据库排他锁、Zookeeper、缓存(redis、memcached)等,本文选用Redis实现。原因如下:
1、Redis有很高的性能;
2、Redis命令对此支持较好,实现起来比较方便;
Redis官网上对C#的使用推荐有ServiceStack.Redis和StackExchange.Redis
由于StackExchange.Redis虽然有提供LockTake方法,很方便的使用锁,但是只支持.Net4.5以上。公司很多站点都是3.5和4.0的,所以选用ServiceStack.Redis,自己封装一下。
Redis使用命令介绍
1、SETNX
SETNX key val:当且仅当key不存在时,set一个key为val的字符串,返回1;若key存在,则什么都不做,返回0。
2、expire
expire key timeout:为key设置一个超时时间,单位为second,超过这个时间锁会自动释放,避免死锁。
3、delete
delete key:删除key
在使用Redis实现分布式锁的时候,主要就会使用到这三个命令。
C#代码
1、锁方法
/// <summary> /// 分布式锁 /// </summary> /// <param name="key">锁key</param> /// <param name="lockExpirySeconds">锁自动超时时间(秒)</param> /// <param name="waitLockMs">等待锁时间(秒)</param> /// <returns></returns> public static bool Lock(string key, int lockExpirySeconds = 10, double waitLockSeconds = 0) { //间隔等待50毫秒 int waitIntervalMs = 50;
RedisClient client = (RedisClient)RedisClientManager.GetClient(RedisProviderName.RedisCommon, 0); string lockKey = "LockForSetNX:" + key;
DateTime begin = DateTime.Now; using (client)
{ while (true)
{ //循环获取取锁 if (client.SetNX(lockKey, new byte[] { 1 }) == 1)
{ //设置锁的过期时间 client.Expire(lockKey, lockExpirySeconds); return true;
} //不等待锁则返回 if (waitLockSeconds == 0) break; //超过等待时间,则不再等待 if ((DateTime.Now - begin).TotalSeconds >= waitLockSeconds) break;
Thread.Sleep(waitIntervalMs);
} return false;
}
}
2、释放锁
/// <summary> /// 删除锁 执行完代码以后调用释放锁 /// </summary> /// <param name="key"></param> public static void DelLock(string key) {
RedisClient client = (RedisClient)RedisClientManager.GetClient(RedisProviderName.RedisCommon, 0); string lockKey = "LockForSetNX:" + key; using (client)
{
client.Del(lockKey);
}
}
3、业务应用代码
try { //取锁,设置key10秒后失效,最大等待锁5秒 if (RedisHelper.Lock("LockKey", 10, 5))
{ //取到锁,执行具体业务 //例如商品购买,库存-1 }
} finally { //释放锁 DelLock("LockKey");
}
【如果文章对你有所帮助,请在评论区留言点赞,以资鼓励。】