/** Hadoop external communication class. */
public class HadoopExternalCommunication {
  /** IPC error message. */
  public static final String OUT_OF_RESOURCES_TCP_MSG =
      "Failed to allocate shared memory segment " + "(switching to TCP, may be slower).";

  /** Default port which node sets listener to (value is <tt>47100</tt>). */
  public static final int DFLT_PORT = 27100;

  /** Default connection timeout (value is <tt>1000</tt>ms). */
  public static final long DFLT_CONN_TIMEOUT = 1000;

  /** Default Maximum connection timeout (value is <tt>600,000</tt>ms). */
  public static final long DFLT_MAX_CONN_TIMEOUT = 10 * 60 * 1000;

  /** Default reconnect attempts count (value is <tt>10</tt>). */
  public static final int DFLT_RECONNECT_CNT = 10;

  /** Default message queue limit per connection (for incoming and outgoing . */
  public static final int DFLT_MSG_QUEUE_LIMIT = GridNioServer.DFLT_SEND_QUEUE_LIMIT;

  /**
   * Default count of selectors for TCP server equals to {@code "Math.min(4,
   * Runtime.getRuntime().availableProcessors())"}.
   */
  public static final int DFLT_SELECTORS_CNT = 1;

  /** Node ID meta for session. */
  private static final int PROCESS_META = GridNioSessionMetaKey.nextUniqueKey();

  /** Handshake timeout meta for session. */
  private static final int HANDSHAKE_FINISH_META = GridNioSessionMetaKey.nextUniqueKey();

  /** Message tracker meta for session. */
  private static final int TRACKER_META = GridNioSessionMetaKey.nextUniqueKey();

  /**
   * Default local port range (value is <tt>100</tt>). See {@link #setLocalPortRange(int)} for
   * details.
   */
  public static final int DFLT_PORT_RANGE = 100;

  /** Default value for {@code TCP_NODELAY} socket option (value is <tt>true</tt>). */
  public static final boolean DFLT_TCP_NODELAY = true;

  /** Server listener. */
  private final GridNioServerListener<HadoopMessage> srvLsnr =
      new GridNioServerListenerAdapter<HadoopMessage>() {
        @Override
        public void onConnected(GridNioSession ses) {
          HadoopProcessDescriptor desc = ses.meta(PROCESS_META);

          assert desc != null
              : "Received connected notification without finished handshake: " + ses;
        }

        /** {@inheritDoc} */
        @Override
        public void onDisconnected(GridNioSession ses, @Nullable Exception e) {
          if (log.isDebugEnabled()) log.debug("Closed connection for session: " + ses);

          if (e != null) U.error(log, "Session disconnected due to exception: " + ses, e);

          HadoopProcessDescriptor desc = ses.meta(PROCESS_META);

          if (desc != null) {
            HadoopCommunicationClient rmv = clients.remove(desc.processId());

            if (rmv != null) rmv.forceClose();
          }

          HadoopMessageListener lsnr0 = lsnr;

          if (lsnr0 != null)
            // Notify listener about connection close.
            lsnr0.onConnectionLost(desc);
        }

        /** {@inheritDoc} */
        @Override
        public void onMessage(GridNioSession ses, HadoopMessage msg) {
          notifyListener(ses.<HadoopProcessDescriptor>meta(PROCESS_META), msg);

          if (msgQueueLimit > 0) {
            GridNioMessageTracker tracker = ses.meta(TRACKER_META);

            assert tracker != null : "Missing tracker for limited message queue: " + ses;

            tracker.run();
          }
        }
      };

  /** Logger. */
  private IgniteLogger log;

  /** Local process descriptor. */
  private HadoopProcessDescriptor locProcDesc;

  /** Marshaller. */
  private Marshaller marsh;

  /** Message notification executor service. */
  private ExecutorService execSvc;

  /** Grid name. */
  private String gridName;

  /** Complex variable that represents this node IP address. */
  private volatile InetAddress locHost;

  /** Local port which node uses. */
  private int locPort = DFLT_PORT;

  /** Local port range. */
  private int locPortRange = DFLT_PORT_RANGE;

  /** Local port which node uses to accept shared memory connections. */
  private int shmemPort = -1;

  /** Allocate direct buffer or heap buffer. */
  private boolean directBuf = true;

  /** Connect timeout. */
  private long connTimeout = DFLT_CONN_TIMEOUT;

  /** Maximum connect timeout. */
  private long maxConnTimeout = DFLT_MAX_CONN_TIMEOUT;

  /** Reconnect attempts count. */
  @SuppressWarnings({"FieldAccessedSynchronizedAndUnsynchronized"})
  private int reconCnt = DFLT_RECONNECT_CNT;

  /** Socket send buffer. */
  private int sockSndBuf;

  /** Socket receive buffer. */
  private int sockRcvBuf;

  /** Message queue limit. */
  private int msgQueueLimit = DFLT_MSG_QUEUE_LIMIT;

  /** NIO server. */
  private GridNioServer<HadoopMessage> nioSrvr;

  /** Shared memory server. */
  private IpcSharedMemoryServerEndpoint shmemSrv;

  /** {@code TCP_NODELAY} option value for created sockets. */
  private boolean tcpNoDelay = DFLT_TCP_NODELAY;

  /** Shared memory accept worker. */
  private ShmemAcceptWorker shmemAcceptWorker;

  /** Shared memory workers. */
  private final Collection<ShmemWorker> shmemWorkers = new ConcurrentLinkedDeque8<>();

  /** Clients. */
  private final ConcurrentMap<UUID, HadoopCommunicationClient> clients =
      GridConcurrentFactory.newMap();

  /** Message listener. */
  private volatile HadoopMessageListener lsnr;

  /** Bound port. */
  private int boundTcpPort = -1;

  /** Bound port for shared memory server. */
  private int boundTcpShmemPort = -1;

  /** Count of selectors to use in TCP server. */
  private int selectorsCnt = DFLT_SELECTORS_CNT;

