com.wavefront.opentracing.WavefrontSpanContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wavefront-opentracing-sdk-java Show documentation

Implements OpenTracing API for collecting and sending tracing data to Wavefront from Java applications.

The newest version!

package com.wavefront.opentracing;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

import javax.annotation.Nullable;

import io.opentracing.SpanContext;

/**
 * Represents a Wavefront SpanContext based on OpenTracing's {@link SpanContext}.
 *
 * @author Vikram Raman
 */
public class WavefrontSpanContext implements SpanContext {

  private final UUID traceId;
  private final UUID spanId;
  private final Boolean samplingDecision;
  private final Map baggage;

  public WavefrontSpanContext(UUID traceId, UUID spanId) {
    this(traceId, spanId, null, null);
  }

  public WavefrontSpanContext(UUID traceId, UUID spanId, Map baggage, Boolean decision) {
    this.traceId = traceId;
    this.spanId = spanId;
    this.samplingDecision = decision;

    // expected that most contexts will have no bagagge items except when propagated
    this.baggage = (baggage == null) ? Collections.emptyMap() : baggage;
  }

  @Override
  public String toTraceId() {
    return traceId.toString();
  }

  @Override
  public String toSpanId() {
    return spanId.toString();
  }

  @Override
  public Iterable> baggageItems() {
    return Collections.unmodifiableMap(baggage).entrySet();
  }

  @Nullable
  public String getBaggageItem(String key) {
    return baggage.get(key);
  }

  public WavefrontSpanContext withBaggageItem(String key, String value) {
    Map items = new HashMap<>(baggage);
    items.put(key, value);
    return new WavefrontSpanContext(traceId, spanId, items, samplingDecision);
  }

  Map getBaggage() {
    // protected method for internal use. don't wrap within unmodifiable for memory efficiency.
    return baggage;
  }

  WavefrontSpanContext withSamplingDecision(boolean decision) {
    return new WavefrontSpanContext(traceId, spanId, baggage, Boolean.valueOf(decision));
  }

  public UUID getTraceId() {
    return traceId;
  }

  public UUID getSpanId() {
    return spanId;
  }

  public boolean isSampled() {
    return samplingDecision != null;
  }

  @Nullable
  public Boolean getSamplingDecision() {
    return samplingDecision;
  }

  @Override
  public String toString() {
    return "WavefrontSpanContext{" +
        "traceId=" + traceId +
        ", spanId=" + spanId +
        '}';
  }
}