|
using EzCoreNet.Redis;
|
using Furion;
|
using Furion.FriendlyException;
|
using Microsoft.AspNetCore.Mvc;
|
using Microsoft.AspNetCore.Mvc.Filters;
|
using System;
|
using System.Collections.Generic;
|
using System.IO;
|
using System.Linq;
|
using System.Security.Claims;
|
using System.Text;
|
using System.Threading.Tasks;
|
|
namespace Cylsg.Filter
|
{
|
|
public enum Limttype
|
{
|
/// <summary>
|
///共用限流
|
/// </summary>
|
All,
|
/// <summary>
|
/// 用户级限流
|
/// </summary>
|
User
|
}
|
|
/// <summary>
|
/// 限流
|
/// </summary>
|
|
public class LimitFilterAttribute : Attribute,IAsyncActionFilter
|
{
|
|
/// <summary>
|
/// redis限流文件夹名称
|
/// </summary>
|
private const string DirKey = "UserRateLimit:";
|
/// <summary>
|
/// 限流发生时 返回的提示
|
/// </summary>
|
public string ResponseMeg { get; set; } = "请不要在短时间内重复访问该资源";
|
|
/// <summary>
|
/// 限流时长 单位秒 默认为1秒
|
/// </summary>
|
public int timespan { get; set; } = 1;
|
/// <summary>
|
/// 限流次数 默认为1次
|
/// </summary>
|
public int InCount { get; set; } = 1;
|
|
/// <summary>
|
/// 默认为用户级
|
/// </summary>
|
public Limttype LimiType { get; set; }= Limttype.All;
|
|
public async Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next)
|
{
|
var redisHelper = App.GetService<IEzCoreNetRedisService>();
|
if(redisHelper == null)
|
{
|
throw Oops.Oh("内存服务异常，请求终止");
|
}
|
|
var path = context.HttpContext?.Request?.Path.Value;
|
if (string.IsNullOrEmpty(path))
|
{
|
//非API控制器，直接放行 没有找到路径
|
await next.Invoke();
|
return;
|
}
|
string key = "";
|
if (LimiType == Limttype.User)
|
{
|
if (App.User?.FindFirstValue("UserID") == null)
|
{
|
//没有用户直接放行
|
await next.Invoke();
|
return;
|
}
|
|
key = $"{DirKey}:{path}:{App.User?.FindFirstValue("UserID")}";
|
}
|
else
|
if(LimiType == Limttype.All)
|
{
|
key = $"{DirKey}:{path}:All";
|
}
|
else
|
{
|
await next.Invoke();
|
return;
|
}
|
|
|
long? count = redisHelper.Get<int>(key);
|
if (count == null || count == 0)
|
{
|
redisHelper.Incrby(key);
|
redisHelper.SetTtl(key, timespan);
|
}
|
else
|
if (count >= (InCount))
|
{
|
throw Oops.Oh(ResponseMeg);
|
|
}
|
else
|
{
|
redisHelper.Incrby(key);
|
}
|
|
await next.Invoke();
|
|
|
}
|
}
|
}
|