  /** Local node ID message. */
  private ProcessHandshakeMessage locIdMsg;

  /** Locks. */
  private final GridKeyLock locks = new GridKeyLock();

  /**
   * @param parentNodeId Parent node ID.
   * @param procId Process ID.
   * @param marsh Marshaller to use.
   * @param log Logger.
   * @param execSvc Executor service for message notification.
   * @param gridName Grid name.
   */
  public HadoopExternalCommunication(
      UUID parentNodeId,
      UUID procId,
      Marshaller marsh,
      IgniteLogger log,
      ExecutorService execSvc,
      String gridName) {
    locProcDesc = new HadoopProcessDescriptor(parentNodeId, procId);

    this.marsh = marsh;
    this.log = log.getLogger(HadoopExternalCommunication.class);
    this.execSvc = execSvc;
    this.gridName = gridName;
  }

  /**
   * Sets local port for socket binding.
   *
   * <p>If not provided, default value is {@link #DFLT_PORT}.
   *
   * @param locPort Port number.
   */
  public void setLocalPort(int locPort) {
    this.locPort = locPort;
  }

  /**
   * Gets local port for socket binding.
   *
   * @return Local port.
   */
  public int getLocalPort() {
    return locPort;
  }

  /**
   * Sets local port range for local host ports (value must greater than or equal to <tt>0</tt>). If
   * provided local port (see {@link #setLocalPort(int)}} is occupied, implementation will try to
   * increment the port number for as long as it is less than initial value plus this range.
   *
   * <p>If port range value is <tt>0</tt>, then implementation will try bind only to the port
   * provided by {@link #setLocalPort(int)} method and fail if binding to this port did not succeed.
   *
   * <p>Local port range is very useful during development when more than one grid nodes need to run
   * on the same physical machine.
   *
   * <p>If not provided, default value is {@link #DFLT_PORT_RANGE}.
   *
   * @param locPortRange New local port range.
   */
  public void setLocalPortRange(int locPortRange) {
    this.locPortRange = locPortRange;
  }

  /** @return Local port range. */
  public int getLocalPortRange() {
    return locPortRange;
  }

  /**
   * Sets local port to accept shared memory connections.
   *
   * <p>If set to {@code -1} shared memory communication will be disabled.
   *
   * <p>If not provided, shared memory is disabled.
   *
   * @param shmemPort Port number.
   */
  public void setSharedMemoryPort(int shmemPort) {
    this.shmemPort = shmemPort;
  }

  /**
   * Gets shared memory port to accept incoming connections.
   *
   * @return Shared memory port.
   */
  public int getSharedMemoryPort() {
    return shmemPort;
  }

  /**
   * Sets connect timeout used when establishing connection with remote nodes.
   *
   * <p>{@code 0} is interpreted as infinite timeout.
   *
   * <p>If not provided, default value is {@link #DFLT_CONN_TIMEOUT}.
   *
   * @param connTimeout Connect timeout.
   */
  public void setConnectTimeout(long connTimeout) {
    this.connTimeout = connTimeout;
  }

  /** @return Connection timeout. */
  public long getConnectTimeout() {
    return connTimeout;
  }

  /**
   * Sets maximum connect timeout. If handshake is not established within connect timeout, then SPI
   * tries to repeat handshake procedure with increased connect timeout. Connect timeout can grow
   * till maximum timeout value, if maximum timeout value is reached then the handshake is
   * considered as failed.
   *
   * <p>{@code 0} is interpreted as infinite timeout.
   *
   * <p>If not provided, default value is {@link #DFLT_MAX_CONN_TIMEOUT}.
   *
   * @param maxConnTimeout Maximum connect timeout.
   */
  public void setMaxConnectTimeout(long maxConnTimeout) {
    this.maxConnTimeout = maxConnTimeout;
  }

  /**
   * Gets maximum connection timeout.
   *
   * @return Maximum connection timeout.
   */
  public long getMaxConnectTimeout() {
    return maxConnTimeout;
  }

  /**
   * Sets maximum number of reconnect attempts used when establishing connection with remote nodes.
   *
   * <p>If not provided, default value is {@link #DFLT_RECONNECT_CNT}.
   *
   * @param reconCnt Maximum number of reconnection attempts.
   */
  public void setReconnectCount(int reconCnt) {
    this.reconCnt = reconCnt;
  }

  /** @return Reconnect count. */
  public int getReconnectCount() {
    return reconCnt;
  }

  /**
   * Sets flag to allocate direct or heap buffer in SPI. If value is {@code true}, then SPI will use
   * {@link ByteBuffer#allocateDirect(int)} call. Otherwise, SPI will use {@link
   * ByteBuffer#allocate(int)} call.
   *
   * <p>If not provided, default value is {@code true}.
   *
   * @param directBuf Flag indicates to allocate direct or heap buffer in SPI.
   */
  public void setDirectBuffer(boolean directBuf) {
    this.directBuf = directBuf;
  }

  /** @return Direct buffer flag. */
  public boolean isDirectBuffer() {
    return directBuf;
  }

  /**
   * Sets the count of selectors te be used in TCP server.
   *
   * <p>If not provided, default value is {@link #DFLT_SELECTORS_CNT}.
   *
   * @param selectorsCnt Selectors count.
   */
  public void setSelectorsCount(int selectorsCnt) {
    this.selectorsCnt = selectorsCnt;
  }

  /** @return Number of selectors to use. */
  public int getSelectorsCount() {
    return selectorsCnt;
  }

  /**
   * Sets value for {@code TCP_NODELAY} socket option. Each socket will be opened using provided
   * value.
   *
   * <p>Setting this option to {@code true} disables Nagle's algorithm for socket decreasing latency
   * and delivery time for small messages.
   *
   * <p>For systems that work under heavy network load it is advisable to set this value to {@code
   * false}.
   *
   * <p>If not provided, default value is {@link #DFLT_TCP_NODELAY}.
   *
   * @param tcpNoDelay {@code True} to disable TCP delay.
   */
  public void setTcpNoDelay(boolean tcpNoDelay) {
    this.tcpNoDelay = tcpNoDelay;
  }

