Add shouldStart() call to armeria server instrumentation (#4843)

open-telemetry · Dec 8, 2021 · 8d65542 · 8d65542
1 parent b5516c8
commit 8d65542
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...ary/src/main/java/io/opentelemetry/instrumentation/armeria/v1_3/OpenTelemetryService.java b/...ary/src/main/java/io/opentelemetry/instrumentation/armeria/v1_3/OpenTelemetryService.java
@@ -29,7 +29,12 @@ final class OpenTelemetryService extends SimpleDecoratingHttpService {
 
   @Override
   public HttpResponse serve(ServiceRequestContext ctx, HttpRequest req) throws Exception {
-    Context context = instrumenter.start(Context.current(), ctx);
+    Context parentContext = Context.current();
+    if (!instrumenter.shouldStart(parentContext, ctx)) {
+      return unwrap().serve(ctx, req);
+    }
+
+    Context context = instrumenter.start(parentContext, ctx);
 
     Span span = Span.fromContext(context);
     if (span.isRecording()) {