加入KeepAlive设置

ying32 · Jan 16, 2017 · 553217e · 553217e
1 parent 6300abd
commit 553217e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -6,6 +6,7 @@ import (
 	"log"
 	"net"
 	"net/http"
+	"time"
 )
 
 type TRPClient struct {
@@ -26,6 +27,8 @@ func (c *TRPClient) Start() error {
 	if err != nil {
 		return err
 	}
+	conn.(*net.TCPConn).SetKeepAlive(true)
+	conn.(*net.TCPConn).SetKeepAlivePeriod(time.Duration(2 * time.Second))
 	c.conn = conn
 	return c.process()
 }

diff --git a/server.go b/server.go
@@ -37,6 +37,10 @@ func (s *TRPServer) Start() error {
 }
 
 func (s *TRPServer) Close() error {
+	if s.conn != nil {
+		s.conn.Close()
+		s.conn = nil
+	}
 	if s.listener != nil {
 		err := s.listener.Close()
 		s.listener = nil
@@ -109,7 +113,8 @@ func (s *TRPServer) cliProcess(conn *net.TCPConn) error {
 		s.conn = nil
 	}
 	log.Println("连接新的客户端：", conn.RemoteAddr())
-
+	conn.SetKeepAlive(true)
+	conn.SetKeepAlivePeriod(time.Duration(2 * time.Second))
 	s.conn = conn
 	return nil
 }