You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@ignite.apache.org by vo...@apache.org on 2015/03/05 07:54:20 UTC
[7/7] incubator-ignite git commit: Merge branch 'sprint-2' into
ignite-386-tmp
Merge branch 'sprint-2' into ignite-386-tmp
Project: http://git-wip-us.apache.org/repos/asf/incubator-ignite/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-ignite/commit/0bef3fc8
Tree: http://git-wip-us.apache.org/repos/asf/incubator-ignite/tree/0bef3fc8
Diff: http://git-wip-us.apache.org/repos/asf/incubator-ignite/diff/0bef3fc8
Branch: refs/heads/ignite-386-tmp
Commit: 0bef3fc8538fa53692b6d963f982ce7d49e08a74
Parents: 498dcfa 162760b
Author: vozerov-gridgain <vo...@gridgain.com>
Authored: Thu Mar 5 09:48:11 2015 +0300
Committer: vozerov-gridgain <vo...@gridgain.com>
Committed: Thu Mar 5 09:48:11 2015 +0300
----------------------------------------------------------------------
bin/ignitevisorcmd.sh | 2 +-
.../misc/lifecycle/LifecycleExample.java | 4 +-
.../apache/ignite/internal/IgniteKernal.java | 8 +--
.../shmem/IpcSharedMemoryServerEndpoint.java | 3 +-
.../apache/ignite/lifecycle/LifecycleBean.java | 36 +++++-----
.../ignite/lifecycle/LifecycleEventType.java | 20 +++---
.../communication/tcp/TcpCommunicationSpi.java | 2 +-
.../internal/GridLifecycleBeanSelfTest.java | 76 ++++++++++----------
.../dht/GridCacheDhtPreloadBigDataSelfTest.java | 2 +-
.../dht/GridCacheDhtPreloadUnloadSelfTest.java | 2 +-
...achePartitionedPreloadLifecycleSelfTest.java | 8 +--
...CacheReplicatedPreloadLifecycleSelfTest.java | 8 +--
.../clock/GridTimeSyncProcessorSelfTest.java | 2 +-
.../colocation/GridTestLifecycleBean.java | 2 +-
.../loadtests/dsi/GridDsiLifecycleBean.java | 8 +--
.../ignite/testsuites/IgniteCacheTestSuite.java | 6 +-
.../HadoopExternalCommunication.java | 2 +-
.../ignite/internal/GridFactorySelfTest.java | 10 +--
.../apache/ignite/visor/plugin/VisorPlugin.java | 2 +-
19 files changed, 100 insertions(+), 103 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/incubator-ignite/blob/0bef3fc8/modules/core/src/main/java/org/apache/ignite/internal/IgniteKernal.java
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/incubator-ignite/blob/0bef3fc8/modules/hadoop/src/main/java/org/apache/ignite/internal/processors/hadoop/taskexecutor/external/communication/HadoopExternalCommunication.java
----------------------------------------------------------------------
diff --cc modules/hadoop/src/main/java/org/apache/ignite/internal/processors/hadoop/taskexecutor/external/communication/HadoopExternalCommunication.java
index 4df8188,0000000..499f2fa
mode 100644,000000..100644
--- a/modules/hadoop/src/main/java/org/apache/ignite/internal/processors/hadoop/taskexecutor/external/communication/HadoopExternalCommunication.java
+++ b/modules/hadoop/src/main/java/org/apache/ignite/internal/processors/hadoop/taskexecutor/external/communication/HadoopExternalCommunication.java
@@@ -1,1431 -1,0 +1,1431 @@@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.ignite.internal.processors.hadoop.taskexecutor.external.communication;
+
+import org.apache.ignite.*;
+import org.apache.ignite.internal.processors.hadoop.message.*;
+import org.apache.ignite.internal.processors.hadoop.taskexecutor.external.*;
+import org.apache.ignite.internal.util.*;
+import org.apache.ignite.internal.util.ipc.*;
+import org.apache.ignite.internal.util.ipc.shmem.*;
+import org.apache.ignite.internal.util.nio.*;
+import org.apache.ignite.internal.util.typedef.*;
+import org.apache.ignite.internal.util.typedef.internal.*;
+import org.apache.ignite.internal.util.worker.*;
+import org.apache.ignite.marshaller.*;
+import org.apache.ignite.thread.*;
+import org.jdk8.backport.*;
+import org.jetbrains.annotations.*;
+
+import java.io.*;
+import java.net.*;
+import java.nio.*;
+import java.nio.channels.*;
+import java.util.*;
+import java.util.concurrent.*;
+
+/**
+ * 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)."; // TODO IGNITE-70 Add link to documentation
++ "(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", 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", 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).listenAsync(new CI1<GridNioFuture<?>>() {
+ @Override public void apply(GridNioFuture<?> fut) {
+ try {
+ // Make sure there were no errors.
+ fut.get();
+ }
+ catch (IgniteCheckedException | IOException 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);
+ }
+ }
+}