com.wavefront.agent.SSLConnectionSocketFactoryImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of proxy Show documentation

Service for batching and relaying metric traffic to Wavefront

There is a newer version: 4.36

package com.wavefront.agent;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.Socket;
import org.apache.http.HttpHost;
import org.apache.http.conn.socket.LayeredConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.protocol.HttpContext;

/**
 * Delegated SSLConnectionSocketFactory that sets SoTimeout explicitly (for Apache HttpClient).
 *
 * @author [email protected]
 */
public class SSLConnectionSocketFactoryImpl implements LayeredConnectionSocketFactory {
  private final SSLConnectionSocketFactory delegate;
  private final int soTimeout;

  public SSLConnectionSocketFactoryImpl(SSLConnectionSocketFactory delegate, int soTimeoutMs) {
    this.delegate = delegate;
    this.soTimeout = soTimeoutMs;
  }

  @Override
  public Socket createSocket(HttpContext context) throws IOException {
    Socket socket1 = delegate.createSocket(context);
    socket1.setSoTimeout(soTimeout);
    return socket1;
  }

  @Override
  public Socket connectSocket(
      int connectTimeout,
      Socket sock,
      HttpHost host,
      InetSocketAddress remoteAddress,
      InetSocketAddress localAddress,
      HttpContext context)
      throws IOException {
    Socket socket1 =
        delegate.connectSocket(soTimeout, sock, host, remoteAddress, localAddress, context);
    socket1.setSoTimeout(soTimeout);
    return socket1;
  }

  @Override
  public Socket createLayeredSocket(Socket socket, String target, int port, HttpContext context)
      throws IOException {
    Socket socket1 = delegate.createLayeredSocket(socket, target, port, context);
    socket1.setSoTimeout(soTimeout);
    return socket1;
  }
}