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

Don’t change resource name of span unless it’s for Rabbit.

This commit is contained in:
Tyler Benson 2018-10-19 12:27:34 +10:00
parent 64595cf485
commit 97dd303909
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
dd-java-agent/instrumentation/rabbitmq-amqp-2.7/src/main/java/datadog/trace/instrumentation/rabbitmq/amqp/RabbitCommandInstrumentation.java
View File

@ -11,6 +11,7 @@ import com.rabbitmq.client.Command;
import com.rabbitmq.client.Method;
import datadog.trace.agent.tooling.Instrumenter;
import datadog.trace.api.DDTags;
import datadog.trace.api.interceptor.MutableSpan;
import io.opentracing.Span;
import io.opentracing.util.GlobalTracer;
import java.util.Collections;
@ -50,13 +51,16 @@ public class RabbitCommandInstrumentation extends Instrumenter.Default {
final Span span = GlobalTracer.get().activeSpan();
final Method method = command.getMethod();
if (span != null && method != null) {
final String name = method.protocolMethodName();
if (!name.equals("basic.publish")) {
// Don't overwrite the name already set.
span.setTag(DDTags.RESOURCE_NAME, name);
if (span instanceof MutableSpan && method != null) {
if (((MutableSpan) span).getOperationName().equals("amqp.command")) {
final String name = method.protocolMethodName();
if (!name.equals("basic.publish")) {
// Don't overwrite the name already set.
span.setTag(DDTags.RESOURCE_NAME, name);
}
span.setTag("amqp.command", name);
}
span.setTag("amqp.command", name);
}
}