com.wavefront.agent.preprocessor.SpanExtractAnnotationIfNotExistsTransformer Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of proxy-test Show documentation

Service for batching and relaying metric traffic to Wavefront

There is a newer version: 9999.0

package com.wavefront.agent.preprocessor;

import java.util.function.Predicate;

import javax.annotation.Nullable;

import wavefront.report.Span;

/**
 * Create a new span annotation by extracting a portion of a span name, source name or another annotation
 *
 * @author [email protected]
 */
public class SpanExtractAnnotationIfNotExistsTransformer extends SpanExtractAnnotationTransformer {

  public SpanExtractAnnotationIfNotExistsTransformer(final String key,
                                                     final String input,
                                                     final String patternSearch,
                                                     final String patternReplace,
                                                     @Nullable final String replaceInput,
                                                     @Nullable final String patternMatch,
                                                     final boolean firstMatchOnly,
                                                     @Nullable final Predicate v2Predicate,
                                                     final PreprocessorRuleMetrics ruleMetrics) {
    super(key, input, patternSearch, patternReplace, replaceInput, patternMatch, firstMatchOnly,
        v2Predicate, ruleMetrics);
  }

  @Nullable
  @Override
  public Span apply(@Nullable Span span) {
    if (span == null) return null;
    long startNanos = ruleMetrics.ruleStart();
    try {
      if (!v2Predicate.test(span)) return span;

      if (span.getAnnotations().stream().noneMatch(a -> a.getKey().equals(key))) {
        internalApply(span);
      }
      return span;
    } finally {
      ruleMetrics.ruleEnd(startNanos);
    }
  }
}