diff --git a/src/ws/init.go b/src/ws/init.go index e6993ff..cc5e526 100644 --- a/src/ws/init.go +++ b/src/ws/init.go @@ -44,11 +44,14 @@ func welcomeAgents(ws *websocket.Conn, welcome wsWelcome, t string) { func gcConnOnError(ws *websocket.Conn) (err error) { listeners.Range(func(index, value interface{}) bool { - if value.(*connectionInfo).ConnectionPS == ws || value.(*connectionInfo).ConnectionRR == ws { - value.(*connectionInfo).ConnectionPS.Close() - value.(*connectionInfo).ConnectionRR.Close() - listeners.Delete(index) + var ci = value.(*connectionInfo) + if ci.ConnectionPS == ws { + ci.ConnectionPS.Close() } + if ci.ConnectionRR == ws { + ci.ConnectionRR.Close() + } + listeners.Delete(index) return true }) return err