  /** @return {@code TCP_NO_DELAY} flag. */
  public boolean isTcpNoDelay() {
    return tcpNoDelay;
  }

  /**
   * Sets receive buffer size for sockets created or accepted by this SPI.
   *
   * <p>If not provided, default is {@code 0} which leaves buffer unchanged after socket creation
   * (OS defaults).
   *
   * @param sockRcvBuf Socket receive buffer size.
   */
  public void setSocketReceiveBuffer(int sockRcvBuf) {
    this.sockRcvBuf = sockRcvBuf;
  }

  /** @return Socket receive buffer size. */
  public int getSocketReceiveBuffer() {
    return sockRcvBuf;
  }

  /**
   * Sets send buffer size for sockets created or accepted by this SPI.
   *
   * <p>If not provided, default is {@code 0} which leaves the buffer unchanged after socket
   * creation (OS defaults).
   *
   * @param sockSndBuf Socket send buffer size.
   */
  public void setSocketSendBuffer(int sockSndBuf) {
    this.sockSndBuf = sockSndBuf;
  }

  /** @return Socket send buffer size. */
  public int getSocketSendBuffer() {
    return sockSndBuf;
  }

  /**
   * Sets message queue limit for incoming and outgoing messages.
   *
   * <p>When set to positive number send queue is limited to the configured value. {@code 0}
   * disables the size limitations.
   *
   * <p>If not provided, default is {@link #DFLT_MSG_QUEUE_LIMIT}.
   *
   * @param msgQueueLimit Send queue size limit.
   */
  public void setMessageQueueLimit(int msgQueueLimit) {
    this.msgQueueLimit = msgQueueLimit;
  }

  /** @return Message queue size limit. */
  public int getMessageQueueLimit() {
    return msgQueueLimit;
  }

  /**
   * Sets Hadoop communication message listener.
   *
   * @param lsnr Message listener.
   */
  public void setListener(HadoopMessageListener lsnr) {
    this.lsnr = lsnr;
  }

  /** @return Outbound message queue size. */
  public int getOutboundMessagesQueueSize() {
    return nioSrvr.outboundMessagesQueueSize();
  }

  /**
   * Starts communication.
   *
   * @throws IgniteCheckedException If failed.
   */
  public void start() throws IgniteCheckedException {
    try {
      locHost = U.getLocalHost();
    } catch (IOException e) {
      throw new IgniteCheckedException("Failed to initialize local address.", e);
    }

    try {
      shmemSrv = resetShmemServer();
    } catch (IgniteCheckedException e) {
      U.warn(log, "Failed to start shared memory communication server.", e);
    }

    try {
      // This method potentially resets local port to the value
      // local node was bound to.
      nioSrvr = resetNioServer();
    } catch (IgniteCheckedException e) {
      throw new IgniteCheckedException("Failed to initialize TCP server: " + locHost, e);
    }

    locProcDesc.address(locHost.getHostAddress());
    locProcDesc.sharedMemoryPort(boundTcpShmemPort);
    locProcDesc.tcpPort(boundTcpPort);

    locIdMsg = new ProcessHandshakeMessage(locProcDesc);

    if (shmemSrv != null) {
      shmemAcceptWorker = new ShmemAcceptWorker(shmemSrv);

      new IgniteThread(shmemAcceptWorker).start();
    }

    nioSrvr.start();
  }

  /**
   * Gets local process descriptor.
   *
   * @return Local process descriptor.
   */
  public HadoopProcessDescriptor localProcessDescriptor() {
    return locProcDesc;
  }

  /**
   * Gets filters used by communication.
   *
   * @return Filters array.
   */
  private GridNioFilter[] filters() {
    return new GridNioFilter[] {
      new GridNioAsyncNotifyFilter(gridName, execSvc, log),
      new HandshakeAndBackpressureFilter(),
      new HadoopMarshallerFilter(marsh),
      new GridNioCodecFilter(new GridBufferedParser(directBuf, ByteOrder.nativeOrder()), log, false)
    };
  }

  /**
   * Recreates tpcSrvr socket instance.
   *
   * @return Server instance.
   * @throws IgniteCheckedException Thrown if it's not possible to create server.
   */
  private GridNioServer<HadoopMessage> resetNioServer() throws IgniteCheckedException {
    if (boundTcpPort >= 0)
      throw new IgniteCheckedException(
          "Tcp NIO server was already created on port " + boundTcpPort);

    IgniteCheckedException lastEx = null;

    // If configured TCP port is busy, find first available in range.
    for (int port = locPort; port < locPort + locPortRange; port++) {
      try {
        GridNioServer<HadoopMessage> srvr =
            GridNioServer.<HadoopMessage>builder()
                .address(locHost)
                .port(port)
                .listener(srvLsnr)
                .logger(log.getLogger(GridNioServer.class))
                .selectorCount(selectorsCnt)
                .gridName(gridName)
                .tcpNoDelay(tcpNoDelay)
                .directBuffer(directBuf)
                .byteOrder(ByteOrder.nativeOrder())
                .socketSendBufferSize(sockSndBuf)
                .socketReceiveBufferSize(sockRcvBuf)
                .sendQueueLimit(msgQueueLimit)
                .directMode(false)
                .filters(filters())
                .build();

        boundTcpPort = port;

        // Ack Port the TCP server was bound to.
        if (log.isInfoEnabled())
          log.info(
              "Successfully bound to TCP port [port="
                  + boundTcpPort
                  + ", locHost="
                  + locHost
                  + ']');

        return srvr;
      } catch (IgniteCheckedException e) {
        lastEx = e;

        if (log.isDebugEnabled())
          log.debug(
              "Failed to bind to local port (will try next port within range) [port="
                  + port
                  + ", locHost="
                  + locHost
                  + ']');
      }
    }

    // If free port wasn't found.
    throw new IgniteCheckedException(
        "Failed to bind to any port within range [startPort="
            + locPort
            + ", portRange="
            + locPortRange
            + ", locHost="
            + locHost
            + ']',
        lastEx);
  }

