com.datadoghq.trace.integration.DDSpanContextDecorator 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;

/**
 * Span decorators are called when new tags are written and proceed to various remappings and enrichments
 */
public abstract class DDSpanContextDecorator {

	private String matchingTag;

	private String matchingValue;

	private String setTag;

	private String setValue;

	public boolean afterSetTag(DDSpanContext context, String tag, Object value) {
		if ((this.getMatchingValue() == null || value.equals(this.getMatchingValue()))) {
			String targetTag = getSetTag() == null ? tag : getSetTag();
			String targetValue = getSetValue() == null ? String.valueOf(value) : getSetValue();
			
			if (targetTag.equals(DDTags.SERVICE_NAME)) {
				context.setServiceName(targetValue);
			} else if (targetTag.equals(DDTags.RESOURCE_NAME)) {
				context.setResourceName(targetValue);
			} else if (targetTag.equals(DDTags.SPAN_TYPE)) {
				context.setSpanType(targetValue);
			} else {
				context.setTag(targetTag, targetValue);
			}
			return true;
		} else {
			return false;
		}
	}

	public String getMatchingTag() {
		return matchingTag;
	}

	public void setMatchingTag(String tag) {
		this.matchingTag = tag;
	}

	public String getMatchingValue() {
		return matchingValue;
	}

	public void setMatchingValue(String value) {
		this.matchingValue = value;
	}

	public String getSetTag() {
		return setTag;
	}

	public void setSetTag(String targetTag) {
		this.setTag = targetTag;
	}

	public String getSetValue() {
		return setValue;
	}

	public void setSetValue(String targetValue) {
		this.setValue = targetValue;
	}
}