Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 10 additions & 1 deletion src/AspNetCoreRateLimit.Redis/RedisProcessingStrategy.cs
Original file line number Diff line number Diff line change
Expand Up @@ -3,20 +3,23 @@
using System;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Extensions.Options;

namespace AspNetCoreRateLimit.Redis
{
public class RedisProcessingStrategy : ProcessingStrategy
{
private readonly IConnectionMultiplexer _connectionMultiplexer;
private readonly IRateLimitConfiguration _config;
private readonly RedisRateLimitOptions _options;
private readonly ILogger<RedisProcessingStrategy> _logger;

public RedisProcessingStrategy(IConnectionMultiplexer connectionMultiplexer, IRateLimitConfiguration config, ILogger<RedisProcessingStrategy> logger)
public RedisProcessingStrategy(IConnectionMultiplexer connectionMultiplexer, IRateLimitConfiguration config, IOptions<RedisRateLimitOptions> redisOptions, ILogger<RedisProcessingStrategy> logger)
: base(config)
{
_connectionMultiplexer = connectionMultiplexer ?? throw new ArgumentException("IConnectionMultiplexer was null. Ensure StackExchange.Redis was successfully registered");
_config = config;
_options = redisOptions.Value;
_logger = logger;
}

Expand All @@ -42,5 +45,11 @@ public async Task<RateLimitCounter> IncrementAsync(string counterId, TimeSpan in
Timestamp = intervalStart
};
}

protected override string BuildCounterKey(ClientRequestIdentity requestIdentity, RateLimitRule rule, ICounterKeyBuilder counterKeyBuilder,
RateLimitOptions rateLimitOptions)
{
return $"{_options.KeyPrefix}{base.BuildCounterKey(requestIdentity, rule, counterKeyBuilder, rateLimitOptions)}";
}
}
}
10 changes: 10 additions & 0 deletions src/AspNetCoreRateLimit.Redis/RedisRateLimitOptions.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
namespace AspNetCoreRateLimit.Redis
{
public class RedisRateLimitOptions
{
/// <summary>
/// Gets or sets the Redis key prefix
/// </summary>
public string KeyPrefix { get; set; } = "ratelimits:";
}
}
9 changes: 6 additions & 3 deletions test/AspNetCoreRateLimit.Demo/Startup.cs
Original file line number Diff line number Diff line change
Expand Up @@ -28,12 +28,15 @@ public void ConfigureServices(IServiceCollection services)
// configure client rate limiting middleware
services.Configure<ClientRateLimitOptions>(Configuration.GetSection("ClientRateLimiting"));
services.Configure<ClientRateLimitPolicies>(Configuration.GetSection("ClientRateLimitPolicies"));

// configure Redis rate limiting strategy
services.Configure<RedisRateLimitOptions>(Configuration.GetSection("RedisRateLimitPolicies"));

// register stores
services.AddInMemoryRateLimiting();
//services.AddDistributedRateLimiting<AsyncKeyLockProcessingStrategy>();
//services.AddDistributedRateLimiting<RedisProcessingStrategy>();
//services.AddRedisRateLimiting();
// services.AddDistributedRateLimiting<AsyncKeyLockProcessingStrategy>();
// services.AddDistributedRateLimiting<RedisProcessingStrategy>();
// services.AddRedisRateLimiting();

services.AddMvc((options) =>
{
Expand Down
4 changes: 4 additions & 0 deletions test/AspNetCoreRateLimit.Demo/appsettings.json
Original file line number Diff line number Diff line change
Expand Up @@ -216,5 +216,9 @@
]
}
]
},

"RedisRateLimitPolicies": {
"KeyPrefix": "ratelimits:"
}
}