ASP.NET Core 的 Web Api 实现限流 中间件

在ASP.NET Core中实现Web API限流的一个常见方法是使用中间件来限制并发请求的数量。以下是一个简单的中间件示例，用于限制API端点的并发访问量：

using Microsoft.AspNetCore.Http;
using System.Collections.Concurrent;
using System.Threading.Tasks;
 
public class ConcurrencyLimiterMiddleware
{
    private static readonly ConcurrentDictionary<string, int> _requests = new ConcurrentDictionary<string, int>();
    private readonly RequestDelegate _next;
    private readonly int _maxConcurrentRequests;
 
    public ConcurrencyLimiterMiddleware(RequestDelegate next, int maxConcurrentRequests)
    {
        _next = next;
        _maxConcurrentRequests = maxConcurrentRequests;
    }
 
    public async Task InvokeAsync(HttpContext context)
    {
        if (!context.Request.Path.StartsWithSegments("/api/"))
        {
            await _next(context);
            return;
        }
 
        var key = context.Request.HttpContext.Connection.RemoteIpAddress.ToString();
        int count;
 
        while (true)
        {
            // 尝试添加或更新key的计数
            if (_requests.TryAdd(key, 1) || _requests.TryUpdate(key, 1, 0))
            {
                // 成功添加或更新，继续处理请求
                count = _requests[key];
                break;
            }
            // 如果key已经在字典中，尝试更新其计数
            else if (_requests.TryGetValue(key, out count) && count < _maxConcurrentRequests)
            {
                // 更新成功，继续处理请求
                _requests[key] = count + 1;
                break;
            }
            // 如果已达到最大并发请求限制，则等待一段时间后重试
            await Task.Delay(100);
        }
 
        try
        {
            await _next(context);
        }
        finally
        {
            // 请求处理完毕后，将计数器减一
            _requests[key] = count - 1;
        }
    }
}
 
// 在Startup.cs中配置服务和中间件
public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    // ...
 
    // 添加并发请求限制中间件
    app.UseMiddleware<ConcurrencyLimiterMiddleware>(10); // 最多允许10个并发请求
 
    // ...
}

这个中间件ConcurrencyLimiterMiddleware会根据远程IP地址对每个请求进行计数，并且如果并发请求数量超过了设定的阈值，它会等待并重试。这个简单的实现没有考虑清理过期的IP地址记录或者更复杂的限流策略，但它展示了如何使用中间件来实现基本的限流功能。