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

[mongo] update the db.statement tag to use the quantized query

This commit is contained in:
Emanuele Palazzetti 2017-08-01 11:46:50 +02:00
parent 2f22b98cd1
commit e236f44b12
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
dd-java-agent/src/main/java/com/datadoghq/trace/agent/integration/MongoHelper.java
View File

@ -5,6 +5,7 @@ import com.mongodb.MongoClientOptions;
import com.mongodb.event.CommandStartedEvent; import com.mongodb.event.CommandStartedEvent;
import io.opentracing.Span; import io.opentracing.Span;
import io.opentracing.contrib.mongo.TracingCommandListener; import io.opentracing.contrib.mongo.TracingCommandListener;
import io.opentracing.tag.Tags;
import java.util.Arrays; import java.util.Arrays;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
@ -50,8 +51,14 @@ public class MongoHelper extends DDAgentTracingHelper<MongoClientOptions.Builder
public void decorate(final Span span, final CommandStartedEvent event) { public void decorate(final Span span, final CommandStartedEvent event) {
try { try {
// normalize the Mongo command so that parameters are removed from the string
final BsonDocument normalized = norm(event.getCommand()); final BsonDocument normalized = norm(event.getCommand());
span.setTag(DDTags.RESOURCE_NAME, normalized.toString()); final String mongoCmd = normalized.toString();
// add specific resource name and replace the `db.statement` OpenTracing
// tag with the quantized version of the Mongo command
span.setTag(DDTags.RESOURCE_NAME, mongoCmd);
span.setTag(Tags.DB_STATEMENT.getKey(), mongoCmd);
} catch (final Throwable e) { } catch (final Throwable e) {
log.warn("Couldn't decorate the mongo query: " + e.getMessage(), e); log.warn("Couldn't decorate the mongo query: " + e.getMessage(), e);
} }