com.wavefront.agent.handlers.ReportPointHandlerImpl 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.api.agent.ValidationConfiguration;
import com.wavefront.common.Clock;
import com.wavefront.common.Utils;
import com.wavefront.data.DeltaCounterValueException;
import com.wavefront.ingester.ReportPointSerializer;
import com.yammer.metrics.Metrics;
import com.yammer.metrics.core.Counter;
import com.yammer.metrics.core.MetricName;
import com.yammer.metrics.core.MetricsRegistry;
import wavefront.report.Histogram;
import wavefront.report.ReportPoint;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.util.Collection;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.logging.Logger;

import static com.wavefront.data.Validation.validatePoint;

/**
 * Handler that processes incoming ReportPoint objects, validates them and hands them over to one of
 * the {@link SenderTask} threads.
 *
 * @author [email protected]
 */
class ReportPointHandlerImpl extends AbstractReportableEntityHandler {

  final Logger validItemsLogger;
  final ValidationConfiguration validationConfig;
  final Function recompressor;
  final com.yammer.metrics.core.Histogram receivedPointLag;
  final com.yammer.metrics.core.Histogram receivedTagCount;
  final Supplier discardedCounterSupplier;

  /**
   * Creates a new instance that handles either histograms or points.
   *
   * @param handlerKey           handler key for the metrics pipeline.
   * @param blockedItemsPerBatch controls sample rate of how many blocked points are written
   *                             into the main log file.
   * @param senderTasks          sender tasks.
   * @param validationConfig     validation configuration.
   * @param setupMetrics         Whether we should report counter metrics.
   * @param receivedRateSink     Where to report received rate.
   * @param blockedItemLogger    logger for blocked items (optional).
   * @param validItemsLogger     sampling logger for valid items (optional).
   * @param recompressor         histogram recompressor (optional)
   */
  ReportPointHandlerImpl(final HandlerKey handlerKey,
                         final int blockedItemsPerBatch,
                         @Nullable final Collection> senderTasks,
                         @Nonnull final ValidationConfiguration validationConfig,
                         final boolean setupMetrics,
                         @Nullable final Consumer receivedRateSink,
                         @Nullable final Logger blockedItemLogger,
                         @Nullable final Logger validItemsLogger,
                         @Nullable final Function recompressor) {
    super(handlerKey, blockedItemsPerBatch, new ReportPointSerializer(), senderTasks,
        setupMetrics, receivedRateSink, blockedItemLogger);
    this.validationConfig = validationConfig;
    this.validItemsLogger = validItemsLogger;
    this.recompressor = recompressor;
    MetricsRegistry registry = setupMetrics ? Metrics.defaultRegistry() : LOCAL_REGISTRY;
    this.receivedPointLag = registry.newHistogram(new MetricName(handlerKey.toString() +
        ".received", "", "lag"), false);
    this.receivedTagCount = registry.newHistogram(new MetricName(handlerKey.toString() +
        ".received", "", "tagCount"), false);
    this.discardedCounterSupplier = Utils.lazySupplier(() ->
        Metrics.newCounter(new MetricName(handlerKey.toString(), "", "discarded")));
  }

  @Override
  void reportInternal(ReportPoint point) {
    receivedTagCount.update(point.getAnnotations().size());
    try {
      validatePoint(point, validationConfig);
    } catch (DeltaCounterValueException e) {
      discardedCounterSupplier.get().inc();
      return;
    }
    receivedPointLag.update(Clock.now() - point.getTimestamp());
    if (point.getValue() instanceof Histogram && recompressor != null) {
      Histogram histogram = (Histogram) point.getValue();
      point.setValue(recompressor.apply(histogram));
    }
    final String strPoint = serializer.apply(point);
    getTask().add(strPoint);
    getReceivedCounter().inc();
    if (validItemsLogger != null) validItemsLogger.info(strPoint);
  }
}