Implement producer part of RocketMQ new client instrumentation

instrumentation/rocketmq/rocketmq-client-java/rocketmq-client-java-5.0/javaagent/build.gradle.kts

@@ -0,0 +1,21 @@
+plugins {
+  id("otel.javaagent-instrumentation")
+}
+
+muzzle {
+  pass {
+    group.set("org.apache.rocketmq")
+    module.set("rocketmq-client-java")
+    versions.set("[5.0.2,)")
+  }
+}
+
+dependencies {
+  library("org.apache.rocketmq:rocketmq-client-java:5.0.2")
+
+  testImplementation(project(":instrumentation:rocketmq:rocketmq-client-java:rocketmq-client-java-5.0:testing"))
+}
+
+tasks.withType<Test>().configureEach {
+  systemProperty("testLatestDeps", findProperty("testLatestDeps") as Boolean)
+}

instrumentation/rocketmq/rocketmq-client-java/rocketmq-client-java-5.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/rocketmqclientjava/v5_0/MapSetter.java

@@ -0,0 +1,22 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.rocketmqclientjava.v5_0;
+
+import io.opentelemetry.context.propagation.TextMapSetter;
+import javax.annotation.Nullable;
+import org.apache.rocketmq.client.java.message.PublishingMessageImpl;
+
+enum MapSetter implements TextMapSetter<PublishingMessageImpl> {
+  INSTANCE;
+
+  @Override
+  public void set(@Nullable PublishingMessageImpl carrier, String key, String value) {
+    if (null == carrier || !"traceparent".equals(key)) {
+      return;
+    }
+    carrier.setTraceContext(value);
+  }
+}

instrumentation/rocketmq/rocketmq-client-java/rocketmq-client-java-5.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/rocketmqclientjava/v5_0/ParentContextExtractor.java

@@ -0,0 +1,48 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.rocketmqclientjava.v5_0;
+
+import io.opentelemetry.api.trace.propagation.W3CTraceContextPropagator;
+import io.opentelemetry.context.Context;
+import io.opentelemetry.context.propagation.TextMapGetter;
+import java.util.Collections;
+import java.util.Map;
+import java.util.Optional;
+import org.apache.rocketmq.client.java.message.PublishingMessageImpl;
+
+public class ParentContextExtractor {
+  private ParentContextExtractor() {}
+
+  public static Context fromMessage(PublishingMessageImpl message) {
+    Optional<String> parentTraceContext = message.getParentTraceContext();
+    if (!parentTraceContext.isPresent()) {
+      return Context.root();
+    }
+
+    // Use W3C standard propagator to propagate trace context.
+    return W3CTraceContextPropagator.getInstance()
+        .extract(
+            Context.root(),
+            Collections.singletonMap("traceparent", parentTraceContext.get()),
+            MapGetter.INSTANCE);
+  }
+
+  private enum MapGetter implements TextMapGetter<Map<String, String>> {
+    INSTANCE;
+
+    @Override
+    public Iterable<String> keys(Map<String, String> carrier) {
+      return carrier.keySet();
+    }
+
+    @SuppressWarnings("NullableProblems")
+    @Override
+    public String get(Map<String, String> carrier, String key) {
+      assert carrier != null;
+      return carrier.get(key);
+    }
+  }
+}

instrumentation/rocketmq/rocketmq-client-java/rocketmq-client-java-5.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/rocketmqclientjava/v5_0/RocketMqInstrumentationModule.java

@@ -0,0 +1,24 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.rocketmqclientjava.v5_0;
+
+import com.google.auto.service.AutoService;
+import io.opentelemetry.javaagent.extension.instrumentation.InstrumentationModule;
+import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
+import java.util.Collections;
+import java.util.List;
+
+@AutoService(InstrumentationModule.class)
+public class RocketMqInstrumentationModule extends InstrumentationModule {
+  public RocketMqInstrumentationModule() {
+    super("rocketmq-client-java", "rocketmq-client-java-5.0");
+  }
+
+  @Override
+  public List<TypeInstrumentation> typeInstrumentations() {
+    return Collections.singletonList(new RocketMqProducerInstrumentation());
+  }
+}

instrumentation/rocketmq/rocketmq-client-java/rocketmq-client-java-5.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/rocketmqclientjava/v5_0/RocketMqInstrumenterFactory.java

