Disable UseDefaultForwarders

microsoft · Apr 10, 2024 · 4d2d28d · 4d2d28d
1 parent af7a775
commit 4d2d28d
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/testassets/BenchmarkApp/Program.cs b/testassets/BenchmarkApp/Program.cs
@@ -14,6 +14,7 @@
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using Yarp.ReverseProxy.Forwarder;
+using Yarp.ReverseProxy.Transforms.Builder;
 
 BenchmarksEventSource.MeasureAspNetVersion();
 BenchmarksEventSource.MeasureNetCoreAppVersion();
@@ -53,10 +54,11 @@
     var forwarder = app.ApplicationServices.GetRequiredService<IHttpForwarder>();
     var clusterUrl = GetClusterUrl();
     var httpClient = new HttpMessageInvoker(CreateHandler());
+    var transformer = CreateHttpTransformer(app);
 
     app.Run(async context =>
     {
-        await forwarder.SendAsync(context, clusterUrl, httpClient, ForwarderRequestConfig.Empty, HttpTransformer.Default);
+        await forwarder.SendAsync(context, clusterUrl, httpClient, ForwarderRequestConfig.Empty, transformer);
     });
 });
 
@@ -95,3 +97,13 @@ static SocketsHttpHandler CreateHandler()
 
     return handler;
 }
+
+static HttpTransformer CreateHttpTransformer(IApplicationBuilder app)
+{
+    var transformBuilder = app.ApplicationServices.GetRequiredService<ITransformBuilder>();
+
+    return transformBuilder.Create(context =>
+    {
+        context.UseDefaultForwarders = false;
+    });
+}