await closing sockets on shutdown

Author: kimchy

modules/elasticsearch/src/main/java/org/elasticsearch/transport/netty/NettyTransport.java

@@ -324,7 +324,7 @@ ThreadPool threadPool() {
         for (Iterator<NodeChannels> it = connectedNodes.values().iterator(); it.hasNext();) {
             NodeChannels nodeChannels = it.next();
             it.remove();
-            nodeChannels.close();
+            nodeChannels.closeAndWait();
         }
 
         if (clientBootstrap != null) {
@@ -615,5 +615,23 @@ private void closeChannels(Channel[] channels) {
                 }
             }
         }
+
+        public void closeAndWait() {
+            closeChannelsAndWait(low);
+            closeChannelsAndWait(med);
+            closeChannelsAndWait(high);
+        }
+
+        private void closeChannelsAndWait(Channel[] channels) {
+            for (Channel channel : channels) {
+                try {
+                    if (channel != null && channel.isOpen()) {
+                        channel.close().awaitUninterruptibly();
+                    }
+                } catch (Exception e) {
+                    //ignore
+                }
+            }
+        }
     }
 }