From 382bead5482e8e379aea3bf7857cf8e28d6da850 Mon Sep 17 00:00:00 2001 From: Alexandre Thibault Date: Fri, 29 Sep 2017 11:32:30 +0200 Subject: [PATCH] rpc: fix client websocket timeout (#687) --- rpc/lib/server/handlers.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/rpc/lib/server/handlers.go b/rpc/lib/server/handlers.go index 0fafd2a84..9de6564dd 100644 --- a/rpc/lib/server/handlers.go +++ b/rpc/lib/server/handlers.go @@ -488,6 +488,10 @@ func (wsc *wsConnection) readRoutine() { wsc.baseConn.Close() }() + wsc.baseConn.SetPongHandler(func(m string) error { + return wsc.baseConn.SetReadDeadline(time.Now().Add(wsc.readWait)) + }) + for { select { case <-wsc.Quit: