From 0d85c7dd7dfbb7c6290b44036d9a150e13a4302a Mon Sep 17 00:00:00 2001 From: "arraykeys@gmail.com" Date: Tue, 13 Mar 2018 17:31:51 +0800 Subject: [PATCH] bridge add timeout Signed-off-by: arraykeys@gmail.com --- services/mux_bridge.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/services/mux_bridge.go b/services/mux_bridge.go index 6c33246..605a187 100644 --- a/services/mux_bridge.go +++ b/services/mux_bridge.go @@ -180,7 +180,9 @@ func (s *MuxBridge) callback(inConn net.Conn, serverID, key string) { index := keys[i] log.Printf("select client : %s-%s", key, index) session, _ := group.Get(index) + session.(*smux.Session).SetDeadline(time.Now().Add(time.Millisecond * time.Duration(*s.cfg.Timeout))) stream, err := session.(*smux.Session).OpenStream() + session.(*smux.Session).SetDeadline(time.Time{}) if err != nil { log.Printf("%s client session open stream %s fail, err: %s, retrying...", key, serverID, err) time.Sleep(time.Second * 3)