fix: nil mark_as_read

wdvxdr1123 · Jul 28, 2023 · a5f46e1 · a5f46e1
1 parent a433140
commit a5f46e1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/driver/wsclient.go b/driver/wsclient.go
@@ -123,11 +123,11 @@ func (ws *WSClient) Listen(handler func([]byte, zero.APICaller)) {
 				}
 				close(c) // channel only use once
 			}
-			return
+			continue
 		}
 		if rsp.Get("meta_event_type").Str != "heartbeat" { // 忽略心跳事件
 			log.Debug("[ws] 接收到事件: ", helper.BytesToString(payload))
-			return
+			continue
 		}
 		handler(payload, ws)
 	}

diff --git a/driver/wsserver.go b/driver/wsserver.go
@@ -194,11 +194,11 @@ func (wssc *WSSCaller) listen(handler func([]byte, zero.APICaller)) {
 				}
 				close(c) // channel only use once
 			}
-			return
+			continue
 		}
 		if rsp.Get("meta_event_type").Str != "heartbeat" { // 忽略心跳事件
 			log.Debug("[wss] 接收到事件: ", helper.BytesToString(payload))
-			return
+			continue
 		}
 		handler(payload, wssc)
 	}