浏览代码

avoid sending msg onb closed channel

abhishek9686 1 年之前
父节点
当前提交
0de762240b
共有 1 个文件被更改,包括 4 次插入1 次删除
  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
 				}