  /**
   * Creates new shared memory communication server.
   *
   * @return Server.
   * @throws IgniteCheckedException If failed.
   */
  @Nullable
  private IpcSharedMemoryServerEndpoint resetShmemServer() throws IgniteCheckedException {
    if (boundTcpShmemPort >= 0)
      throw new IgniteCheckedException(
          "Shared memory server was already created on port " + boundTcpShmemPort);

    if (shmemPort == -1 || U.isWindows()) return null;

    IgniteCheckedException lastEx = null;

    // If configured TCP port is busy, find first available in range.
    for (int port = shmemPort; port < shmemPort + locPortRange; port++) {
      try {
        IpcSharedMemoryServerEndpoint srv =
            new IpcSharedMemoryServerEndpoint(
                log.getLogger(IpcSharedMemoryServerEndpoint.class),
                locProcDesc.processId(),
                gridName);

        srv.setPort(port);

        srv.omitOutOfResourcesWarning(true);

        srv.start();

        boundTcpShmemPort = port;

        // Ack Port the TCP server was bound to.
        if (log.isInfoEnabled())
          log.info(
              "Successfully bound shared memory communication to TCP port [port="
                  + boundTcpShmemPort
                  + ", locHost="
                  + locHost
                  + ']');

        return srv;
      } catch (IgniteCheckedException e) {
        lastEx = e;

        if (log.isDebugEnabled())
          log.debug(
              "Failed to bind to local port (will try next port within range) [port="
                  + port
                  + ", locHost="
                  + locHost
                  + ']');
      }
    }

    // If free port wasn't found.
    throw new IgniteCheckedException(
        "Failed to bind shared memory communication to any port within range [startPort="
            + locPort
            + ", portRange="
            + locPortRange
            + ", locHost="
            + locHost
            + ']',
        lastEx);
  }

  /**
   * Stops the server.
   *
   * @throws IgniteCheckedException
   */
  public void stop() throws IgniteCheckedException {
    // Stop TCP server.
    if (nioSrvr != null) nioSrvr.stop();

    U.cancel(shmemAcceptWorker);
    U.join(shmemAcceptWorker, log);

    U.cancel(shmemWorkers);
    U.join(shmemWorkers, log);

    shmemWorkers.clear();

    // Force closing on stop (safety).
    for (HadoopCommunicationClient client : clients.values()) client.forceClose();

    // Clear resources.
    nioSrvr = null;

    boundTcpPort = -1;
  }

  /**
   * Sends message to Hadoop process.
   *
   * @param desc
   * @param msg
   * @throws IgniteCheckedException
   */
  public void sendMessage(HadoopProcessDescriptor desc, HadoopMessage msg)
      throws IgniteCheckedException {
    assert desc != null;
    assert msg != null;

    if (log.isTraceEnabled())
      log.trace("Sending message to Hadoop process [desc=" + desc + ", msg=" + msg + ']');

    HadoopCommunicationClient client = null;

    boolean closeOnRelease = true;

    try {
      client = reserveClient(desc);

      client.sendMessage(desc, msg);

      closeOnRelease = false;
    } finally {
      if (client != null) {
        if (closeOnRelease) {
          client.forceClose();

          clients.remove(desc.processId(), client);
        } else client.release();
      }
    }
  }

  /**
   * Returns existing or just created client to node.
   *
   * @param desc Node to which client should be open.
   * @return The existing or just created client.
   * @throws IgniteCheckedException Thrown if any exception occurs.
   */
  private HadoopCommunicationClient reserveClient(HadoopProcessDescriptor desc)
      throws IgniteCheckedException {
    assert desc != null;

    UUID procId = desc.processId();

    while (true) {
      HadoopCommunicationClient client = clients.get(procId);

      if (client == null) {
        if (log.isDebugEnabled())
          log.debug(
              "Did not find client for remote process [locProcDesc="
                  + locProcDesc
                  + ", desc="
                  + desc
                  + ']');

        // Do not allow concurrent connects.
        Object sync = locks.lock(procId);

        try {
          client = clients.get(procId);

          if (client == null) {
            HadoopCommunicationClient old = clients.put(procId, client = createNioClient(desc));

            assert old == null;
          }
        } finally {
          locks.unlock(procId, sync);
        }

        assert client != null;
      }

      if (client.reserve()) return client;
      else
        // Client has just been closed by idle worker. Help it and try again.
        clients.remove(procId, client);
    }
  }

  /**
   * @param desc Process descriptor.
   * @return Client.
   * @throws IgniteCheckedException If failed.
   */
  @Nullable
  protected HadoopCommunicationClient createNioClient(HadoopProcessDescriptor desc)
      throws IgniteCheckedException {
    assert desc != null;

    int shmemPort = desc.sharedMemoryPort();

    // If remote node has shared memory server enabled and has the same set of MACs
    // then we are likely to run on the same host and shared memory communication could be tried.
    if (shmemPort != -1 && locProcDesc.parentNodeId().equals(desc.parentNodeId())) {
      try {
        return createShmemClient(desc, shmemPort);
      } catch (IgniteCheckedException e) {
        if (e.hasCause(IpcOutOfSystemResourcesException.class))
          // Has cause or is itself the IpcOutOfSystemResourcesException.
          LT.warn(log, null, OUT_OF_RESOURCES_TCP_MSG);
        else if (log.isDebugEnabled())
          log.debug(
              "Failed to establish shared memory connection with local hadoop process: " + desc);
      }
    }

    return createTcpClient(desc);
  }

