Merge pull request docker-java#982 from pjdarton/add_netty_read_timeout

KostyaSha · web-flow · commit 528cc69b7187 · 2018-03-12T16:02:06.000+03:00
Add withReadTimeout(Integer) method to NettyDockerCmdExecFactory
diff --git a/src/main/java/com/github/dockerjava/netty/NettyDockerCmdExecFactory.java b/src/main/java/com/github/dockerjava/netty/NettyDockerCmdExecFactory.java
@@ -6,7 +6,9 @@
 import java.net.InetAddress;
 import java.net.InetSocketAddress;
 import java.net.SocketAddress;
+import java.net.SocketTimeoutException;
 import java.security.Security;
+import java.util.concurrent.TimeUnit;
 
 import javax.net.ssl.SSLEngine;
 import javax.net.ssl.SSLParameters;
@@ -23,6 +25,7 @@
 import io.netty.channel.Channel;
 import io.netty.channel.ChannelConfig;
 import io.netty.channel.ChannelFactory;
+import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelInitializer;
 import io.netty.channel.EventLoopGroup;
 import io.netty.channel.epoll.EpollDomainSocketChannel;
@@ -38,6 +41,9 @@
 import io.netty.handler.codec.http.HttpClientCodec;
 import io.netty.handler.logging.LoggingHandler;
 import io.netty.handler.ssl.SslHandler;
+import io.netty.handler.timeout.IdleState;
+import io.netty.handler.timeout.IdleStateEvent;
+import io.netty.handler.timeout.IdleStateHandler;
 import io.netty.util.concurrent.DefaultThreadFactory;
 import org.bouncycastle.jce.provider.BouncyCastleProvider;
 
@@ -84,6 +90,8 @@ public DuplexChannel getChannel() {
 
     private Integer connectTimeout = null;
 
+    private Integer readTimeout = null;
+
     @Override
     public void init(DockerClientConfig dockerClientConfig) {
         super.init(dockerClientConfig);
@@ -273,16 +281,54 @@ public NettyDockerCmdExecFactory withConnectTimeout(Integer connectTimeout) {
         return this;
     }
 
+    /**
+     * Configure read timeout in milliseconds
+     */
+    public NettyDockerCmdExecFactory withReadTimeout(Integer readTimeout) {
+        this.readTimeout = readTimeout;
+        return this;
+    }
+
     private <T extends Channel> T configure(T channel) {
         ChannelConfig channelConfig = channel.config();
 
         if (connectTimeout != null) {
             channelConfig.setConnectTimeoutMillis(connectTimeout);
         }
+        if (readTimeout != null) {
+            channel.pipeline().addLast("readTimeoutHandler", new ReadTimeoutHandler());
+        }
 
         return channel;
     }
 
+    private final class ReadTimeoutHandler extends IdleStateHandler {
+        private boolean alreadyTimedOut;
+
+        ReadTimeoutHandler() {
+            super(readTimeout, 0, 0, TimeUnit.MILLISECONDS);
+        }
+
+        /**
+         * Called when a read timeout was detected.
+         */
+        @Override
+        protected synchronized void channelIdle(ChannelHandlerContext ctx, IdleStateEvent evt) throws Exception {
+            assert evt.state() == IdleState.READER_IDLE;
+            final Channel channel = ctx.channel();
+            if (channel == null || !channel.isActive() || alreadyTimedOut) {
+                return;
+            }
+            final Object dockerAPIEndpoint = dockerClientConfig.getDockerHost();
+            final String msg = "Read timed out: No data received within " + readTimeout
+                    + "ms.  Perhaps the docker API (" + dockerAPIEndpoint
+                    + ") is not responding normally, or perhaps you need to increase the readTimeout value.";
+            final Exception ex = new SocketTimeoutException(msg);
+            ctx.fireExceptionCaught(ex);
+            alreadyTimedOut = true;
+        }
+    }
+
     protected WebTarget getBaseResource() {
         checkNotNull(baseResource, "Factory not initialized, baseResource not set. You probably forgot to call init()!");
         return baseResource;