@@ -0,0 +1,70 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.rocketmqclientjava.v5_0;
+
+import static io.opentelemetry.instrumentation.api.instrumenter.AttributesExtractor.constant;
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MESSAGING_OPERATION;
+
+import io.opentelemetry.api.OpenTelemetry;
+import io.opentelemetry.api.trace.StatusCode;
+import io.opentelemetry.instrumentation.api.instrumenter.AttributesExtractor;
+import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
+import io.opentelemetry.instrumentation.api.instrumenter.InstrumenterBuilder;
+import io.opentelemetry.instrumentation.api.instrumenter.SpanKindExtractor;
+import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessageOperation;
+import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingAttributesExtractor;
+import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingAttributesGetter;
+import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingSpanNameExtractor;
+import java.util.List;
+import org.apache.rocketmq.client.java.impl.producer.SendReceiptImpl;
+import org.apache.rocketmq.client.java.message.PublishingMessageImpl;
+
+public class RocketMqInstrumenterFactory {
+  private static final String INSTRUMENTATION_NAME = "io.opentelemetry.rocketmq-client-java-5.0";
+
+  private RocketMqInstrumenterFactory() {}
+
+  public static Instrumenter<PublishingMessageImpl, SendReceiptImpl> createProducerInstrumenter(
+      OpenTelemetry openTelemetry, List<String> capturedHeaders, boolean propagationEnabled) {
+
+    RocketMqProducerAttributeGetter getter = RocketMqProducerAttributeGetter.INSTANCE;
+    MessageOperation operation = MessageOperation.SEND;
+
+    AttributesExtractor<PublishingMessageImpl, SendReceiptImpl> attributesExtractor =
+        buildMessagingAttributesExtractor(getter, operation, capturedHeaders);
+
+    InstrumenterBuilder<PublishingMessageImpl, SendReceiptImpl> instrumenterBuilder =
+        Instrumenter.<PublishingMessageImpl, SendReceiptImpl>builder(
+                openTelemetry,
+                INSTRUMENTATION_NAME,
+                MessagingSpanNameExtractor.create(getter, operation))
+            .addAttributesExtractor(attributesExtractor)
+            .addAttributesExtractor(RocketMqProducerAttributeExtractor.INSTANCE)
+            .addAttributesExtractor(constant(MESSAGING_OPERATION, "send"))
+            .setSpanStatusExtractor(
+                (spanStatusBuilder, message, sendReceipt, error) -> {
+                  if (null == sendReceipt) {
+                    spanStatusBuilder.setStatus(StatusCode.ERROR);
+                    return;
+                  }
+                  spanStatusBuilder.setStatus(StatusCode.OK);
+                });
+
+    if (propagationEnabled) {
+      return instrumenterBuilder.buildProducerInstrumenter(MapSetter.INSTANCE);
+    }
+    return instrumenterBuilder.buildInstrumenter(SpanKindExtractor.alwaysProducer());
+  }
+
+  private static <T, R> MessagingAttributesExtractor<T, R> buildMessagingAttributesExtractor(
+      MessagingAttributesGetter<T, R> getter,
+      MessageOperation operation,
+      List<String> capturedHeaders) {
+    return MessagingAttributesExtractor.builder(getter, operation)
+        .setCapturedHeaders(capturedHeaders)
+        .build();
+  }
+}

instrumentation/rocketmq/rocketmq-client-java/rocketmq-client-java-5.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/rocketmqclientjava/v5_0/RocketMqProducerAttributeExtractor.java