  /**
   * @param desc Process descriptor.
   * @param port Port.
   * @return Client.
   * @throws IgniteCheckedException If failed.
   */
  @Nullable
  protected HadoopCommunicationClient createShmemClient(HadoopProcessDescriptor desc, int port)
      throws IgniteCheckedException {
    int attempt = 1;

    int connectAttempts = 1;

    long connTimeout0 = connTimeout;

    while (true) {
      IpcEndpoint clientEndpoint;

      try {
        clientEndpoint = new IpcSharedMemoryClientEndpoint(port, (int) connTimeout, log);
      } catch (IgniteCheckedException e) {
        // Reconnect for the second time, if connection is not established.
        if (connectAttempts < 2 && X.hasCause(e, ConnectException.class)) {
          connectAttempts++;

          continue;
        }

        throw e;
      }

      HadoopCommunicationClient client = null;

      try {
        ShmemWorker worker = new ShmemWorker(clientEndpoint, false);

        shmemWorkers.add(worker);

        GridNioSession ses = worker.session();

        HandshakeFinish fin = new HandshakeFinish();

        // We are in lock, it is safe to get session and attach
        ses.addMeta(HANDSHAKE_FINISH_META, fin);

        client = new HadoopTcpNioCommunicationClient(ses);

        new IgniteThread(worker).start();

        fin.await(connTimeout0);
      } catch (HadoopHandshakeTimeoutException e) {
        if (log.isDebugEnabled())
          log.debug(
              "Handshake timed out (will retry with increased timeout) [timeout="
                  + connTimeout0
                  + ", err="
                  + e.getMessage()
                  + ", client="
                  + client
                  + ']');

        if (client != null) client.forceClose();

        if (attempt == reconCnt || connTimeout0 > maxConnTimeout) {
          if (log.isDebugEnabled())
            log.debug(
                "Handshake timedout (will stop attempts to perform the handshake) "
                    + "[timeout="
                    + connTimeout0
                    + ", maxConnTimeout="
                    + maxConnTimeout
                    + ", attempt="
                    + attempt
                    + ", reconCnt="
                    + reconCnt
                    + ", err="
                    + e.getMessage()
                    + ", client="
                    + client
                    + ']');

          throw e;
        } else {
          attempt++;

          connTimeout0 *= 2;

          continue;
        }
      } catch (RuntimeException | Error e) {
        if (log.isDebugEnabled())
          log.debug(
              "Caught exception (will close client) [err="
                  + e.getMessage()
                  + ", client="
                  + client
                  + ']');

        if (client != null) client.forceClose();

        throw e;
      }

      return client;
    }
  }

  /**
   * Establish TCP connection to remote hadoop process and returns client.
   *
   * @param desc Process descriptor.
   * @return Client.
   * @throws IgniteCheckedException If failed.
   */
  protected HadoopCommunicationClient createTcpClient(HadoopProcessDescriptor desc)
      throws IgniteCheckedException {
    String addr = desc.address();

    int port = desc.tcpPort();

    if (log.isDebugEnabled())
      log.debug(
          "Trying to connect to remote process [locProcDesc="
              + locProcDesc
              + ", desc="
              + desc
              + ']');

    boolean conn = false;
    HadoopTcpNioCommunicationClient client = null;
    IgniteCheckedException errs = null;

    int connectAttempts = 1;

    long connTimeout0 = connTimeout;

    int attempt = 1;

    while (!conn) { // Reconnection on handshake timeout.
      try {
        SocketChannel ch = SocketChannel.open();

        ch.configureBlocking(true);

        ch.socket().setTcpNoDelay(tcpNoDelay);
        ch.socket().setKeepAlive(true);

        if (sockRcvBuf > 0) ch.socket().setReceiveBufferSize(sockRcvBuf);

        if (sockSndBuf > 0) ch.socket().setSendBufferSize(sockSndBuf);

        ch.socket().connect(new InetSocketAddress(addr, port), (int) connTimeout);

        HandshakeFinish fin = new HandshakeFinish();

        GridNioSession ses = nioSrvr.createSession(ch, F.asMap(HANDSHAKE_FINISH_META, fin)).get();

        client = new HadoopTcpNioCommunicationClient(ses);

        if (log.isDebugEnabled()) log.debug("Waiting for handshake finish for client: " + client);

        fin.await(connTimeout0);

        conn = true;
      } catch (HadoopHandshakeTimeoutException e) {
        if (client != null) {
          client.forceClose();

          client = null;
        }

        if (log.isDebugEnabled())
          log.debug(
              "Handshake timedout (will retry with increased timeout) [timeout="
                  + connTimeout0
                  + ", desc="
                  + desc
                  + ", port="
                  + port
                  + ", err="
                  + e
                  + ']');

        if (attempt == reconCnt || connTimeout0 > maxConnTimeout) {
          if (log.isDebugEnabled())
            log.debug(
                "Handshake timed out (will stop attempts to perform the handshake) "
                    + "[timeout="
                    + connTimeout0
                    + ", maxConnTimeout="
                    + maxConnTimeout
                    + ", attempt="
                    + attempt
                    + ", reconCnt="
                    + reconCnt
                    + ", err="
                    + e.getMessage()
                    + ", addr="
                    + addr
                    + ']');

          if (errs == null)
            errs =
                new IgniteCheckedException(
                    "Failed to connect to remote Hadoop process "
                        + "(is process still running?) [desc="
                        + desc
                        + ", addrs="
                        + addr
                        + ']');

          errs.addSuppressed(e);

          break;
        } else {
          attempt++;

          connTimeout0 *= 2;

          // Continue loop.
        }
      } catch (Exception e) {
        if (client != null) {
          client.forceClose();

          client = null;
        }

        if (log.isDebugEnabled())
          log.debug("Client creation failed [addr=" + addr + ", port=" + port + ", err=" + e + ']');

        if (X.hasCause(e, SocketTimeoutException.class))
          LT.warn(
              log,
              null,
              "Connect timed out (consider increasing 'connTimeout' "
                  + "configuration property) [addr="
                  + addr
                  + ", port="
                  + port
                  + ']');

        if (errs == null)
          errs =
              new IgniteCheckedException(
                  "Failed to connect to remote Hadoop process (is process still running?) "
                      + "[desc="
                      + desc
                      + ", addrs="
                      + addr
                      + ']');

        errs.addSuppressed(e);

        // Reconnect for the second time, if connection is not established.
        if (connectAttempts < 2
            && (e instanceof ConnectException || X.hasCause(e, ConnectException.class))) {
          connectAttempts++;

          continue;
        }

        break;
      }
    }

    if (client == null) {
      assert errs != null;

      if (X.hasCause(errs, ConnectException.class))
        LT.warn(
            log,
            null,
            "Failed to connect to a remote Hadoop process (is process still running?). "
                + "Make sure operating system firewall is disabled on local and remote host) "
                + "[addrs="
                + addr
                + ", port="
                + port
                + ']');

      throw errs;
    }

    if (log.isDebugEnabled()) log.debug("Created client: " + client);

    return client;
  }

