72 lines
2.3 KiB
C#
72 lines
2.3 KiB
C#
|
using Hncore.Infrastructure.Extension;
|
|||
|
|
using System;
|
|||
|
|
using System.Diagnostics;
|
|||
|
|
using System.Reflection;
|
|||
|
|
using System.Runtime.InteropServices;
|
|||
|
|
|
|||
|
|
namespace Hncore.Infrastructure.Common
|
|||
|
|
{
|
|||
|
|
public class RedisLocker
|
|||
|
|
{
|
|||
|
|
private string _key;
|
|||
|
|
private long _lockTime;
|
|||
|
|
/// <summary>
|
|||
|
|
///
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="key">锁的键</param>
|
|||
|
|
/// <param name="outTime">锁超时时间 单位毫秒</param>
|
|||
|
|
public RedisLocker(string key,long outTime)
|
|||
|
|
{
|
|||
|
|
_key =$"Lock:{Assembly.GetCallingAssembly().GetFriendName()}:{key}" ;
|
|||
|
|
_lockTime = outTime;
|
|||
|
|
}
|
|||
|
|
public void Exec(Action action)
|
|||
|
|
{
|
|||
|
|
if (GetLock())
|
|||
|
|
{
|
|||
|
|
action();
|
|||
|
|
ReleaseLock();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected bool GetLock()
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
var currentTime = DateTime.Now.GetUnixTimeStamp();
|
|||
|
|
if (RedisHelper.SetNx(this._key, currentTime + _lockTime))
|
|||
|
|
{
|
|||
|
|
Console.WriteLine("获取到Redis锁了");
|
|||
|
|
RedisHelper.Expire(_key, TimeSpan.FromMilliseconds(_lockTime)); //设置过期时间
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//防止SetNx成功但是设置过期时间（Expire）失败造成死锁
|
|||
|
|
var lockValue = Convert.ToInt64(RedisHelper.Get(_key));
|
|||
|
|
currentTime = DateTime.Now.GetUnixTimeStamp();
|
|||
|
|
if (lockValue > 0 && currentTime > lockValue)
|
|||
|
|
{
|
|||
|
|
var getsetResult = Convert.ToInt64(RedisHelper.GetSet(_key, currentTime));
|
|||
|
|
if (getsetResult == 0 || getsetResult == lockValue)
|
|||
|
|
{
|
|||
|
|
Console.WriteLine("获取到Redis锁了");
|
|||
|
|
RedisHelper.Expire(_key, TimeSpan.FromMilliseconds(_lockTime));
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
Console.WriteLine("没有获取到锁");
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
catch
|
|||
|
|
{
|
|||
|
|
ReleaseLock();
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected bool ReleaseLock()
|
|||
|
|
{
|
|||
|
|
return RedisHelper.Del(_key) > 0;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|