Răsfoiți Sursa

Merge pull request #17 from linjiacheng/patch-1

fix: IpRateLimit未注入IProcessingStrategy导致启动时报错
zhontai 3 ani în urmă
părinte
comite
ea21b7f2d1

+ 3 - 5
Admin.Core/Extensions/RateLimitServiceCollectionExtensions.cs

@@ -27,19 +27,17 @@ namespace Admin.Core.Extensions
                 //redis
                 var redisRateLimit = new CSRedis.CSRedisClient(cacheConfig.Redis.ConnectionStringRateLimit);
                 services.AddSingleton<IDistributedCache>(new CSRedisCache(redisRateLimit));
-                services.AddSingleton<IIpPolicyStore, DistributedCacheIpPolicyStore>();
-                services.AddSingleton<IRateLimitCounterStore, DistributedCacheRateLimitCounterStore>();
+                services.AddDistributedRateLimiting();
             }
             else
             {
                 //内存
                 services.AddMemoryCache();
-                services.AddSingleton<IIpPolicyStore, MemoryCacheIpPolicyStore>();
-                services.AddSingleton<IRateLimitCounterStore, MemoryCacheRateLimitCounterStore>();
+                services.AddInMemoryRateLimiting();
             }
             services.AddSingleton<IRateLimitConfiguration, RateLimitConfiguration>();
 
             #endregion IP限流
         }
     }
-}
+}