Redis 系列五之分布式锁

        #region ---分布式锁---

        /// <summary>
        /// 加锁
        /// </summary>
        /// <param name="key">锁key</param>
        /// <param name="selfMark">自己唯一标记</param>
        /// <param name="lockExpirySeconds">锁自动过期时间[默认10](s)</param>
        /// <param name="waitLockMilliseconds">等待锁时间(ms)</param>
        /// <returns></returns>
        public bool Lock(string key, out string selfMark, int lockExpirySeconds = 10, long waitLockMilliseconds = 0)
        {
            DateTime begin = DateTime.Now;
            selfMark = Guid.NewGuid().ToString("N");//自己标记,释放锁时会用到,自己加的锁除非过期否则只能自己打开
            using (IRedisClient redisClient = GetClient())
            {
                string lockKey = "Lock_" + key;
                while (true)
                {
                    string script = string.Format("if redis.call('SETNX', KEYS[1], ARGV[1]) == 1 then redis.call('PEXPIRE',KEYS[1],{0}) return 1 else return 0 end", lockExpirySeconds * 1000);
                    //循环获取取锁
                    if (redisClient.ExecLuaAsInt(script, new[] { lockKey }, new[] { selfMark }) == 1)
                    {
                        return true;
                    }
                    //不等待锁则返回
                    if (waitLockMilliseconds == 0)
                    {
                        break;
                    }
                    //超过等待时间,则不再等待
                    if ((DateTime.Now - begin).TotalMilliseconds >= waitLockMilliseconds)
                    {
                        break;
                    }
                }
                return false;
            }
        }

        /// <summary>
        /// 释放锁
        /// </summary>
        /// <param name="key">锁key</param>
        /// <param name="selfMark">自己唯一标记</param>
        public void UnLock(string key, string selfMark)
        {
            using (IRedisClient redisClient = GetClient())
            {
                string lockKey = "Lock_" + key;
                var script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
                redisClient.ExecLuaAsString(script, new[] { lockKey }, new[] { selfMark });
            }
        }

        #endregion
            
            for (int i = 1; i <= 50; i++)
            {
                Task.Run(() =>
                {
                    Random random = new Random();
                    int userId = random.Next(1, 5);
                    string lock_key = "user_{0}";
                    string selfMark = string.Empty;
                    try
                    {
                        lock_key = string.Format(lock_key, userId);
                        //加锁
                        if (RedisUtil.Instance().Lock(lock_key, out selfMark, 10, 0))
                        {
                            //逻辑代码......



                            Thread.Sleep(500);
                        }
                    }
                    catch (Exception ex)
                    {
                        throw ex;
                    }
                    finally
                    {
                        //解锁
                        RedisUtil.Instance().UnLock(lock_key, selfMark);
                    }
                });
            }

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值