chore: add tracing endpoint

baloise-incubator · Oct 17, 2023 · f9df14d · f9df14d
1 parent d90f300
commit f9df14d
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/src/main/java/ch/baloise/observability/gabelstaplerbuggyapp/ProfilingTracingFilter.java b/src/main/java/ch/baloise/observability/gabelstaplerbuggyapp/ProfilingTracingFilter.java
@@ -24,16 +24,17 @@ public class ProfilingTracingFilter implements WebFilter {
     @Override
     public Mono<Void> filter(ServerWebExchange serverWebExchange,
                              WebFilterChain webFilterChain) {
-        String traceId = "unknown";
-        String spanId = "unknown";
-        Span span = tracer.currentSpan();
-        if (span != null) {
-            traceId = span.context().traceId();
-            spanId = span.context().spanId();
-        }
-        log.info("Filter called with trace-id %s and span-id %s".formatted(traceId, spanId));
-
-        return Pyroscope.LabelsWrapper.run(new LabelsSet("trace-id", traceId, "span-id", spanId), () -> webFilterChain.filter(serverWebExchange));
+        return Pyroscope.LabelsWrapper.run(new LabelsSet("trace-id", traceId, "span-id", spanId), () -> {
+            String traceId = "unknown";
+            String spanId = "unknown";
+            Span span = tracer.currentSpan();
+            if (span != null) {
+                traceId = span.context().traceId();
+                spanId = span.context().spanId();
+            }
+            log.info("Filter called with trace-id %s and span-id %s".formatted(traceId, spanId));
+            return webFilterChain.filter(serverWebExchange);
+        });
     }
 
 }