56 lines
2.1 KiB
C#
56 lines
2.1 KiB
C#
using Microsoft.AspNetCore.Mvc;
|
||
using Microsoft.AspNetCore.Mvc.Filters;
|
||
|
||
namespace HT.Cloud.Code
|
||
{
|
||
/// <summary>
|
||
/// 防重复锁，MVC使用
|
||
/// </summary>
|
||
public class HandlerLockAttribute : ActionFilterAttribute
|
||
{
|
||
public HandlerLockAttribute()
|
||
{
|
||
}
|
||
|
||
public override void OnActionExecuting(ActionExecutingContext filterContext)
|
||
{
|
||
if (OperatorProvider.Provider.GetCurrent() == null)
|
||
{
|
||
WebHelper.WriteCookie("WaterCloud_login_error", "overdue");
|
||
//filterContext.HttpContext.Response.WriteAsync("<script>top.location.href ='" + filterContext.HttpContext.Request.PathBase + "/Home/Error?msg=408" + "';if(document.all) window.event.returnValue = false;</script>");
|
||
OperatorProvider.Provider.EmptyCurrent("pc_").GetAwaiter().GetResult();
|
||
filterContext.Result = new RedirectResult(filterContext.HttpContext.Request.PathBase + "/Home/Error?msg=408");
|
||
return;
|
||
}
|
||
else
|
||
{
|
||
string token = filterContext.HttpContext.Request.Cookies["pc_" + GlobalContext.SystemConfig.TokenName];
|
||
string cacheToken = CacheHelper.GetAsync<string>("pc_" + GlobalContext.SystemConfig.TokenName + "_" + OperatorProvider.Provider.GetCurrent().UserId + "_" + OperatorProvider.Provider.GetCurrent().LoginTime).GetAwaiter().GetResult();
|
||
if (string.IsNullOrWhiteSpace(token))
|
||
{
|
||
filterContext.Result = new JsonResult(new AlwaysResult { state = ResultType.error.ToString(), message = "token不能空" });
|
||
return;
|
||
}
|
||
if (string.IsNullOrWhiteSpace(cacheToken))
|
||
{
|
||
filterContext.Result = new JsonResult(new AlwaysResult { state = ResultType.error.ToString(), message = "token不能空" });
|
||
return;
|
||
}
|
||
if (token != cacheToken)
|
||
{
|
||
filterContext.Result = new JsonResult(new AlwaysResult { state = ResultType.error.ToString(), message = "请求异常" });
|
||
return;
|
||
}
|
||
//固定加锁5秒
|
||
bool result = CacheHelper.SetNx(token, token, 5);
|
||
if (!result)
|
||
{
|
||
filterContext.Result = new JsonResult(new AlwaysResult { state = ResultType.error.ToString(), message = "请求太频繁，请稍后" });
|
||
return;
|
||
}
|
||
}
|
||
//随机值
|
||
base.OnActionExecuting(filterContext);
|
||
}
|
||
}
|
||
} |