Set write deadline in client/agent connections

rancher · Sep 19, 2018 · 0f6027a · 0f6027a
1 parent a0f6626
commit 0f6027a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/remotedialer/wsconn.go b/pkg/remotedialer/wsconn.go
@@ -24,6 +24,7 @@ func newWSConn(conn *websocket.Conn) *wsConn {
 func (w *wsConn) WriteMessage(messageType int, data []byte) error {
 	w.Lock()
 	defer w.Unlock()
+	w.conn.SetWriteDeadline(time.Now().Add(PingWaitDuration))
 	return w.conn.WriteMessage(messageType, data)
 }