Skip to content

Commit 87b5288

Browse files
Clean up Rate Limit sample (dotnet#19)
1 parent b2be189 commit 87b5288

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

fundamentals/middleware/rate-limit/WebRateLimitAuth/Program.cs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,8 +148,6 @@
148148
using Microsoft.AspNetCore.Identity;
149149
using Microsoft.AspNetCore.RateLimiting;
150150
using Microsoft.EntityFrameworkCore;
151-
using Microsoft.Extensions.Logging.Abstractions;
152-
using Microsoft.Extensions.Options;
153151
using WebRateLimitAuth;
154152
using WebRateLimitAuth.Data;
155153
using WebRateLimitAuth.Models;

fundamentals/middleware/rate-limit/WebRateLimitAuth/SampleRateLimiterPolicy.cs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
1+
// <snippet_1>
12
using System.Threading.RateLimiting;
23
using Microsoft.AspNetCore.RateLimiting;
34
using Microsoft.Extensions.Options;
45
using WebRateLimitAuth.Models;
56

67
namespace WebRateLimitAuth;
78

9+
// <snippet>
810
public class SampleRateLimiterPolicy : IRateLimiterPolicy<string>
911
{
1012
private Func<OnRejectedContext, CancellationToken, ValueTask>? _onRejected;
@@ -15,7 +17,7 @@ public SampleRateLimiterPolicy(ILogger<SampleRateLimiterPolicy> logger,
1517
{
1618
_onRejected = (context, token) =>
1719
{
18-
context.HttpContext.Response.StatusCode = 429;
20+
context.HttpContext.Response.StatusCode = StatusCodes.Status429TooManyRequests;
1921
logger.LogWarning($"Request rejected by {nameof(SampleRateLimiterPolicy)}");
2022
return ValueTask.CompletedTask;
2123
};
@@ -24,6 +26,7 @@ public SampleRateLimiterPolicy(ILogger<SampleRateLimiterPolicy> logger,
2426

2527
public Func<OnRejectedContext, CancellationToken, ValueTask>?
2628
OnRejected { get => _onRejected; }
29+
// </snippet>
2730

2831
public RateLimitPartition<string> GetPartition(HttpContext httpContext)
2932
{
@@ -37,3 +40,4 @@ public RateLimitPartition<string> GetPartition(HttpContext httpContext)
3740

3841
}
3942
}
43+
// </snippet_1>

0 commit comments

Comments
 (0)