com.wavefront.integrations.AbstractProxyConnectionHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-client Show documentation

Java client for sending data to Wavefront

There is a newer version: 4.36

package com.wavefront.integrations;

import com.wavefront.metrics.ReconnectingSocket;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.function.Supplier;

import javax.annotation.Nullable;
import javax.net.SocketFactory;

/**
 * Abstract base class for sending data to a Wavefront proxy.
 *
 * @author Clement Pang ([email protected]).
 * @author Vikram Raman ([email protected]).
 */
public abstract class AbstractProxyConnectionHandler implements WavefrontConnectionHandler {

  private final InetSocketAddress address;
  private final SocketFactory socketFactory;
  private volatile ReconnectingSocket reconnectingSocket;
  @Nullable
  private final Long connectionTimeToLiveMillis;
  @Nullable
  private final Supplier timeSupplier;

  protected AbstractProxyConnectionHandler(InetSocketAddress address, SocketFactory socketFactory) {
    this(address, socketFactory, null, null);
  }

  protected AbstractProxyConnectionHandler(InetSocketAddress address, SocketFactory socketFactory,
                                           @Nullable Long connectionTimeToLiveMillis,
                                           @Nullable Supplier timeSupplier) {
    this.address = address;
    this.socketFactory = socketFactory;
    this.connectionTimeToLiveMillis = connectionTimeToLiveMillis;
    this.timeSupplier = timeSupplier;
    this.reconnectingSocket = null;
  }

  @Override
  public synchronized void connect() throws IllegalStateException, IOException {
    if (reconnectingSocket != null) {
      throw new IllegalStateException("Already connected");
    }
    try {
      reconnectingSocket = new ReconnectingSocket(address.getHostName(), address.getPort(), socketFactory,
          connectionTimeToLiveMillis, timeSupplier);
    } catch (Exception e) {
      throw new IOException(e);
    }
  }

  @Override
  public boolean isConnected() {
    return reconnectingSocket != null;
  }

  @Override
  public void flush() throws IOException {
    if (reconnectingSocket != null) {
      reconnectingSocket.flush();
    }
  }

  @Override
  public synchronized void close() throws IOException {
    if (reconnectingSocket != null) {
      reconnectingSocket.close();
      reconnectingSocket = null;
    }
  }

  /**
   * Sends the given data to the Wavefront proxy.
   *
   * @param lineData line data in a Wavefront supported format
   * @throws Exception If there was failure sending the data
   */
  protected void sendData(String lineData) throws Exception {
    reconnectingSocket.write(lineData);
  }
}