Bläddra i källkod

avoid sending msg onb closed channel

abhishek9686 1 år sedan
förälder
incheckning
0de762240b
1 ändrade filer med 4 tillägg och 1 borttagningar
  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
 				}