open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add async propagation flags for kafka consumer.

This commit is contained in:
Tyler Benson 2019-12-18 10:40:54 -08:00
parent b3f15ca133
commit 7095ea3426
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/TracingIterator.java
View File

@ -59,6 +59,7 @@ public class TracingIterator implements Iterator<ConsumerRecord> {
decorator.afterStart(span); decorator.afterStart(span);
decorator.onConsume(span, next); decorator.onConsume(span, next);
currentScope = activateSpan(span, true); currentScope = activateSpan(span, true);
currentScope.setAsyncPropagation(true);
} }
} catch (final Exception e) { } catch (final Exception e) {
log.debug("Error during decoration", e); log.debug("Error during decoration", e);

3
dd-java-agent/instrumentation/kafka-streams-0.11/src/main/java/datadog/trace/instrumentation/kafka_streams/KafkaStreamsProcessorInstrumentation.java
View File

@ -81,7 +81,7 @@ public class KafkaStreamsProcessorInstrumentation {
CONSUMER_DECORATE.afterStart(span); CONSUMER_DECORATE.afterStart(span);
CONSUMER_DECORATE.onConsume(span, record); CONSUMER_DECORATE.onConsume(span, record);
activateSpan(span, true); activateSpan(span, true).setAsyncPropagation(true);
} }
} }
} }
@ -119,6 +119,7 @@ public class KafkaStreamsProcessorInstrumentation {
@Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class) @Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class)
public static void stopSpan(@Advice.Thrown final Throwable throwable) { public static void stopSpan(@Advice.Thrown final Throwable throwable) {
// This is dangerous... we assume the span/scope is the one we expect, but it may not be.
final AgentSpan span = activeSpan(); final AgentSpan span = activeSpan();
if (span != null) { if (span != null) {
CONSUMER_DECORATE.onError(span, throwable); CONSUMER_DECORATE.onError(span, throwable);