com.wavefront.ingester.SpanIngesterFormatter Maven / Gradle / Ivy

Go to download
package com.wavefront.ingester;

import wavefront.report.Annotation;
import wavefront.report.Span;

import javax.annotation.Nullable;
import java.util.Iterator;
import java.util.List;
import java.util.function.Supplier;

import com.wavefront.data.ParseException;

/**
 * Builder for Span formatter.
 *
 * @author [email protected]
 */
public class SpanIngesterFormatter extends AbstractIngesterFormatter {

  private SpanIngesterFormatter(List> elements) {
    super(elements);
  }

  public static class SpanFormatBuilder extends IngesterFormatBuilder {
    @Override
    public SpanIngesterFormatter build() {
      return new SpanIngesterFormatter(elements);
    }
  }

  public static IngesterFormatBuilder newBuilder() {
    return new SpanFormatBuilder();
  }

  @Override
  public Span drive(String input, @Nullable Supplier defaultHostNameSupplier,
                    String customerId, @Nullable List customSourceTags,
                    @Nullable List customLogTimestampTags,
                    @Nullable List customLogMessageTags,
                    List customLogApplicationTags, List customLogServiceTags,
                    @Nullable List customLogLevelTags,@Nullable List customLogExceptionTags, @Nullable IngesterContext ingesterContext) {
    Span span = new Span();
    span.setCustomer(customerId);
    StringParser parser = new StringParser(input);
    for (FormatterElement element : elements) {
      element.consume(parser, span);
    }
    if (parser.hasNext()) {
      throw new ParseException("Unexpected extra input: " + parser.next());
    }
    List annotations = span.getAnnotations();
    if (annotations != null) {
      boolean hasTrueSource = false;
      Iterator iterator = annotations.iterator();
      while (iterator.hasNext()) {
        final Annotation annotation = iterator.next();
        if (customSourceTags != null && !hasTrueSource &&
            customSourceTags.contains(annotation.getKey())) {
          span.setSource(annotation.getValue());
        }
        switch (annotation.getKey()) {
          case "source":
          case "host":
            span.setSource(annotation.getValue());
            iterator.remove();
            hasTrueSource = true;
            break;
          case "spanId":
            span.setSpanId(annotation.getValue());
            iterator.remove();
            break;
          case "traceId":
            span.setTraceId(annotation.getValue());
            iterator.remove();
            break;
          default:
            break;
        }
      }
    }
    if (span.getSource() == null && defaultHostNameSupplier != null) {
      span.setSource(defaultHostNameSupplier.get());
    }
    if (span.getSource() == null) {
      throw new IllegalArgumentException("source can't be null: " + input);
    }
    if (span.getSpanId() == null) {
      throw new IllegalArgumentException("spanId can't be null: " + input);
    }
    if (span.getTraceId() == null) {
      throw new IllegalArgumentException("traceId can't be null: " + input);
    }
    return span;
  }
}