com.datadoghq.trace.integration.DBComponent Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dd-trace Show documentation

Datadog core library

There is a newer version: 0.2.12

package com.datadoghq.trace.integration;

import com.datadoghq.trace.DDSpanContext;
import com.datadoghq.trace.DDTags;
import io.opentracing.tag.Tags;

/**
 * This span decorator leverages DB tags. It allows the dev to define a custom
 * service name  and retrieves some DB meta such as the statement
 */
public class DBComponent extends DDSpanContextDecorator {

	public DBComponent() {
		super();
		this.setMatchingTag(Tags.COMPONENT.getKey());
		this.setSetTag(DDTags.SERVICE_NAME);
	}

	@Override
	public boolean afterSetTag(DDSpanContext context, String tag, Object value) {
		//Assign service name
		if (super.afterSetTag(context, tag, value)) {
			//Assign span type to DB
			context.setSpanType("db");
			return true;
		}
		return false;
	}
}