R66ThriftServerService.java

/*
 * This file is part of Waarp Project (named also Waarp or GG).
 *
 *  Copyright (c) 2019, Waarp SAS, and individual contributors by the @author
 *  tags. See the COPYRIGHT.txt in the distribution for a full listing of
 * individual contributors.
 *
 *  All Waarp Project is free software: you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 *
 * Waarp is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 * A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along with
 * Waarp . If not, see <http://www.gnu.org/licenses/>.
 */
package org.waarp.openr66.thrift;

import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TThreadPoolServer;
import org.apache.thrift.transport.TServerSocket;
import org.apache.thrift.transport.TServerTransport;
import org.apache.thrift.transport.TTransportException;
import org.waarp.common.future.WaarpFuture;
import org.waarp.common.logging.WaarpLogger;
import org.waarp.common.logging.WaarpLoggerFactory;
import org.waarp.thrift.r66.R66Service;

import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.UnknownHostException;

/**
 * Main Thrift server service
 */
public class R66ThriftServerService implements Runnable {
  /**
   * Internal Logger
   */
  private static final WaarpLogger logger =
      WaarpLoggerFactory.getLogger(R66ThriftServerService.class);

  protected int port = 4266;
  protected TServerTransport serverTransport;
  protected TServer server;
  protected final WaarpFuture serviceReady;

  public R66ThriftServerService(final WaarpFuture serviceReady,
                                final int port) {
    this.serviceReady = serviceReady;
    this.port = port;
  }

  public final boolean awaitInitialization() {
    if (serviceReady != null) {
      serviceReady.awaitOrInterruptible();
      return serviceReady.isSuccess();
    }
    return true;
  }

  @Override
  public void run() {
    try {
      logger.warn("Will start Thrift service on port: " + port);
      final byte[] local = { 127, 0, 0, 1 };
      InetAddress addr;
      try {
        addr = InetAddress.getByAddress(local);
      } catch (final UnknownHostException e) {
        try {
          addr = InetAddress.getLocalHost();
        } catch (final UnknownHostException e1) {
          logger.error("Cannot start the Thrift service", e1);
          serviceReady.setFailure(e);
          releaseResources();
          return;
        }
      }
      final InetSocketAddress address = new InetSocketAddress(addr, port);
      serverTransport = new TServerSocket(address);
      final R66Service.Processor<R66EmbeddedServiceImpl> processor =
          new R66Service.Processor<R66EmbeddedServiceImpl>(
              new R66EmbeddedServiceImpl());
      server = new TThreadPoolServer(
          new TThreadPoolServer.Args(serverTransport).processor(processor));
      serviceReady.setSuccess();
      server.serve();
    } catch (final TTransportException e) {
      logger.error("An error occurs during initialization of Thrift support",
                   e);
      serviceReady.setFailure(e);
      releaseResources();
    }
  }

  public final void releaseResources() {
    if (server != null) {
      logger.debug("Stop Thrift Server");
      server.stop();
    }
    if (serverTransport != null) {
      logger.debug("Stop Thrift Transport");
      serverTransport.close();
    }
    logger.debug("Thrift stopped");
  }
}