com.wavefront.agent.handlers.EventSenderTask 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.agent.data.EntityProperties;
import com.wavefront.agent.data.EventDataSubmissionTask;
import com.wavefront.agent.data.QueueingReason;
import com.wavefront.agent.data.TaskResult;
import com.wavefront.agent.queueing.TaskQueue;
import com.wavefront.api.EventAPI;
import com.wavefront.dto.Event;

import javax.annotation.Nullable;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.ScheduledExecutorService;

/**
 * This class is responsible for accumulating events and sending them batch. This
 * class is similar to PostPushDataTimedTask.
 *
 * @author [email protected]
 */
class EventSenderTask extends AbstractSenderTask {

  private final EventAPI proxyAPI;
  private final UUID proxyId;
  private final TaskQueue backlog;

  /**
   * @param handlerKey   handler key, that serves as an identifier of the metrics pipeline.
   * @param proxyAPI     handles interaction with Wavefront servers as well as queueing.
   * @param proxyId      id of the proxy.
   * @param threadId     thread number.
   * @param properties   container for mutable proxy settings.
   * @param scheduler    executor service for running this task
   * @param backlog      backing queue
   */
  EventSenderTask(HandlerKey handlerKey, EventAPI proxyAPI, UUID proxyId, int threadId,
                  EntityProperties properties, ScheduledExecutorService scheduler,
                  TaskQueue backlog) {
    super(handlerKey, threadId, properties, scheduler);
    this.proxyAPI = proxyAPI;
    this.proxyId = proxyId;
    this.backlog = backlog;
  }

  @Override
  TaskResult processSingleBatch(List batch) {
    EventDataSubmissionTask task = new EventDataSubmissionTask(proxyAPI, proxyId, properties,
        backlog, handlerKey.getHandle(), batch, null);
    return task.execute();
  }

  @Override
  public void flushSingleBatch(List batch, @Nullable QueueingReason reason) {
    EventDataSubmissionTask task = new EventDataSubmissionTask(proxyAPI, proxyId, properties,
        backlog, handlerKey.getHandle(), batch, null);
    task.enqueue(reason);
  }
}