process accept error

FengJiaQi · Oct 18, 2015 · 85027ce · 85027ce
1 parent d125311
commit 85027ce
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/network/tcp_server.go b/network/tcp_server.go
@@ -4,6 +4,7 @@ import (
 	"github.com/name5566/leaf/log"
 	"net"
 	"sync"
+	"time"
 )
 
 type TCPServer struct {
@@ -62,11 +63,26 @@ func (server *TCPServer) run() {
 	server.wgLn.Add(1)
 	defer server.wgLn.Done()
 
+	var tempDelay time.Duration
 	for {
 		conn, err := server.ln.Accept()
 		if err != nil {
+			if ne, ok := err.(net.Error); ok && ne.Temporary() {
+				if tempDelay == 0 {
+					tempDelay = 5 * time.Millisecond
+				} else {
+					tempDelay *= 2
+				}
+				if max := 1 * time.Second; tempDelay > max {
+					tempDelay = max
+				}
+				log.Release("accept error: %v; retrying in %v", err, tempDelay)
+				time.Sleep(tempDelay)
+				continue
+			}
 			return
 		}
+		tempDelay = 0
 
 		server.mutexConns.Lock()
 		if len(server.conns) >= server.MaxConnNum {