浏览代码

Merge pull request #17 from linjiacheng/patch-1

fix: IpRateLimit未注入IProcessingStrategy导致启动时报错
zhontai 3 年之前
父节点
当前提交
ea21b7f2d1
共有 1 个文件被更改,包括 3 次插入5 次删除
  1. 3 5
      Admin.Core/Extensions/RateLimitServiceCollectionExtensions.cs

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