rogerferdinan/safe-web-socket
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: adding lock for cancel function slices

Browse Source
This commit is contained in:
Roger Ferdinan
2025-09-29 22:53:29 +07:00
parent 2225391fc3
commit c550701dfa
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
v1/client/client.go
View File

@@ -235,7 +235,10 @@ func (wsClient *SafeWebsocketClient) connect() error {
pingCtx, pingCancel := context.WithCancel(context.Background()) pingCtx, pingCancel := context.WithCancel(context.Background())
go wsClient.startPingTicker(pingCtx) go wsClient.startPingTicker(pingCtx)
wsClient.mu.WriteHandler(func() error {
wsClient.cancelFuncs = append(wsClient.cancelFuncs, pingCancel) wsClient.cancelFuncs = append(wsClient.cancelFuncs, pingCancel)
return nil
})
if wsClient.conn != nil { if wsClient.conn != nil {
wsClient.conn.Close() wsClient.conn.Close()
@@ -251,7 +254,10 @@ func (wsClient *SafeWebsocketClient) connect() error {
go func() { go func() {
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
wsClient.mu.WriteHandler(func() error {
wsClient.cancelFuncs = append(wsClient.cancelFuncs, cancel) wsClient.cancelFuncs = append(wsClient.cancelFuncs, cancel)
return nil
})
for { for {
select { select {
case <-ctx.Done(): case <-ctx.Done():
@@ -285,7 +291,10 @@ func (wsClient *SafeWebsocketClient) connect() error {
go func() { go func() {
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
wsClient.mu.WriteHandler(func() error {
wsClient.cancelFuncs = append(wsClient.cancelFuncs, cancel) wsClient.cancelFuncs = append(wsClient.cancelFuncs, cancel)
return nil
})
for { for {
select { select {
case <-ctx.Done(): case <-ctx.Done():