ASP.NET Core 限流中间件：深入架构、配置与自定义策略

ASP.NET Core 限流中间件：架构与配置

ASP.NET Core 提供了一种强大而灵活的机制来限制传入请求的速率，这对于维护 Web 应用程序的稳定性和安全性至关重要。此功能的核心是一个专门的_限流中间件_，它利用通用的 .NET 限流组件高效地管理请求流。理解其架构、配置原则和可扩展性，能帮助中高级开发人员构建健壮且可扩展的解决方案。本文将深入探讨该组件的概念和运行机制，超越表层使用。

ASP.NET Core 中限流中间件的集成与配置

_限流中间件_在 ASP.NET Core 的限流系统中扮演着核心角色。它拦截每一个传入的 HTTP 请求，根据预定义的规则决定是处理还是拒绝。该中间件充当适配器，利用通用的 .NET 限流组件提供的底层限流算法。

要将_中间件组件_集成到 ASP.NET Core 请求处理管道中，需要使用 IApplicationBuilder 接口的 UseRateLimiter 扩展方法。其在管道中的关键位置要求是将其置于 UseRouting 和 UseEndpoints 调用之间。这确保了限流在请求路由确定之后、但在请求到达其最终处理端点之前应用。在基于 WebApplication/WebApplicationBuilder 模板构建的应用程序中（从 ASP.NET Core 6.0 开始），此条件通常默认满足。

_中间件组件_的行为通过_限流器配置参数_进行配置，这些参数封装在 RateLimiterOptions 类的实例中。UseRateLimiter 方法有两种主要形式：一种允许直接传递 RateLimiterOptions 实例，而另一种更常用，它从应用程序的服务容器中检索参数，这与 Options 模式保持一致。

为了实现完整功能，包括指标收集，必须在依赖注入容器中注册必要的服务。这通过 IServiceCollection 的 AddRateLimiter 扩展方法完成。从 ASP.NET Core 8 开始，注册这些服务是强制性的。AddRateLimiter 方法还通过作为参数传递的委托来配置 RateLimiterOptions。在 ASP.NET Core 9 中，新增了一种不带委托的 AddRateLimiter 形式，这意味着提供了其他方式来提供设置。这些参数的有效_配置_对于应用程序的性能和安全性至关重要。

全局与基于策略的限流机制

ASP.NET Core 的限流功能通过两种独立但互补的机制运行：_全局限流_和_基于策略的限流_。只有当请求获得这两种机制的许可时，才能成功处理。如果其中一种机制未配置或不适用于当前请求，则其许可被视为自动授予。

全局请求限流

_全局限流_无一例外地应用于所有传入请求。它通过将分区限流器实例分配给 RateLimiterOptions 中的 GlobalLimiter 属性来配置。此限流器必须是 PartitionedRateLimiter 类的对象，并以 HttpContext 资源类型进行参数化。在 ASP.NET Core 的上下文中，HttpContext 始终用作限流器的资源类型。缺少全局限流器配置会停用此机制。在全局层面管理流量是第一道防线。

基于策略的限流

为了更精细地控制请求流，使用_基于策略的限流_。_请求限流策略_允许将特定的限流规则应用于某些请求组，这些请求组与特定的路由或路由端点相关联。

策略通过 RateLimiterOptions 中 AddPolicy 方法的多个重载进行配置。这些方法允许将命名的_限流策略_添加到配置中。

策略可以是：

命名策略： 在 RateLimiterOptions 中注册，具有唯一标识符。它们可以多次重用并绑定到各种路由。
匿名策略： 直接绑定到特定路由，不打算在其他地方重用。

配置命名策略的过程包括两个阶段：

使用 AddPolicy 将命名策略添加到 RateLimiterOptions。
在路由配置中将此策略绑定到一个或多个路由。

匿名策略只需绑定到路由。这为_管理_对各种端点的_访问_提供了灵活性。

处理拒绝请求与创建自定义策略

当请求被限流机制拒绝时，_中间件组件_必须执行特定的操作。此行为通过 RateLimiterOptions 中的 OnRejected 和 RejectionStatusCode 属性进行配置。

RejectionStatusCode 属性定义了将返回给客户端的 HTTP 状态码（例如，503 Service Unavailable 或 429 Too Many Requests）。OnRejected 属性允许指定一个_拒绝处理程序_——一个接受 HttpContext 和拒绝上下文对象的异步委托（ValueTask）。此委托旨在修改响应，例如，添加 Retry-After 头或提供有关拒绝原因的更详细信息。

值得注意的是，_限流策略_可以拥有自己特定的拒绝处理程序。如果一个受策略约束的请求同时定义了全局处理程序和策略特定的处理程序，则始终优先使用策略中指定的处理程序。这种机制允许对_被拒绝的请求_进行精细调整。

开发自定义限流策略

为了实现超出标准配置的复杂和自定义_限流_场景，ASP.NET Core 提供了创建_自定义请求限流策略_的能力。此类策略必须实现泛型接口 IRateLimiterPolicy<TPartitionKey>，其中 TPartitionKey 定义了分区键类型。

IRateLimiterPolicy<TPartitionKey> 接口定义了两个关键元素：

GetPartition 方法：这是_分区方法_，它根据当前的 HttpContext 返回具有指定分区键类型的分区数据。它允许动态选择请求的底层限流器，例如，基于客户端的 IP 地址、用户 ID、请求头值或其他 HttpContext 参数。这使得限流具有选择性，即使在同一策略下，也可以对不同组的请求应用不同的限制。
OnRejected 属性：如前所述，它允许为此特定策略定义一个特定的拒绝处理程序，从而覆盖全局处理程序。

实现 IRateLimiterPolicy<TPartitionKey> 的对象被_中间件组件_用于创建所谓的_预备策略_，这些策略直接应用于_请求处理_。在当前版本中，除了实现此接口的类之外，以 HttpContext 作为资源类型的分区委托也可以作为命名策略的基础。

自定义策略和_分区_的应用为开发人员提供了极大的灵活性，以创建精确且适应性强的限流系统，能够满足高负载应用程序、微服务架构或公共 API 的独特要求。这有助于有效的_资源管理_和滥用预防。

核心要点

ASP.NET Core 中的限流中间件是流量控制和安全的关键组件，与通用的 .NET 限流机制集成。
功能通过 RateLimiterOptions 配置，使用 UseRateLimiter 用于管道，AddRateLimiter 用于服务注册和配置。
该系统支持两个级别的限流：针对所有请求的_全局限流_和针对特定路由进行精细控制的_基于策略的限流_。
拒绝处理通过 OnRejected 和 RejectionStatusCode 配置，并可在策略级别进行覆盖。
实现 IRateLimiterPolicy<TPartitionKey> 的_自定义策略_，能够基于 HttpContext 实现复杂的_分区_，用于动态选择限流器和自定义拒绝逻辑。