asp .net core 中间件实现，避免请求并发过高

zxb11c

已于 2024-08-01 19:43:14 修改

阅读量148

点赞数 3

文章标签： .netcore

于 2024-08-01 19:42:31 首次发布

本文链接：https://blog.csdn.net/zxb11c/article/details/140855757

版权

1，直接上代码

  /// <summary>
  /// 10 秒内 同ip最多放100次
  /// </summary>
  public class RequestThrottlingMiddleware
  {
      private const int outtime = 10;
      private const int total = 100;

      private readonly RequestDelegate _next;
      private readonly IMemoryCache _cache;
      public RequestThrottlingMiddleware(RequestDelegate next, IMemoryCache cache)
      {
          _next = next;
          _cache = cache;
      }

      public async Task Invoke(HttpContext context)
      {
          var ipAddress = context.Connection.RemoteIpAddress.ToString();
            
          // 用于设置10秒时间窗口
          var timeWindow = TimeSpan.FromSeconds(outtime);
          var cacheKey = $"{ipAddress}_{DateTime.UtcNow.Ticks / timeWindow.Ticks}";

          if (_cache.TryGetValue(cacheKey, out int requestCount))
          {
              // 如果请求次数超过100次，可以将IP地址加入黑名单或执行其他操作
              if (requestCount > total)
              {
                  // 加入黑名单的逻辑
                  context.Response.StatusCode = 403; // Forbidden
                  await context.Response.WriteAsync("IP Address is blacklisted.请求太频繁了，休息一下发请求吧！");
              }
              else
              {
                  // 请求次数加一
                  _cache.Set(cacheKey, requestCount + 1, timeWindow);
                  await _next(context);
              }
          }
          else
          {
              // 如果IP地址不在缓存中，将其添加到缓存并设置请求次数为1
              _cache.Set(cacheKey, 1, timeWindow);
              await _next(context);
          }
      }
  }

2, 配置中间件

    public void Configure(IApplicationBuilder app, IWebHostEnvironment env){
//...
     //过滤请求频繁的
     app.UseMiddleware<RequestThrottlingMiddleware>();
//...
}