Fix received connection not closing

Signed-off-by: Yilun <[email protected]>
nknorg · Sep 6, 2018 · 0be60e0 · 0be60e0
1 parent c96eb17
commit 0be60e0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/net/node/node.go b/net/node/node.go
@@ -145,6 +145,14 @@ func NewNode() *node {
 		state: INIT,
 	}
 	n.time = time.Now()
+	go func() {
+		time.Sleep(ConnectingTimeout)
+		if n.local != &n && n.GetState() != ESTABLISH {
+			log.Warn("Connecting timeout:", n.GetAddrStr())
+			n.SetState(INACTIVITY)
+			n.CloseConn()
+		}
+	}()
 	return &n
 }