From 13b9a77b510e1351267fcaa233a98a7c6a3ed9bd Mon Sep 17 00:00:00 2001
From: lifeiyi <736926938@qq.com>
Date: Wed, 30 Dec 2020 14:26:09 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96tcp=E8=BF=9E=E6=8E=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 network/tcp_client.go | 5 ++++-
 network/tcp_server.go | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/network/tcp_client.go b/network/tcp_client.go
index e800c24..7a2e977 100644
--- a/network/tcp_client.go
+++ b/network/tcp_client.go
@@ -72,7 +72,10 @@ func (client *TCPClient) init() {
 func (client *TCPClient) dial() net.Conn {
 	for {
 		conn, err := net.Dial("tcp", client.Addr)
-		if err == nil || client.closeFlag {
+		if client.closeFlag {
+			return conn
+		} else if err == nil && conn != nil {
+			conn.(*net.TCPConn).SetNoDelay(true)
 			return conn
 		}
 
diff --git a/network/tcp_server.go b/network/tcp_server.go
index 14bf138..9040e7f 100644
--- a/network/tcp_server.go
+++ b/network/tcp_server.go
@@ -95,6 +95,7 @@ func (server *TCPServer) run() {
 			}
 			return
 		}
+		conn.(*net.TCPConn).SetNoDelay(true)
 		tempDelay = 0
 
 		server.mutexConns.Lock()