com.wavefront.agent.preprocessor.ReportPointForceLowercaseTransformer 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 com.google.common.base.Function;
import com.google.common.base.Preconditions;

import java.util.function.Predicate;
import java.util.regex.Pattern;

import javax.annotation.Nullable;

import wavefront.report.ReportPoint;

/**
 * Force lowercase transformer. Converts a specified component of a point (metric name,
 * source name or a point tag value, depending on "scope" parameter) to lower case to
 * enforce consistency.
 *
 * @author [email protected]
 */
public class ReportPointForceLowercaseTransformer implements Function {

  private final String scope;
  @Nullable
  private final Pattern compiledMatchPattern;
  private final PreprocessorRuleMetrics ruleMetrics;
  private final Predicate v2Predicate;


  public ReportPointForceLowercaseTransformer(final String scope,
                                              @Nullable final String patternMatch,
                                              @Nullable Predicate v2Predicate,
                                              final PreprocessorRuleMetrics ruleMetrics) {
    this.scope = Preconditions.checkNotNull(scope, "[scope] can't be null");
    Preconditions.checkArgument(!scope.isEmpty(), "[scope] can't be blank");
    this.compiledMatchPattern = patternMatch != null ? Pattern.compile(patternMatch) : null;
    Preconditions.checkNotNull(ruleMetrics, "PreprocessorRuleMetrics can't be null");
    this.ruleMetrics = ruleMetrics;
    this.v2Predicate = v2Predicate != null ? v2Predicate : x -> true;
  }

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

      switch (scope) {
        case "metricName":
          if (compiledMatchPattern != null && !compiledMatchPattern.matcher(
              reportPoint.getMetric()).matches()) {
            break;
          }
          reportPoint.setMetric(reportPoint.getMetric().toLowerCase());
          ruleMetrics.incrementRuleAppliedCounter();
          break;
        case "sourceName": // source name is not case sensitive in Wavefront, but we'll do it anyway
          if (compiledMatchPattern != null && !compiledMatchPattern.matcher(
              reportPoint.getHost()).matches()) {
            break;
          }
          reportPoint.setHost(reportPoint.getHost().toLowerCase());
          ruleMetrics.incrementRuleAppliedCounter();
          break;
        default:
          if (reportPoint.getAnnotations() != null) {
            String tagValue = reportPoint.getAnnotations().get(scope);
            if (tagValue != null) {
              if (compiledMatchPattern != null && !compiledMatchPattern.matcher(tagValue).matches()) {
                break;
              }
              reportPoint.getAnnotations().put(scope, tagValue.toLowerCase());
              ruleMetrics.incrementRuleAppliedCounter();
            }
          }
      }
      return reportPoint;
    } finally {
      ruleMetrics.ruleEnd(startNanos);
    }
  }
}