feat: ws断连后自动重试

2024-10-20 15:47:30 +08:00 · 2024-10-20 15:47:30 +08:00 · 33ef75c54f
commit 33ef75c54f
parent 5a56f6fdc7
1 changed files with 6 additions and 0 deletions
--- a/ws_client/client.go
+++ b/ws_client/client.go
@ -31,6 +31,12 @@ func NewWebSocketClient(scheme, host, path string) (*WebSocketClient, error) {
 		conn: conn,
 		done: make(chan struct{}),
 	}
+	defer func() {
+		if r := recover(); r != nil {
+			log.Print("Recovered in NewWebSocketClient:", r)
+			go client.receiveMessages()
+		}
+	}()

 	go client.receiveMessages()