From c470d316059234bb080f08d52bbf4183de03e8a4 Mon Sep 17 00:00:00 2001
From: Redkale <22250530@qq.com>
Date: Sat, 28 Apr 2018 20:03:01 +0800
Subject: [PATCH]

---
 src/org/redkale/net/ProtocolServer.java | 31 +++++++++++++++++++++++++
 src/org/redkale/net/Transport.java      | 29 +++++------------------
 2 files changed, 37 insertions(+), 23 deletions(-)

diff --git a/src/org/redkale/net/ProtocolServer.java b/src/org/redkale/net/ProtocolServer.java
index 62bf12a88..3284ec81f 100644
--- a/src/org/redkale/net/ProtocolServer.java
+++ b/src/org/redkale/net/ProtocolServer.java
@@ -23,6 +23,24 @@ import java.util.concurrent.atomic.AtomicLong;
  */
 public abstract class ProtocolServer {
 
+    protected static final boolean supportTcpNoDelay;
+
+    protected static final boolean supportTcpKeepAlive;
+
+    static {
+        boolean tcpNoDelay = false;
+        boolean keepAlive = false;
+        try {
+            AsynchronousSocketChannel channel = AsynchronousSocketChannel.open();
+            tcpNoDelay = channel.supportedOptions().contains(StandardSocketOptions.TCP_NODELAY);
+            keepAlive = channel.supportedOptions().contains(StandardSocketOptions.SO_KEEPALIVE);
+            channel.close();
+        } catch (Exception e) {
+        }
+        supportTcpNoDelay = tcpNoDelay;
+        supportTcpKeepAlive = keepAlive;
+    }
+
     //创建数
     protected final AtomicLong createCounter = new AtomicLong();
 
@@ -72,6 +90,14 @@ public abstract class ProtocolServer {
         throw new RuntimeException("ProtocolServer not support protocol " + protocol);
     }
 
+    public static boolean supportTcpNoDelay() {
+        return supportTcpNoDelay;
+    }
+
+    public static boolean supportTcpKeepAlive() {
+        return supportTcpKeepAlive;
+    }
+
     private static final class ProtocolUDPServer extends ProtocolServer {
 
         private boolean running;
@@ -214,6 +240,11 @@ public abstract class ProtocolServer {
                     }
                     createCounter.incrementAndGet();
                     livingCounter.incrementAndGet();
+                    try {
+                        if (supportTcpNoDelay()) channel.setOption(StandardSocketOptions.TCP_NODELAY, true);
+                        if (supportTcpKeepAlive()) channel.setOption(StandardSocketOptions.SO_KEEPALIVE, true);
+                    } catch (IOException e) {
+                    }
                     AsyncConnection conn = AsyncConnection.create(channel, null, context);
                     conn.livingCounter = livingCounter;
                     conn.closedCounter = closedCounter;
diff --git a/src/org/redkale/net/Transport.java b/src/org/redkale/net/Transport.java
index c3551ae93..6d6d19a9d 100644
--- a/src/org/redkale/net/Transport.java
+++ b/src/org/redkale/net/Transport.java
@@ -18,6 +18,7 @@ import java.util.logging.Level;
 import javax.net.ssl.SSLContext;
 import org.redkale.convert.*;
 import org.redkale.convert.json.JsonConvert;
+import static org.redkale.net.ProtocolServer.*;
 import org.redkale.util.*;
 
 /**
@@ -32,24 +33,6 @@ public final class Transport {
 
     public static final String DEFAULT_PROTOCOL = "TCP";
 
-    protected static final boolean supportTcpNoDelay;
-
-    protected static final boolean supportKeepAlive;
-
-    static {
-        boolean tcpNoDelay = false;
-        boolean keepAlive = false;
-        try {
-            AsynchronousSocketChannel channel = AsynchronousSocketChannel.open();
-            tcpNoDelay = channel.supportedOptions().contains(StandardSocketOptions.TCP_NODELAY);
-            keepAlive = channel.supportedOptions().contains(StandardSocketOptions.SO_KEEPALIVE);
-            channel.close();
-        } catch (Exception e) {
-        }
-        supportTcpNoDelay = tcpNoDelay;
-        supportKeepAlive = keepAlive;
-    }
-
     protected final AtomicInteger seq = new AtomicInteger(-1);
 
     protected final TransportFactory factory;
@@ -238,7 +221,7 @@ public final class Transport {
             if (!rand) { //指定地址
                 TransportNode node = findTransportNode(addr);
                 if (node == null) {
-                    return AsyncConnection.createTCP(group, sslContext, addr, supportTcpNoDelay, factory.readTimeoutSeconds, factory.writeTimeoutSeconds);
+                    return AsyncConnection.createTCP(group, sslContext, addr, supportTcpNoDelay(), factory.readTimeoutSeconds, factory.writeTimeoutSeconds);
                 }
                 final BlockingQueue<AsyncConnection> queue = node.conns;
                 if (!queue.isEmpty()) {
@@ -247,7 +230,7 @@ public final class Transport {
                         if (conn.isOpen()) return CompletableFuture.completedFuture(conn);
                     }
                 }
-                return AsyncConnection.createTCP(group, sslContext, addr, supportTcpNoDelay, factory.readTimeoutSeconds, factory.writeTimeoutSeconds);
+                return AsyncConnection.createTCP(group, sslContext, addr, supportTcpNoDelay(), factory.readTimeoutSeconds, factory.writeTimeoutSeconds);
             }
 
             //---------------------随机取地址------------------------
@@ -269,8 +252,8 @@ public final class Transport {
                 }
                 CompletableFuture future = new CompletableFuture();
                 final AsynchronousSocketChannel channel = AsynchronousSocketChannel.open(group);
-                if (supportTcpNoDelay) channel.setOption(StandardSocketOptions.TCP_NODELAY, true);
-                if (supportKeepAlive) channel.setOption(StandardSocketOptions.SO_KEEPALIVE, true);
+                if (supportTcpNoDelay()) channel.setOption(StandardSocketOptions.TCP_NODELAY, true);
+                if (supportTcpKeepAlive()) channel.setOption(StandardSocketOptions.SO_KEEPALIVE, true);
                 channel.connect(one.address, one, new CompletionHandler<Void, TransportNode>() {
                     @Override
                     public void completed(Void result, TransportNode attachment) {
@@ -320,7 +303,7 @@ public final class Transport {
             if (node == exclude) continue;
             if (future.isDone()) return future;
             final AsynchronousSocketChannel channel = AsynchronousSocketChannel.open(group);
-            if (supportTcpNoDelay) channel.setOption(StandardSocketOptions.TCP_NODELAY, true);
+            if (supportTcpNoDelay()) channel.setOption(StandardSocketOptions.TCP_NODELAY, true);
             channel.connect(node.address, node, new CompletionHandler<Void, TransportNode>() {
                 @Override
                 public void completed(Void result, TransportNode attachment) {