Websocket: Protect Setup() from races

gbjk · Feb 16, 2024 · ebaa55d · ebaa55d
1 parent 78a7e83
commit ebaa55d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/exchanges/stream/websocket.go b/exchanges/stream/websocket.go
@@ -103,6 +103,9 @@ func (w *Websocket) Setup(s *WebsocketSetup) error {
 		return errWebsocketSetupIsNil
 	}
 
+	w.m.Lock()
+	defer w.m.Unlock()
+
 	if w.IsInitialised() {
 		return fmt.Errorf("%s %w", w.exchangeName, errWebsocketAlreadyInitialised)
 	}