[Bugfix] fix unclosed gorountine for websocket ping (#126)

300902b7 · Eric_Lee · GitHub · cd3dca8d · 300902b7
Unverified Commit 300902b7 authored Oct 30, 2019 by Eric_Lee Committed by GitHub Oct 30, 2019
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

websshws.go pkg/httpd/websshws.go +4 -0

No files found.
--- a/pkg/httpd/websshws.go
+++ b/pkg/httpd/websshws.go
@@ -58,6 +58,10 @@ func OnNamespaceConnected(c *neffos.NSConn, msg neffos.Message) error {
 	go func() {
 		for {
 			<-time.After(30 * time.Second)
+			if c.Conn.IsClosed() {
+				logger.Infof("User %s from %s websocket connect closed", user.Username, remoteIP)
+				return
+			}
 			c.Emit("ping", []byte(""))
 		}
 	}()