Add telemetry to HttpEventListener

trinodb · Dec 5, 2024 · 7b026b3 · 7b026b3
1 parent 9dfafec
commit 7b026b3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...http-event-listener/src/main/java/io/trino/plugin/httpquery/HttpEventListenerFactory.java b/...http-event-listener/src/main/java/io/trino/plugin/httpquery/HttpEventListenerFactory.java
@@ -16,6 +16,8 @@
 import com.google.inject.Scopes;
 import io.airlift.bootstrap.Bootstrap;
 import io.airlift.json.JsonModule;
+import io.opentelemetry.api.OpenTelemetry;
+import io.opentelemetry.api.trace.Tracer;
 import io.trino.spi.eventlistener.EventListener;
 import io.trino.spi.eventlistener.EventListenerFactory;
 import io.trino.spi.eventlistener.QueryCompletedEvent;
@@ -43,6 +45,8 @@ public EventListener create(Map<String, String> config, EventListenerContext con
         Bootstrap app = new Bootstrap(
                 new JsonModule(),
                 binder -> {
+                    binder.bind(OpenTelemetry.class).toInstance(context.getOpenTelemetry());
+                    binder.bind(Tracer.class).toInstance(context.getTracer());
                     jsonCodecBinder(binder).bindJsonCodec(QueryCompletedEvent.class);
                     jsonCodecBinder(binder).bindJsonCodec(QueryCreatedEvent.class);
                     jsonCodecBinder(binder).bindJsonCodec(SplitCompletedEvent.class);