com.datadoghq.trace.integration.AbstractDecorator 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 AbstractDecorator {

  private String matchingTag;

  private Object matchingValue;

  private String setTag;

  private String setValue;

  public boolean afterSetTag(final DDSpanContext context, final String tag, final Object value) {
    if ((this.getMatchingValue() == null || value.equals(this.getMatchingValue()))) {
      final String targetTag = getSetTag() == null ? tag : getSetTag();
      final 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(final String tag) {
    this.matchingTag = tag;
  }

  public Object getMatchingValue() {
    return matchingValue;
  }

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

  public String getSetTag() {
    return setTag;
  }

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

  public String getSetValue() {
    return setValue;
  }

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