Parcourir la source

fix: IpRateLimit未注入IProcessingStrategy导致启动时报错

AspNetCoreRateLimit更新到新版后,IpRateLimit启用时,未注入IProcessingStrategy导致启动时报错
donkie il y a 3 ans
Parent
commit
9d4761d217

+ 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限流
         }
     }
-}
+}