com.wavefront.agent.handlers.InternalProxyWavefrontClient 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.handlers;

import com.wavefront.common.Clock;
import com.wavefront.data.ReportableEntityType;
import com.wavefront.sdk.common.Pair;
import com.wavefront.sdk.common.WavefrontSender;
import com.wavefront.sdk.entities.histograms.HistogramGranularity;
import com.wavefront.sdk.entities.tracing.SpanLog;
import wavefront.report.Histogram;
import wavefront.report.HistogramType;
import wavefront.report.ReportPoint;

import javax.annotation.Nullable;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import java.util.function.Supplier;
import java.util.stream.Collectors;

import static com.wavefront.common.Utils.lazySupplier;

public class InternalProxyWavefrontClient implements WavefrontSender {
  private final Supplier> pointHandlerSupplier;
  private final Supplier> histogramHandlerSupplier;
  private final String clientId;

  public InternalProxyWavefrontClient(ReportableEntityHandlerFactory handlerFactory,
                                      String handle) {
    this.pointHandlerSupplier = lazySupplier(() ->
        handlerFactory.getHandler(HandlerKey.of(ReportableEntityType.POINT, handle)));
    this.histogramHandlerSupplier = lazySupplier(() ->
        handlerFactory.getHandler(HandlerKey.of(ReportableEntityType.HISTOGRAM, handle)));
    this.clientId = handle;
  }

  @Override
  public void flush() {
    // noop
  }

  @Override
  public int getFailureCount() {
    return 0;
  }

  @Override
  public void sendDistribution(String name, List> centroids,
                               Set histogramGranularities, Long timestamp,
                               String source, Map tags) {
    final List bins = centroids.stream().map(x -> x._1).collect(Collectors.toList());
    final List counts = centroids.stream().map(x -> x._2).collect(Collectors.toList());
    for (HistogramGranularity granularity : histogramGranularities) {
      int duration;
      switch (granularity) {
        case MINUTE:
          duration = 60000;
          break;
        case HOUR:
          duration = 3600000;
          break;
        case DAY:
          duration = 86400000;
          break;
        default:
          throw new IllegalArgumentException("Unknown granularity: " + granularity);
      }
      Histogram histogram = Histogram.newBuilder().
          setType(HistogramType.TDIGEST).
          setBins(bins).
          setCounts(counts).
          setDuration(duration).
          build();
      ReportPoint point = ReportPoint.newBuilder().
          setTable("unknown").
          setMetric(name).
          setValue(histogram).
          setTimestamp(timestamp).
          setHost(source).
          setAnnotations(tags).
          build();
      histogramHandlerSupplier.get().report(point);
    }
  }

  @Override
  public void sendMetric(String name, double value, Long timestamp, String source,
                         Map tags) {
    // default to millis
    long timestampMillis = 0;
    timestamp = timestamp == null ? Clock.now() : timestamp;
    if (timestamp < 10_000_000_000L) {
      // seconds
      timestampMillis = timestamp * 1000;
    } else if (timestamp < 10_000_000_000_000L) {
      // millis
      timestampMillis = timestamp;
    } else if (timestamp < 10_000_000_000_000_000L) {
      // micros
      timestampMillis = timestamp / 1000;
    } else if (timestamp <= 999_999_999_999_999_999L) {
      // nanos
      timestampMillis = timestamp / 1000_000;
    }

    final ReportPoint point = ReportPoint.newBuilder().
        setTable("unknown").
        setMetric(name).
        setValue(value).
        setTimestamp(timestampMillis).
        setHost(source).
        setAnnotations(tags).
        build();
    pointHandlerSupplier.get().report(point);
  }

  public void sendFormattedMetric(String s) {
    throw new UnsupportedOperationException("Not applicable");
  }

  @Override
  public void sendSpan(String name, long startMillis, long durationMillis, String source, UUID traceId, UUID spanId,
                       List parents, List followsFrom, List> tags,
                       @Nullable List spanLogs) {
    throw new UnsupportedOperationException("Not applicable");
  }

  @Override
  public void close() {
    // noop
  }

  @Override
  public String getClientId() {
    return clientId;
  }

  @Override
  public void sendEvent(String name, long startMillis, long endMillis, String source, Map tags, Map annotations) throws IOException {
    throw new UnsupportedOperationException("Not applicable");
  }

  @Override
  public void sendLog(String name, double value, Long timestamp, String source, Map tags) throws IOException {
    throw new UnsupportedOperationException("Not applicable");
  }
}