fix

Author: leixm

ratis-common/src/main/java/org/apache/ratis/util/IOUtils.java

@@ -44,6 +44,16 @@
  * IO related utility methods.
  */
 public interface IOUtils {
+  // Network exceptions for reconnection
+  Class<?>[] NETWORK_EXCEPTIONS = {
+      SocketException.class,
+      SocketTimeoutException.class,
+      ClosedChannelException.class,
+      EOFException.class,
+      AlreadyClosedException.class,
+      TimeoutIOException.class
+  };
+
   static InterruptedIOException toInterruptedIOException(
       String message, InterruptedException e) {
     final InterruptedIOException iioe = new InterruptedIOException(message);
@@ -91,9 +101,12 @@ static <T> T getFromFuture(CompletableFuture<T> future, Supplier<Object> name, T
   }
 
   static boolean shouldReconnect(Throwable e) {
-    return ReflectionUtils.isInstance(e,
-        SocketException.class, SocketTimeoutException.class, ClosedChannelException.class, EOFException.class,
-        AlreadyClosedException.class, TimeoutIOException.class);
+    if (e == null) return false;
+    if (ReflectionUtils.isInstance(e, NETWORK_EXCEPTIONS)) {
+      return true;
+    }
+    return e.getCause() != null &&
+        ReflectionUtils.isInstance(e.getCause(), NETWORK_EXCEPTIONS);
   }
 
   static void readFully(InputStream in, int buffSize) throws IOException {

ratis-netty/src/main/java/org/apache/ratis/netty/NettyRpcProxy.java

@@ -128,7 +128,7 @@ protected void channelRead0(ChannelHandlerContext ctx,
 
         @Override
         public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
-          failOutstandingRequests(cause);
+          failOutstandingRequests(new IOException("Caught an exception for the connection to " + peer, cause));
           client.close();
         }