fix: reduce the granularity of the lock.

FGadvancer · Dec 13, 2023 · 4be71e7 · 4be71e7
1 parent d958056
commit 4be71e7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/push/push_to_client.go b/internal/push/push_to_client.go
@@ -299,6 +299,9 @@ func (p *Pusher) GetConnsAndOnlinePush(ctx context.Context, msg *sdkws.MsgData,
 		maxWorkers = 3
 	}
 
+	wg.SetLimit(maxWorkers)
+
+	// Online push message
 	for _, conn := range conns {
 		conn := conn // loop var safe
 		wg.Go(func() error {
@@ -319,6 +322,8 @@ func (p *Pusher) GetConnsAndOnlinePush(ctx context.Context, msg *sdkws.MsgData,
 		})
 	}
 
+	_ = wg.Wait()
+
 	// always return nil
 	return wsResults, nil
 }