  /**
   * @param desc Sender process descriptor.
   * @param msg Communication message.
   */
  protected void notifyListener(HadoopProcessDescriptor desc, HadoopMessage msg) {
    HadoopMessageListener lsnr = this.lsnr;

    if (lsnr != null)
      // Notify listener of a new message.
      lsnr.onMessageReceived(desc, msg);
    else if (log.isDebugEnabled())
      log.debug(
          "Received communication message without any registered listeners (will ignore) "
              + "[senderProcDesc="
              + desc
              + ", msg="
              + msg
              + ']');
  }

  /** {@inheritDoc} */
  @Override
  public String toString() {
    return S.toString(HadoopExternalCommunication.class, this);
  }

  /**
   * This worker takes responsibility to shut the server down when stopping, No other thread shall
   * stop passed server.
   */
  private class ShmemAcceptWorker extends GridWorker {
    /** */
    private final IpcSharedMemoryServerEndpoint srv;

    /** @param srv Server. */
    ShmemAcceptWorker(IpcSharedMemoryServerEndpoint srv) {
      super(gridName, "shmem-communication-acceptor", HadoopExternalCommunication.this.log);

      this.srv = srv;
    }

    /** {@inheritDoc} */
    @Override
    protected void body() throws InterruptedException {
      try {
        while (!Thread.interrupted()) {
          ShmemWorker e = new ShmemWorker(srv.accept(), true);

          shmemWorkers.add(e);

          new IgniteThread(e).start();
        }
      } catch (IgniteCheckedException e) {
        if (!isCancelled()) U.error(log, "Shmem server failed.", e);
      } finally {
        srv.close();
      }
    }

    /** {@inheritDoc} */
    @Override
    public void cancel() {
      super.cancel();

      srv.close();
    }
  }

  /** */
  private class ShmemWorker extends GridWorker {
    /** */
    private final IpcEndpoint endpoint;

    /** Adapter. */
    private HadoopIpcToNioAdapter<HadoopMessage> adapter;

    /** @param endpoint Endpoint. */
    private ShmemWorker(IpcEndpoint endpoint, boolean accepted) {
      super(gridName, "shmem-worker", HadoopExternalCommunication.this.log);

      this.endpoint = endpoint;

      adapter =
          new HadoopIpcToNioAdapter<>(
              HadoopExternalCommunication.this.log, endpoint, accepted, srvLsnr, filters());
    }

    /** {@inheritDoc} */
    @Override
    protected void body() throws InterruptedException {
      try {
        adapter.serve();
      } finally {
        shmemWorkers.remove(this);

        endpoint.close();
      }
    }

    /** {@inheritDoc} */
    @Override
    public void cancel() {
      super.cancel();

      endpoint.close();
    }

    /** @{@inheritDoc} */
    @Override
    protected void cleanup() {
      super.cleanup();

      endpoint.close();
    }

    /** @{@inheritDoc} */
    @Override
    public String toString() {
      return S.toString(ShmemWorker.class, this);
    }

    /** @return NIO session for this worker. */
    public GridNioSession session() {
      return adapter.session();
    }
  }

  /** */
  private static class HandshakeFinish {
    /** Await latch. */
    private CountDownLatch latch = new CountDownLatch(1);

    /** Finishes handshake. */
    public void finish() {
      latch.countDown();
    }

    /**
     * @param time Time to wait.
     * @throws HadoopHandshakeTimeoutException If failed to wait.
     */
    public void await(long time) throws HadoopHandshakeTimeoutException {
      try {
        if (!latch.await(time, TimeUnit.MILLISECONDS))
          throw new HadoopHandshakeTimeoutException(
              "Failed to wait for handshake to finish [timeout=" + time + ']');
      } catch (InterruptedException e) {
        Thread.currentThread().interrupt();

        throw new HadoopHandshakeTimeoutException(
            "Failed to wait for handshake to finish (thread was "
                + "interrupted) [timeout="
                + time
                + ']',
            e);
      }
    }
  }

  /** */
  private class HandshakeAndBackpressureFilter extends GridNioFilterAdapter {
    /** Assigns filter name to a filter. */
    protected HandshakeAndBackpressureFilter() {
      super("HadoopHandshakeFilter");
    }

    /** {@inheritDoc} */
    @Override
    public void onSessionOpened(final GridNioSession ses) throws IgniteCheckedException {
      if (ses.accepted()) {
        if (log.isDebugEnabled()) log.debug("Accepted connection, initiating handshake: " + ses);

        // Server initiates handshake.
        ses.send(locIdMsg)
            .listen(
                new CI1<IgniteInternalFuture<?>>() {
                  @Override
                  public void apply(IgniteInternalFuture<?> fut) {
                    try {
                      // Make sure there were no errors.
                      fut.get();
                    } catch (IgniteCheckedException e) {
                      log.warning(
                          "Failed to send handshake message, will close session: " + ses, e);

                      ses.close();
                    }
                  }
                });
      }
    }

