fix: potential fix

Bisonai · Jul 23, 2024 · 1cf958d · 1cf958d
1 parent 472f1af
commit 1cf958d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/node/pkg/dal/api/controller.go b/node/pkg/dal/api/controller.go
@@ -94,6 +94,8 @@ func (c *Controller) broadcastDataForSymbol(symbol string) {
 func (c *Controller) castSubmissionData(data *dalcommon.OutgoingSubmissionData, symbol *string) {
 	for conn := range c.clients {
 		if _, ok := c.clients[conn][*symbol]; ok {
+			c.mu.Lock()
+			defer c.mu.Unlock()
 			if err := conn.WriteJSON(*data); err != nil {
 				log.Error().Err(err).Msg("failed to write message")
 				delete(c.clients, conn)

diff --git a/node/pkg/dal/api/types.go b/node/pkg/dal/api/types.go
@@ -1,6 +1,8 @@
 package api
 
 import (
+	"sync"
+
 	"bisonai.com/orakl/node/pkg/common/types"
 	"bisonai.com/orakl/node/pkg/dal/collector"
 	dalcommon "bisonai.com/orakl/node/pkg/dal/common"
@@ -20,4 +22,6 @@ type Controller struct {
 	register   chan *websocket.Conn
 	unregister chan *websocket.Conn
 	broadcast  map[string]chan dalcommon.OutgoingSubmissionData
+
+	mu sync.RWMutex
 }