fix: Add workaround for prometheus

GZTimeWalker · Apr 14, 2024 · 5364a89 · 5364a89
1 parent d1f5c51
commit 5364a89
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/src/GZCTF/Extensions/TelemetryExtension.cs b/src/GZCTF/Extensions/TelemetryExtension.cs
@@ -71,10 +71,24 @@ public static void UseTelemetry(this IApplicationBuilder app, TelemetryConfig? c
         if (config is not { Enable: true, Prometheus.Enable: true })
             return;
 
-
         if (config.Prometheus.Port is { } port)
-            app.UseOpenTelemetryPrometheusScrapingEndpoint(context => context.Connection.LocalPort == port);
+            app.UseOpenTelemetryPrometheusScrapingEndpoint(context =>
+            {
+                if (context.Connection.LocalPort == port)
+                {
+                    // FIXME: workaround for prometheus
+                    context.Request.Headers.Accept = "application/openmetrics-text";
+                    return true;
+                }
+
+                return false;
+            });
         else
-            app.UseOpenTelemetryPrometheusScrapingEndpoint();
+            app.UseOpenTelemetryPrometheusScrapingEndpoint(context =>
+            {
+                // FIXME: workaround for prometheus
+                context.Request.Headers.Accept = "application/openmetrics-text";
+                return true;
+            });
     }
-}
+}