28 lines
775 B
Java
28 lines
775 B
Java
package com.datadoghq.trace.integration;
|
|
|
|
import com.datadoghq.trace.DDSpanContext;
|
|
import com.datadoghq.trace.DDTags;
|
|
import io.opentracing.tag.Tags;
|
|
|
|
public class DBStatementAsResourceName extends AbstractDecorator {
|
|
|
|
public DBStatementAsResourceName() {
|
|
super();
|
|
this.setMatchingTag(Tags.DB_STATEMENT.getKey());
|
|
this.setSetTag(DDTags.RESOURCE_NAME);
|
|
}
|
|
|
|
|
|
@Override
|
|
public boolean afterSetTag(final DDSpanContext context, final String tag, final Object value) {
|
|
//Assign service name
|
|
if (super.afterSetTag(context, tag, value)) {
|
|
// Replace the OT db.statement by the DD sql.query
|
|
context.setTag(DDTags.DB_STATEMENT, value);
|
|
context.getTags().remove(Tags.DB_STATEMENT.getKey());
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
}
|