@@ -0,0 +1,55 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.rocketmqclientjava.v5_0;
+
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MESSAGING_ROCKETMQ_MESSAGE_KEYS;
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MESSAGING_ROCKETMQ_MESSAGE_TAG;
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MESSAGING_ROCKETMQ_MESSAGE_TYPE;
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MessagingRocketmqMessageTypeValues.DELAY;
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MessagingRocketmqMessageTypeValues.FIFO;
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MessagingRocketmqMessageTypeValues.NORMAL;
+import static io.opentelemetry.semconv.trace.attributes.SemanticAttributes.MessagingRocketmqMessageTypeValues.TRANSACTION;
+
+import io.opentelemetry.api.common.AttributesBuilder;
+import io.opentelemetry.context.Context;
+import io.opentelemetry.instrumentation.api.instrumenter.AttributesExtractor;
+import java.util.ArrayList;
+import javax.annotation.Nullable;
+import org.apache.rocketmq.client.java.impl.producer.SendReceiptImpl;
+import org.apache.rocketmq.client.java.message.PublishingMessageImpl;
+
+public enum RocketMqProducerAttributeExtractor
+    implements AttributesExtractor<PublishingMessageImpl, SendReceiptImpl> {
+  INSTANCE;
+
+  @Override
+  public void onStart(
+      AttributesBuilder attributes, Context parentContext, PublishingMessageImpl message) {
+    message.getTag().ifPresent(s -> attributes.put(MESSAGING_ROCKETMQ_MESSAGE_TAG, s));
+    attributes.put(MESSAGING_ROCKETMQ_MESSAGE_KEYS, new ArrayList<>(message.getKeys()));
+    switch (message.getMessageType()) {
+      case FIFO:
+        attributes.put(MESSAGING_ROCKETMQ_MESSAGE_TYPE, FIFO);
+        break;
+      case DELAY:
+        attributes.put(MESSAGING_ROCKETMQ_MESSAGE_TYPE, DELAY);
+        break;
+      case TRANSACTION:
+        attributes.put(MESSAGING_ROCKETMQ_MESSAGE_TYPE, TRANSACTION);
+        break;
+      default:
+        attributes.put(MESSAGING_ROCKETMQ_MESSAGE_TYPE, NORMAL);
+    }
+  }
+
+  @Override
+  public void onEnd(
+      AttributesBuilder attributes,
+      Context context,
+      PublishingMessageImpl message,
+      @Nullable SendReceiptImpl sendReceipt,
+      @Nullable Throwable error) {}
+}

instrumentation/rocketmq/rocketmq-client-java/rocketmq-client-java-5.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/rocketmqclientjava/v5_0/RocketMqProducerAttributeGetter.java

@@ -0,0 +1,82 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.rocketmqclientjava.v5_0;
+
+import io.opentelemetry.instrumentation.api.instrumenter.messaging.MessagingAttributesGetter;
+import io.opentelemetry.semconv.trace.attributes.SemanticAttributes;
+import javax.annotation.Nullable;
+import org.apache.rocketmq.client.java.impl.producer.SendReceiptImpl;
+import org.apache.rocketmq.client.java.message.PublishingMessageImpl;
+
+enum RocketMqProducerAttributeGetter
+    implements MessagingAttributesGetter<PublishingMessageImpl, SendReceiptImpl> {
+  INSTANCE;
+
+  @Nullable
+  @Override
+  public String system(PublishingMessageImpl message) {
+    return "rocketmq";
+  }
+
+  @Nullable
+  @Override
+  public String destinationKind(PublishingMessageImpl message) {
+    return SemanticAttributes.MessagingDestinationKindValues.TOPIC;
+  }
+
+  @Nullable
+  @Override
+  public String destination(PublishingMessageImpl message) {
+    return message.getTopic();
+  }
+
+  @Override
+  public boolean temporaryDestination(PublishingMessageImpl message) {
+    return false;
+  }
+
+  @Nullable
+  @Override
+  public String protocol(PublishingMessageImpl message) {
+    return null;
+  }
+
+  @Nullable
+  @Override
+  public String protocolVersion(PublishingMessageImpl message) {
+    return null;
+  }
+
+  @Nullable
+  @Override
+  public String url(PublishingMessageImpl message) {
+    return null;
+  }
+
+  @Nullable
+  @Override
+  public String conversationId(PublishingMessageImpl message) {
+    return null;
+  }
+
+  @Nullable
+  @Override
+  public Long messagePayloadSize(PublishingMessageImpl message) {
+    return (long) message.getBody().remaining();
+  }
+
+  @Nullable
+  @Override
+  public Long messagePayloadCompressedSize(PublishingMessageImpl message) {
+    return null;
+  }
+
+  @Nullable
+  @Override
+  public String messageId(PublishingMessageImpl message, @Nullable SendReceiptImpl sendReceipt) {
+    return message.getMessageId().toString();
+  }
+}