Class OkHttpServer

okhttp/src/main/java/io/grpc/okhttp/OkHttpServer.java:42–189

import javax.net.ServerSocketFactory;

final class OkHttpServer implements InternalServer {
  private static final Logger log = Logger.getLogger(OkHttpServer.class.getName());

  private final SocketAddress originalListenAddress;
  private final ServerSocketFactory socketFactory;
  private final ObjectPool<Executor> transportExecutorPool;
  private final ObjectPool<ScheduledExecutorService> scheduledExecutorServicePool;
  private final OkHttpServerTransport.Config transportConfig;
  private final InternalChannelz channelz;
  private ServerSocket serverSocket;
  private SocketAddress actualListenAddress;
  private InternalInstrumented<InternalChannelz.SocketStats> listenInstrumented;
  private Executor transportExecutor;
  private ScheduledExecutorService scheduledExecutorService;
  private ServerListener listener;
  private boolean shutdown;

  public OkHttpServer(
      OkHttpServerBuilder builder,
      List<? extends ServerStreamTracer.Factory> streamTracerFactories,
      InternalChannelz channelz) {
    this.originalListenAddress = Preconditions.checkNotNull(builder.listenAddress, "listenAddress");
    this.socketFactory = Preconditions.checkNotNull(builder.socketFactory, "socketFactory");
    this.transportExecutorPool =
        Preconditions.checkNotNull(builder.transportExecutorPool, "transportExecutorPool");
    this.scheduledExecutorServicePool =
        Preconditions.checkNotNull(
            builder.scheduledExecutorServicePool, "scheduledExecutorServicePool");
    this.transportConfig = new OkHttpServerTransport.Config(builder, streamTracerFactories);
    this.channelz = Preconditions.checkNotNull(channelz, "channelz");
  }

  @Override
  public void start(ServerListener listener) throws IOException {
    this.listener = Preconditions.checkNotNull(listener, "listener");
    ServerSocket serverSocket = socketFactory.createServerSocket();
    try {
      serverSocket.bind(originalListenAddress);
    } catch (IOException t) {
      serverSocket.close();
      throw t;
    }

    this.serverSocket = serverSocket;
    this.actualListenAddress = serverSocket.getLocalSocketAddress();
    this.listenInstrumented = new ListenSocket(serverSocket);
    this.transportExecutor = transportExecutorPool.getObject();
    // Keep reference alive to avoid frequent re-creation by server transports
    this.scheduledExecutorService = scheduledExecutorServicePool.getObject();
    channelz.addListenSocket(this.listenInstrumented);
    transportExecutor.execute(this::acceptConnections);
  }

  private void acceptConnections() {
    try {
      while (true) {
        Socket socket;
        try {