feat: allow to inject message tracing context to any carrier

gravitee-io · Nov 13, 2024 · 4f11ce5 · 4f11ce5
1 parent 4a04bad
commit 4f11ce5
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -34,7 +34,7 @@
 
     <properties>
         <gravitee-bom.version>8.1.16</gravitee-bom.version>
-        <gravitee-node.version>7.0.0-alpha.1</gravitee-node.version>
+        <gravitee-node.version>7.0.0-alpha.2</gravitee-node.version>
         <gravitee-common.version>4.5.1</gravitee-common.version>
         <gravitee-policy-api.version>1.11.0</gravitee-policy-api.version>
         <gravitee-reporter-api.version>1.31.2</gravitee-reporter-api.version>

diff --git a/src/main/java/io/gravitee/gateway/reactive/api/tracing/Tracer.java b/src/main/java/io/gravitee/gateway/reactive/api/tracing/Tracer.java
@@ -15,9 +15,10 @@
  */
 package io.gravitee.gateway.reactive.api.tracing;
 
-import io.gravitee.common.service.AbstractService;
 import io.gravitee.node.api.opentelemetry.Span;
+import io.opentelemetry.context.propagation.TextMapSetter;
 import io.vertx.core.Context;
+import java.util.function.BiConsumer;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 
@@ -67,4 +68,12 @@ public <R> void endWithResponseAndError(final Span span, final R response, final
     public <R> void endWithResponseAndError(final Span span, final R response, final String message) {
         delegate.endWithResponseAndError(vertxContext, span, response, message);
     }
+
+    public <C> void injectSpanContext(final BiConsumer<String, String> textMapSetter) {
+        delegate.injectSpanContext(vertxContext, textMapSetter);
+    }
+
+    public <C> void injectSpanContext(final Span span, final BiConsumer<String, String> textMapSetter) {
+        delegate.injectSpanContext(vertxContext, span, textMapSetter);
+    }
 }
diff --git a/src/main/java/io/gravitee/gateway/reactive/api/tracing/message/TracingMessage.java b/src/main/java/io/gravitee/gateway/reactive/api/tracing/message/TracingMessage.java
@@ -18,9 +18,12 @@
 import static io.gravitee.gateway.reactive.api.context.InternalContextAttributes.ATTR_INTERNAL_MESSAGE_RECORDABLE;
 import static io.gravitee.gateway.reactive.api.context.InternalContextAttributes.ATTR_INTERNAL_TRACING_MESSAGE_SPAN;
 
+import io.gravitee.gateway.reactive.api.context.base.BaseExecutionContext;
 import io.gravitee.gateway.reactive.api.message.Message;
 import io.gravitee.node.api.opentelemetry.Span;
+import io.opentelemetry.context.propagation.TextMapSetter;
 import java.util.Map;
+import java.util.function.BiConsumer;
 
 /**
  * This interface defines a {@link Message} compatible with tracing
@@ -58,4 +61,19 @@ default boolean isTraceable() {
      * @param value  the value associated
      */
     void addTracingAttribute(final String key, final String value);
+
+    /**
+     * Allow to inject message context into the given carrier
+     *
+     * @param executionContext the current execution context
+     * @param textMapSetter the map setter to use
+     */
+    default void injectSpanContext(final BaseExecutionContext executionContext, final BiConsumer<String, String> textMapSetter) {
+        if (isTraceable()) {
+            Span messageSpan = internalAttribute(ATTR_INTERNAL_TRACING_MESSAGE_SPAN);
+            if (messageSpan != null) {
+                executionContext.getTracer().injectSpanContext(messageSpan, textMapSetter);
+            }
+        }
+    }
 }