    /** {@inheritDoc} */
    @Override
    public void onSessionClosed(GridNioSession ses) throws IgniteCheckedException {
      proceedSessionClosed(ses);
    }

    /** {@inheritDoc} */
    @Override
    public void onExceptionCaught(GridNioSession ses, IgniteCheckedException ex)
        throws IgniteCheckedException {
      proceedExceptionCaught(ses, ex);
    }

    /** {@inheritDoc} */
    @Override
    public GridNioFuture<?> onSessionWrite(GridNioSession ses, Object msg)
        throws IgniteCheckedException {
      if (ses.meta(PROCESS_META) == null && !(msg instanceof ProcessHandshakeMessage))
        log.warning(
            "Writing message before handshake has finished [ses=" + ses + ", msg=" + msg + ']');

      return proceedSessionWrite(ses, msg);
    }

    /** {@inheritDoc} */
    @Override
    public void onMessageReceived(GridNioSession ses, Object msg) throws IgniteCheckedException {
      HadoopProcessDescriptor desc = ses.meta(PROCESS_META);

      UUID rmtProcId = desc == null ? null : desc.processId();

      if (rmtProcId == null) {
        if (!(msg instanceof ProcessHandshakeMessage)) {
          log.warning(
              "Invalid handshake message received, will close connection [ses="
                  + ses
                  + ", msg="
                  + msg
                  + ']');

          ses.close();

          return;
        }

        ProcessHandshakeMessage nId = (ProcessHandshakeMessage) msg;

        if (log.isDebugEnabled())
          log.debug("Received handshake message [ses=" + ses + ", msg=" + msg + ']');

        ses.addMeta(PROCESS_META, nId.processDescriptor());

        if (!ses.accepted())
          // Send handshake reply.
          ses.send(locIdMsg);
        else {
          //
          rmtProcId = nId.processDescriptor().processId();

          if (log.isDebugEnabled()) log.debug("Finished handshake with remote client: " + ses);

          Object sync = locks.tryLock(rmtProcId);

          if (sync != null) {
            try {
              if (clients.get(rmtProcId) == null) {
                if (log.isDebugEnabled())
                  log.debug("Will reuse session for descriptor: " + rmtProcId);

                // Handshake finished flag is true.
                clients.put(rmtProcId, new HadoopTcpNioCommunicationClient(ses));
              } else {
                if (log.isDebugEnabled())
                  log.debug(
                      "Will not reuse client as another already exists [locProcDesc="
                          + locProcDesc
                          + ", desc="
                          + desc
                          + ']');
              }
            } finally {
              locks.unlock(rmtProcId, sync);
            }
          } else {
            if (log.isDebugEnabled())
              log.debug(
                  "Concurrent connection is being established, will not reuse client session ["
                      + "locProcDesc="
                      + locProcDesc
                      + ", desc="
                      + desc
                      + ']');
          }
        }

        if (log.isDebugEnabled())
          log.debug(
              "Handshake is finished for session [ses="
                  + ses
                  + ", locProcDesc="
                  + locProcDesc
                  + ']');

        HandshakeFinish to = ses.meta(HANDSHAKE_FINISH_META);

        if (to != null) to.finish();

        // Notify session opened (both parties).
        proceedSessionOpened(ses);
      } else {
        if (msgQueueLimit > 0) {
          GridNioMessageTracker tracker = ses.meta(TRACKER_META);

          if (tracker == null) {
            GridNioMessageTracker old =
                ses.addMeta(TRACKER_META, tracker = new GridNioMessageTracker(ses, msgQueueLimit));

            assert old == null;
          }

          tracker.onMessageReceived();
        }

        proceedMessageReceived(ses, msg);
      }
    }

    /** {@inheritDoc} */
    @Override
    public GridNioFuture<Boolean> onSessionClose(GridNioSession ses) throws IgniteCheckedException {
      return proceedSessionClose(ses);
    }

    /** {@inheritDoc} */
    @Override
    public void onSessionIdleTimeout(GridNioSession ses) throws IgniteCheckedException {
      proceedSessionIdleTimeout(ses);
    }

    /** {@inheritDoc} */
    @Override
    public void onSessionWriteTimeout(GridNioSession ses) throws IgniteCheckedException {
      proceedSessionWriteTimeout(ses);
    }
  }

  /** Process ID message. */
  @SuppressWarnings("PublicInnerClass")
  public static class ProcessHandshakeMessage implements HadoopMessage {
    /** */
    private static final long serialVersionUID = 0L;

    /** Node ID. */
    private HadoopProcessDescriptor procDesc;

    /** */
    public ProcessHandshakeMessage() {
      // No-op.
    }

    /** @param procDesc Process descriptor. */
    private ProcessHandshakeMessage(HadoopProcessDescriptor procDesc) {
      this.procDesc = procDesc;
    }

    /** @return Process ID. */
    public HadoopProcessDescriptor processDescriptor() {
      return procDesc;
    }

    /** {@inheritDoc} */
    @Override
    public void writeExternal(ObjectOutput out) throws IOException {
      out.writeObject(procDesc);
    }

    /** {@inheritDoc} */
    @Override
    public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
      procDesc = (HadoopProcessDescriptor) in.readObject();
    }

    /** {@inheritDoc} */
    @Override
    public String toString() {
      return S.toString(ProcessHandshakeMessage.class, this);
    }
  }
}
Ejemplo n.º 2
0
  /** {@inheritDoc} */
  @Override
  protected void start0() throws IgniteCheckedException {
    exchLog = cctx.logger(getClass().getName() + ".exchange");

    pendingExplicit = GridConcurrentFactory.newMap();
  }
