Add locks

flashbots · Oct 2, 2023 · ee72ced · ee72ced
1 parent f74e792
commit ee72ced
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/suave/backends/local_store_backend.go b/suave/backends/local_store_backend.go
@@ -79,6 +79,9 @@ func (l *LocalConfidentialStore) Retrieve(bid suave.Bid, caller common.Address,
 }
 
 func (l *LocalConfidentialStore) FetchBidById(bidId suave.BidId) (suave.Bid, error) {
+	l.lock.Lock()
+	defer l.lock.Unlock()
+
 	bid, found := l.bids[bidId]
 	if !found {
 		return suave.Bid{}, errors.New("bid not found")
@@ -88,6 +91,9 @@ func (l *LocalConfidentialStore) FetchBidById(bidId suave.BidId) (suave.Bid, err
 }
 
 func (l *LocalConfidentialStore) FetchBidsByProtocolAndBlock(blockNumber uint64, namespace string) []suave.Bid {
+	l.lock.Lock()
+	defer l.lock.Unlock()
+
 	indexKey := fmt.Sprintf("protocol-%s-bn-%d", namespace, blockNumber)
 	bidIDs, ok := l.index[indexKey]
 	if !ok {