12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- using Admin.Core.Common.Configs;
- using AspNetCoreRateLimit;
- using Microsoft.Extensions.Caching.Distributed;
- using Microsoft.Extensions.Caching.Redis;
- using Microsoft.Extensions.Configuration;
- using Microsoft.Extensions.DependencyInjection;
- namespace Admin.Core.Extensions
- {
- public static class RateLimitServiceCollectionExtensions
- {
- /// <summary>
- /// 添加Ip限流
- /// </summary>
- /// <param name="services"></param>
- /// <param name="configuration"></param>
- /// <param name="cacheConfig"></param>
- public static void AddIpRateLimit(this IServiceCollection services, IConfiguration configuration, CacheConfig cacheConfig)
- {
- #region IP限流
- services.Configure<IpRateLimitOptions>(configuration.GetSection("IpRateLimiting"));
- services.Configure<IpRateLimitPolicies>(configuration.GetSection("IpRateLimitPolicies"));
- if (cacheConfig.TypeRateLimit == Common.Cache.CacheType.Redis)
- {
- //redis
- var redisRateLimit = new CSRedis.CSRedisClient(cacheConfig.Redis.ConnectionStringRateLimit);
- services.AddSingleton<IDistributedCache>(new CSRedisCache(redisRateLimit));
- services.AddDistributedRateLimiting();
- }
- else
- {
- //内存
- services.AddMemoryCache();
- services.AddInMemoryRateLimiting();
- }
- services.AddSingleton<IRateLimitConfiguration, RateLimitConfiguration>();
- #endregion IP限流
- }
- }
- }
|