fix WrapperTracer in server

opensearch-project · Aug 3, 2023 · 71206b9 · 71206b9
1 parent 5ffc34f
commit 71206b9
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/server/src/main/java/org/opensearch/telemetry/tracing/WrappedTracer.java b/server/src/main/java/org/opensearch/telemetry/tracing/WrappedTracer.java
@@ -12,6 +12,7 @@
 import org.opensearch.telemetry.tracing.noop.NoopTracer;
 
 import java.io.IOException;
+import java.util.Map;
 
 /**
  * Wrapper implementation of Tracer. This delegates call to right tracer based on the tracer settings
@@ -36,8 +37,18 @@ public WrappedTracer(TelemetrySettings telemetrySettings, Tracer defaultTracer)
 
     @Override
     public SpanScope startSpan(String spanName) {
+        return startSpan(spanName, null);
+    }
+
+    @Override
+    public SpanScope startSpan(String spanName, Map<String, String> attributes) {
         Tracer delegateTracer = getDelegateTracer();
-        return delegateTracer.startSpan(spanName);
+        return delegateTracer.startSpan(spanName, attributes);
+    }
+
+    @Override
+    public Span getCurrentSpan() {
+        return getDelegateTracer().getCurrentSpan();
     }
 
     @Override