fix dispatcher in library instrumentation

Author: AlixBa

instrumentation/nats/nats-2.21/library/src/main/java/io/nats/client/impl/OpenTelemetryDispatcherFactory.java

@@ -0,0 +1,39 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.nats.client.impl;
+
+import io.nats.client.MessageHandler;
+import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
+import io.opentelemetry.instrumentation.nats.v2_21.OpenTelemetryMessageHandler;
+import io.opentelemetry.instrumentation.nats.v2_21.internal.NatsRequest;
+
+/**
+ * This class is internal and is hence not for public use. Its APIs are unstable and can change at
+ * any time.
+ */
+public final class OpenTelemetryDispatcherFactory extends DispatcherFactory {
+
+  private final DispatcherFactory delegate;
+  private final Instrumenter<NatsRequest, Void> consumerReceiveInstrumenter;
+  private final Instrumenter<NatsRequest, Void> consumerProcessInstrumenter;
+
+  public OpenTelemetryDispatcherFactory(
+      DispatcherFactory delegate,
+      Instrumenter<NatsRequest, Void> consumerReceiveInstrumenter,
+      Instrumenter<NatsRequest, Void> consumerProcessInstrumenter) {
+    this.delegate = delegate;
+    this.consumerReceiveInstrumenter = consumerReceiveInstrumenter;
+    this.consumerProcessInstrumenter = consumerProcessInstrumenter;
+  }
+
+  @Override
+  NatsDispatcher createDispatcher(NatsConnection natsConnection, MessageHandler messageHandler) {
+    return delegate.createDispatcher(
+        natsConnection,
+        new OpenTelemetryMessageHandler(
+            messageHandler, consumerReceiveInstrumenter, consumerProcessInstrumenter));
+  }
+}

instrumentation/nats/nats-2.21/library/src/main/java/io/opentelemetry/instrumentation/nats/v2_21/NatsTelemetry.java

@@ -8,6 +8,7 @@
 import io.nats.client.Connection;
 import io.nats.client.Options;
 import io.nats.client.impl.DispatcherFactory;
+import io.nats.client.impl.OpenTelemetryDispatcherFactory;
 import io.opentelemetry.api.OpenTelemetry;
 import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
 import io.opentelemetry.instrumentation.nats.v2_21.internal.NatsRequest;
@@ -54,7 +55,7 @@ public Options.Builder wrap(Options.Builder options) {
     }
 
     return options.dispatcherFactory(
-        OpenTelemetryDispatcherFactory.wrap(
+        new OpenTelemetryDispatcherFactory(
             factory, consumerReceiveInstrumenter, consumerProcessInstrumenter));
   }
 }

instrumentation/nats/nats-2.21/library/src/main/java/io/opentelemetry/instrumentation/nats/v2_21/OpenTelemetryDispatcherFactory.java

@@ -14,7 +14,11 @@
 import io.opentelemetry.instrumentation.api.internal.Timer;
 import io.opentelemetry.instrumentation.nats.v2_21.internal.NatsRequest;
 
-final class OpenTelemetryMessageHandler implements MessageHandler {
+/**
+ * This class is internal and is hence not for public use. Its APIs are unstable and can change at
+ * any time. Exposed for {@link io.nats.client.impl.OpenTelemetryDispatcherFactory}.
+ */
+public final class OpenTelemetryMessageHandler implements MessageHandler {
 
   private final MessageHandler delegate;
   private final Instrumenter<NatsRequest, Void> consumerReceiveInstrumenter;

instrumentation/nats/nats-2.21/library/src/main/java/io/opentelemetry/instrumentation/nats/v2_21/OpenTelemetryMessageHandler.java

@@ -17,9 +17,7 @@
 
 /**
  * This class is internal and is hence not for public use. Its APIs are unstable and can change at
- * any time.", or "This class is internal and experimental. Its APIs are unstable and can change at
- * any time. Its APIs (or a version of them) may be promoted to the public stable API in the future,
- * but no guarantees are made.
+ * any time.
  */
 public final class NatsInstrumenterFactory {
   private static final String INSTRUMENTATION_NAME = "io.opentelemetry.nats-2.21";

instrumentation/nats/nats-2.21/library/src/main/java/io/opentelemetry/instrumentation/nats/v2_21/internal/NatsInstrumenterFactory.java

@@ -13,9 +13,7 @@
 
 /**
  * This class is internal and is hence not for public use. Its APIs are unstable and can change at
- * any time.", or "This class is internal and experimental. Its APIs are unstable and can change at
- * any time. Its APIs (or a version of them) may be promoted to the public stable API in the future,
- * but no guarantees are made.
+ * any time.
  */
 @AutoValue
 public abstract class NatsRequest {