From 9f8ee49b5c951824de79fa117fc5ee96e41a7cd6 Mon Sep 17 00:00:00 2001 From: Roger Ferdinan Date: Mon, 29 Sep 2025 18:48:38 +0700 Subject: [PATCH] fix: adding SafeWebSocketClient parameter --- v1/client/client.go | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/v1/client/client.go b/v1/client/client.go index b486e66..09ac0e3 100644 --- a/v1/client/client.go +++ b/v1/client/client.go @@ -71,7 +71,7 @@ type SafeWebsocketClientBuilder struct { rawQuery *string useTLS *bool channelSize *int64 - authenticateFn func() error + authenticateFn func(*SafeWebsocketClient) error } func NewSafeWebsocketClientBuilder() *SafeWebsocketClientBuilder { @@ -93,7 +93,7 @@ func (b *SafeWebsocketClientBuilder) UseTLS(useTLS bool) *SafeWebsocketClientBui return b } -func (b *SafeWebsocketClientBuilder) AuthenticateFn(authenticateFn func() error) *SafeWebsocketClientBuilder { +func (b *SafeWebsocketClientBuilder) AuthenticateFn(authenticateFn func(*SafeWebsocketClient) error) *SafeWebsocketClientBuilder { b.authenticateFn = authenticateFn return b } @@ -170,7 +170,7 @@ type SafeWebsocketClient struct { reconnectChans []chan struct{} isConnected bool doneMap *SafeMap[string, chan struct{}] - authenticateFn func() error + authenticateFn func(*SafeWebsocketClient) error } func (wsClient *SafeWebsocketClient) connect() error { @@ -218,13 +218,12 @@ func (wsClient *SafeWebsocketClient) connect() error { wsClient.isConnected = true if wsClient.authenticateFn != nil { - if err := wsClient.authenticateFn(); err != nil { + if err := wsClient.authenticateFn(wsClient); err != nil { return err } } wsClient.mu.WriteHandler(func() error { - pingCtx, pingCancel := context.WithCancel(context.Background()) go wsClient.startPingTicker(pingCtx) wsClient.cancelFuncs = append(wsClient.cancelFuncs, pingCancel)