http server builder

open-telemetry · Sep 27, 2024 · 773c98e · 773c98e
1 parent 41efcc5
commit 773c98e
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 8 deletions.
diff --git a/...main/kotlin/io/opentelemetry/instrumentation/ktor/v2_0/client/KtorClientTracingBuilder.kt b/...main/kotlin/io/opentelemetry/instrumentation/ktor/v2_0/client/KtorClientTracingBuilder.kt
@@ -23,9 +23,11 @@ class KtorClientTracingBuilder {
     }
   }
 
+  private lateinit var openTelemetry: OpenTelemetry
   private lateinit var clientBuilder: DefaultHttpClientInstrumenterBuilder<HttpRequestData, HttpResponse>
 
   fun setOpenTelemetry(openTelemetry: OpenTelemetry) {
+    this.openTelemetry = openTelemetry
     this.clientBuilder = DefaultHttpClientInstrumenterBuilder(
       INSTRUMENTATION_NAME,
       openTelemetry,
@@ -165,6 +167,6 @@ class KtorClientTracingBuilder {
 
   internal fun build(): KtorClientTracing = KtorClientTracing(
     instrumenter = clientBuilder.build(),
-    propagators = clientBuilder.openTelemetry.propagators,
+    propagators = openTelemetry.propagators,
   )
 }
diff --git a/...ry/src/main/kotlin/io/opentelemetry/instrumentation/ktor/v2_0/server/KtorServerTracing.kt b/...ry/src/main/kotlin/io/opentelemetry/instrumentation/ktor/v2_0/server/KtorServerTracing.kt
@@ -51,7 +51,8 @@ class KtorServerTracing private constructor(
         DefaultHttpServerInstrumenterBuilder(
           INSTRUMENTATION_NAME,
           openTelemetry,
-          KtorHttpServerAttributesGetter.INSTANCE
+          KtorHttpServerAttributesGetter.INSTANCE,
+          null
         )
     }
 

diff --git a/.../src/main/java/io/opentelemetry/instrumentation/ratpack/v1_7/RatpackTelemetryBuilder.java b/.../src/main/java/io/opentelemetry/instrumentation/ratpack/v1_7/RatpackTelemetryBuilder.java
@@ -36,8 +36,7 @@ public final class RatpackTelemetryBuilder {
             .setHeaderSetter(RequestHeaderSetter.INSTANCE);
     serverBuilder =
         new DefaultHttpServerInstrumenterBuilder<>(
-                INSTRUMENTATION_NAME, openTelemetry, RatpackHttpAttributesGetter.INSTANCE)
-            .setHeaderGetter(RatpackGetter.INSTANCE);
+                INSTRUMENTATION_NAME, openTelemetry, RatpackHttpAttributesGetter.INSTANCE, RatpackGetter.INSTANCE);
   }
 
   /**
@@ -152,7 +151,7 @@ public RatpackTelemetryBuilder setEmitExperimentalHttpServerMetrics(
   /** Sets custom client {@link SpanNameExtractor} via transform function. */
   @CanIgnoreReturnValue
   public RatpackTelemetryBuilder setClientSpanNameExtractor(
-      Function<SpanNameExtractor<RequestSpec>, ? extends SpanNameExtractor<? super RequestSpec>>
+      Function<SpanNameExtractor<? super RequestSpec>, ? extends SpanNameExtractor<? super RequestSpec>>
           clientSpanNameExtractor) {
     clientBuilder.setSpanNameExtractor(clientSpanNameExtractor);
     return this;
@@ -161,7 +160,7 @@ public RatpackTelemetryBuilder setClientSpanNameExtractor(
   /** Sets custom server {@link SpanNameExtractor} via transform function. */
   @CanIgnoreReturnValue
   public RatpackTelemetryBuilder setServerSpanNameExtractor(
-      Function<SpanNameExtractor<Request>, ? extends SpanNameExtractor<? super Request>>
+      Function<SpanNameExtractor<? super Request>, ? extends SpanNameExtractor<? super Request>>
           serverSpanNameExtractor) {
     serverBuilder.setSpanNameExtractor(serverSpanNameExtractor);
     return this;

diff --git a/...src/main/java/io/opentelemetry/javaagent/instrumentation/undertow/UndertowSingletons.java b/...src/main/java/io/opentelemetry/javaagent/instrumentation/undertow/UndertowSingletons.java
@@ -26,8 +26,7 @@ public final class UndertowSingletons {
                 builder.addContextCustomizer(
                     (context, request, attributes) -> {
                       // span is ended when counter reaches 0, we start from 2 which accounts for
-                      // the
-                      // handler that started the span and exchange completion listener
+                      // the handler that started the span and exchange completion listener
                       context = UndertowActiveHandlers.init(context, 2);
 
                       return new AppServerBridge.Builder()