Add plugin support for registation of trace event listeners

opensearch-project · Aug 3, 2023 · e16a9df · e16a9df
1 parent 71206b9
commit e16a9df
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 0 deletions.
diff --git a/plugins/telemetry-otel/src/main/java/org/opensearch/telemetry/OTelTelemetryPlugin.java b/plugins/telemetry-otel/src/main/java/org/opensearch/telemetry/OTelTelemetryPlugin.java
@@ -16,9 +16,12 @@
 import org.opensearch.telemetry.tracing.OTelResourceProvider;
 import org.opensearch.telemetry.tracing.OTelTelemetry;
 import org.opensearch.telemetry.tracing.OTelTracingTelemetry;
+import org.opensearch.telemetry.tracing.listeners.TraceEventListener;
 
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.List;
+import java.util.Map;
 import java.util.Optional;
 
 /**
@@ -58,6 +61,11 @@ public String getName() {
         return OTEL_TRACER_NAME;
     }
 
+    @Override
+    public Map<String, TraceEventListener> getTraceEventListeners(Telemetry telemetry) {
+        return Collections.emptyMap();
+    }
+
     private Telemetry telemetry() {
         return new OTelTelemetry(new OTelTracingTelemetry(OTelResourceProvider.get(settings)), new MetricsTelemetry() {
         });

diff --git a/server/src/main/java/org/opensearch/plugins/TelemetryPlugin.java b/server/src/main/java/org/opensearch/plugins/TelemetryPlugin.java
@@ -10,7 +10,11 @@
 
 import org.opensearch.telemetry.Telemetry;
 import org.opensearch.telemetry.TelemetrySettings;
+import org.opensearch.telemetry.diagnostics.DiagnosticsEventListener;
+import org.opensearch.telemetry.tracing.listeners.TraceEventListener;
 
+import java.util.Collections;
+import java.util.Map;
 import java.util.Optional;
 
 /**
@@ -22,4 +26,6 @@ public interface TelemetryPlugin {
 
     String getName();
 
+    Map<String, TraceEventListener> getTraceEventListeners(Telemetry telemetry);
+
 }
diff --git a/test/framework/src/main/java/org/opensearch/test/telemetry/MockTelemetryPlugin.java b/test/framework/src/main/java/org/opensearch/test/telemetry/MockTelemetryPlugin.java
@@ -8,11 +8,14 @@
 
 package org.opensearch.test.telemetry;
 
+import java.util.Collections;
+import java.util.Map;
 import java.util.Optional;
 import org.opensearch.plugins.Plugin;
 import org.opensearch.plugins.TelemetryPlugin;
 import org.opensearch.telemetry.Telemetry;
 import org.opensearch.telemetry.TelemetrySettings;
+import org.opensearch.telemetry.tracing.listeners.TraceEventListener;
 
 /**
  * Mock {@link TelemetryPlugin} implementation for testing.
@@ -36,4 +39,9 @@ public Optional<Telemetry> getTelemetry(TelemetrySettings settings) {
     public String getName() {
         return MOCK_TRACER_NAME;
     }
+
+    @Override
+    public Map<String, TraceEventListener> getTraceEventListeners(Telemetry telemetry) {
+        return Collections.emptyMap();
+    }
 }