ASP.NET Core中怎么利用令牌桶限流

本篇文章給大家分享的是有關ASP.NET Core中怎么利用令牌桶限流，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

1、安裝Nuget包

有多種安裝方式，選擇自己喜歡的就行了。

包管理器命令：

Install-Package FireflySoft.RateLimit.AspNetCore

或者.NET命令:

dotnet add package FireflySoft.RateLimit.AspNetCore

或者項目文件直接添加：

<ItemGroup>
<PackageReference Include="FireflySoft.RateLimit.AspNetCore" Version="2.*" />
</ItemGroup>

2、使用中間件

在Startup中使用中間件，演示代碼如下（下邊會有詳細說明）：

public void ConfigureServices(IServiceCollection services)
        {
            ...
            app.AddRateLimit(new InProcessTokenBucketAlgorithm(
                new[] {
                    new TokenBucketRule(30,10,TimeSpan.FromSeconds(1))
                    {
                        ExtractTarget = context =>
                        {
                            return (context as HttpContext).Request.Path.Value;
                        },
                        CheckRuleMatching = context =>
                        {
                            return true;
                        },
                        Name="default limit rule",
                    }
                })
            );
            ...
        }

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
        {
            ...
            app.UseRateLimit();
            ...
        }

如上需要先注冊服務，然后使用中間件。

注冊服務的時候需要提供限流算法和對應的規則：

• 這里使用進程內令牌桶算法，對于分布式服務還可以使用Redis令牌桶算法，支持StackExchange.Redis。

• 桶的容量是30，每秒流入10個令牌。

• ExtractTarget用于提取限流目標，這里是每個不同的請求Path。如果有IO請求，這里還支持對應的異步方法ExtractTargetAsync。

• CheckRuleMatching用于驗證當前請求是否限流。如果有IO請求，這里還支持對應的異步方法CheckRuleMatchingAsync。

• 默認被限流時會返回HttpStatusCode 429，可以在AddRateLimit時使用可選參數error自定義這個值，以及Http Header和Body中的內容。

以上就是ASP.NET Core中怎么利用令牌桶限流，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。