Added some throttling

BoundlessStudio · Jan 8, 2025 · e3d2d32 · e3d2d32
1 parent 245c76d
commit e3d2d32
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/Program.cs b/Program.cs
@@ -2,9 +2,33 @@
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
 using OpenAI.Chat;
+using ThrottlingTroll;
 
 var host = new HostBuilder()
-    .ConfigureFunctionsWebApplication()
+    .ConfigureFunctionsWebApplication((builderContext, workerAppBuilder) => {
+      workerAppBuilder.UseThrottlingTroll(options => {
+        options.Config = new ThrottlingTrollConfig
+        {
+          Rules =
+          [
+            new ThrottlingTrollRule
+            {
+              UriPattern = "/api/Iterator_Start",
+              LimitMethod = new FixedWindowRateLimitMethod
+              {
+                PermitLimit = 10,
+                IntervalInSeconds = 3600,
+              }
+            }
+          ]
+        };
+        options.IdentityIdExtractor = request =>
+        {
+          var httpRequest = ((IIncomingHttpRequestProxy)request).Request;
+          return httpRequest.HttpContext.Connection.RemoteIpAddress?.ToString();
+        };
+      });
+    })
     .ConfigureServices(services =>
     {
       services.AddApplicationInsightsTelemetryWorkerService();

diff --git a/WhiteCrow.csproj b/WhiteCrow.csproj
@@ -16,6 +16,7 @@
     <PackageReference Include="Microsoft.ApplicationInsights.WorkerService" Version="2.22.0" />
     <PackageReference Include="Microsoft.Azure.Functions.Worker.ApplicationInsights" Version="2.0.0" />
     <PackageReference Include="OpenAI" Version="2.1.0" />
+    <PackageReference Include="ThrottlingTroll.AzureFunctionsAspNet" Version="7.3.0" />
   </ItemGroup>
   <ItemGroup>
     <None Update="host.json">