misc: add deadtime of connection(#master)

OhYee · Jan 21, 2020 · 12d1da1 · 12d1da1
1 parent abf372d
commit 12d1da1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/server/send.go b/server/send.go
@@ -2,6 +2,7 @@ package server
 
 import (
 	"github.com/xtaci/kcp-go"
+	"time"
 )
 
 // Send data to addr
@@ -10,8 +11,10 @@ func Send(addr string, data []byte) (repBytes []byte, repErr error, err error) {
 	if err != nil {
 		return
 	}
+	conn.SetDeadline(time.Now().Add(time.Second * time.Duration(connDeadTime)))
 
 	conn.Write(data)
+	Log("Write %v", data)
 	repBytes, repErr, err = messageHandle(conn)
 	return
 }
diff --git a/server/server.go b/server/server.go
@@ -4,11 +4,13 @@ import (
 	"github.com/OhYee/ait/message"
 	"github.com/xtaci/kcp-go"
 	"net"
+	"time"
 )
 
 // Singleton Pattern
 var (
-	closeThread = make([]chan bool, 0)
+	closeThread  = make([]chan bool, 0)
+	connDeadTime = 5000
 )
 
 // Start server, blocking the process after returned
@@ -40,6 +42,7 @@ func thread(threadID int, listener net.Listener) {
 				Err(err)
 				continue
 			}
+			conn.SetDeadline(time.Now().Add(time.Second * time.Duration(connDeadTime)))
 			if handle(threadID, conn) != nil {
 				Err(err)
 				continue