fix DATA RACE ws client sender method

bmstar2012 · Apr 8, 2022 · 7d97eb5 · 7d97eb5
1 parent 227fe28
commit 7d97eb5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/ws/client.go b/api/ws/client.go
@@ -286,11 +286,13 @@ func (c *ClientWs) sender(p bool) error {
 				return err
 			}
 		case <-ticker.C:
+			c.mu[p].Lock()
 			if c.conn[p] != nil && (c.lastTransmit[p] == nil || (c.lastTransmit[p] != nil && time.Since(*c.lastTransmit[p]) > PingPeriod)) {
 				go func() {
 					c.sendChan[p] <- []byte("ping")
 				}()
 			}
+			c.mu[p].Unlock()
 		case <-c.ctx.Done():
 			return c.handleCancel("sender")
 		}