Ejemplo n.º 3
0
/** Native transaction wrapper implementation. */
@SuppressWarnings({"unchecked", "UnusedDeclaration", "TryFinallyCanBeTryWithResources"})
public class PlatformTransactions extends PlatformAbstractTarget {
  /** */
  public static final int OP_CACHE_CONFIG_PARAMETERS = 1;

  /** */
  public static final int OP_METRICS = 2;

  /** */
  public static final int OP_START = 3;

  /** */
  public static final int OP_COMMIT = 4;

  /** */
  public static final int OP_ROLLBACK = 5;

  /** */
  public static final int OP_CLOSE = 6;

  /** */
  public static final int OP_STATE = 7;

  /** */
  public static final int OP_SET_ROLLBACK_ONLY = 8;

  /** */
  public static final int OP_COMMIT_ASYNC = 9;

  /** */
  public static final int OP_ROLLBACK_ASYNC = 10;

  /** */
  public static final int OP_RESET_METRICS = 11;

  /** */
  private final IgniteTransactions txs;

  /** Map with currently active transactions. */
  private final ConcurrentMap<Long, Transaction> txMap = GridConcurrentFactory.newMap();

  /** Transaction ID sequence. Must be static to ensure uniqueness across different caches. */
  private static final AtomicLong TX_ID_GEN = new AtomicLong();

  /**
   * Constructor.
   *
   * @param platformCtx Context.
   */
  public PlatformTransactions(PlatformContext platformCtx) {
    super(platformCtx);

    txs = platformCtx.kernalContext().grid().transactions();
  }

  /**
   * Register transaction.
   *
   * @param tx Transaction.
   * @return Transaction ID.
   */
  private long registerTx(Transaction tx) {
    long id = TX_ID_GEN.incrementAndGet();

    Transaction old = txMap.put(id, tx);

    assert old == null : "Duplicate TX ids: " + old;

    return id;
  }

  /**
   * Unregister transaction.
   *
   * @param id Transaction ID.
   */
  private void unregisterTx(long id) {
    Transaction tx = txMap.remove(id);

    assert tx != null : "Failed to unregister transaction: " + id;
  }

  /**
   * @param id Transaction ID.
   * @return Transaction state.
   */
  private int txClose(long id) {
    Transaction tx = tx(id);

    try {
      tx.close();

      return tx.state().ordinal();
    } finally {
      unregisterTx(id);
    }
  }

  /**
   * Get transaction by ID.
   *
   * @param id ID.
   * @return Transaction.
   */
  private Transaction tx(long id) {
    Transaction tx = txMap.get(id);

    assert tx != null : "Transaction not found for ID: " + id;

    return tx;
  }

  /** {@inheritDoc} */
  @Override
  protected long processInLongOutLong(int type, long val) throws IgniteCheckedException {
    switch (type) {
      case OP_COMMIT:
        tx(val).commit();

        return txClose(val);

      case OP_ROLLBACK:
        tx(val).rollback();

        return txClose(val);

      case OP_CLOSE:
        return txClose(val);

      case OP_SET_ROLLBACK_ONLY:
        return tx(val).setRollbackOnly() ? TRUE : FALSE;

      case OP_STATE:
        return tx(val).state().ordinal();

      case OP_RESET_METRICS:
        txs.resetMetrics();

        return TRUE;
    }

    return super.processInLongOutLong(type, val);
  }

  /** {@inheritDoc} */
  @Override
  protected long processInStreamOutLong(int type, BinaryRawReaderEx reader)
      throws IgniteCheckedException {
    long txId = reader.readLong();

    final Transaction asyncTx = (Transaction) tx(txId).withAsync();

    switch (type) {
      case OP_COMMIT_ASYNC:
        asyncTx.commit();

        break;

      case OP_ROLLBACK_ASYNC:
        asyncTx.rollback();

        break;

      default:
        return super.processInStreamOutLong(type, reader);
    }

    // Future result is the tx itself, we do not want to return it to the platform.
    IgniteFuture fut =
        asyncTx
            .future()
            .chain(
                new C1<IgniteFuture, Object>() {
                  private static final long serialVersionUID = 0L;

                  @Override
                  public Object apply(IgniteFuture fut) {
                    return null;
                  }
                });

    readAndListenFuture(reader, fut);

    return TRUE;
  }

  /** {@inheritDoc} */
  @Override
  protected void processInStreamOutStream(
      int type, BinaryRawReaderEx reader, BinaryRawWriterEx writer) throws IgniteCheckedException {
    switch (type) {
      case OP_START:
        {
          TransactionConcurrency txConcurrency =
              TransactionConcurrency.fromOrdinal(reader.readInt());

          assert txConcurrency != null;

          TransactionIsolation txIsolation = TransactionIsolation.fromOrdinal(reader.readInt());

          assert txIsolation != null;

          Transaction tx =
              txs.txStart(txConcurrency, txIsolation, reader.readLong(), reader.readInt());

          long id = registerTx(tx);

          writer.writeLong(id);

          return;
        }
    }

    super.processInStreamOutStream(type, reader, writer);
  }

  /** {@inheritDoc} */
  @Override
  protected void processOutStream(int type, BinaryRawWriterEx writer)
      throws IgniteCheckedException {
    switch (type) {
      case OP_CACHE_CONFIG_PARAMETERS:
        TransactionConfiguration txCfg =
            platformCtx.kernalContext().config().getTransactionConfiguration();

        writer.writeInt(txCfg.getDefaultTxConcurrency().ordinal());
        writer.writeInt(txCfg.getDefaultTxIsolation().ordinal());
        writer.writeLong(txCfg.getDefaultTxTimeout());

        break;

      case OP_METRICS:
        TransactionMetrics metrics = txs.metrics();

        writer.writeTimestamp(new Timestamp(metrics.commitTime()));
        writer.writeTimestamp(new Timestamp(metrics.rollbackTime()));
        writer.writeInt(metrics.txCommits());
        writer.writeInt(metrics.txRollbacks());

        break;

      default:
        super.processOutStream(type, writer);
    }
  }
}