From 9c6db3055008a621a3a00a6e4e5f5ba5e8dccdab Mon Sep 17 00:00:00 2001 From: Roger Ferdinan Date: Mon, 13 Oct 2025 08:30:42 +0700 Subject: [PATCH] feat: add data retrieval close function --- v1/client/client.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/v1/client/client.go b/v1/client/client.go index fcfc200..80ab0d0 100644 --- a/v1/client/client.go +++ b/v1/client/client.go @@ -338,7 +338,7 @@ func (wsClient *SafeWebsocketClient) startPingTicker(ctx context.Context) { for { select { case <-ctx.Done(): - log.Println("ping ticker stopped") + log.Println("ping ticker canceled by context") return case <-ticker.C: wsClient.writeChan <- Message{ @@ -425,6 +425,10 @@ func (wsClient *SafeWebsocketClient) DataChannel() <-chan []byte { return wsClient.dataChannel } +func (wsClient *SafeWebsocketClient) CloseDataChannel() { + close(wsClient.dataChannel) +} + func (wsClient *SafeWebsocketClient) Write(data []byte) error { wsClient.writeChan <- Message{ MessageType: MessageTypeText,