Prechádzať zdrojové kódy

avoid sending msg onb closed channel

abhishek9686 1 rok pred
rodič
commit
0de762240b
1 zmenil súbory, kde vykonal 4 pridanie a 1 odobranie
  1. 4 1
      auth/host_session.go

+ 4 - 1
auth/host_session.go

@@ -133,7 +133,10 @@ func SessionHandler(conn *websocket.Conn) {
 	go func() {
 		for {
 			msgType, _, err := conn.ReadMessage()
-			if err != nil || msgType == websocket.CloseMessage {
+			if err != nil {
+				return
+			}
+			if msgType == websocket.CloseMessage {
 				if timeout != nil {
 					timeout <- true
 				}