fix: add read locks in ledger query functions

blinklabs-io · Sep 17, 2024 · 69bfa1e · 69bfa1e
1 parent 41c8689
commit 69bfa1e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/state/state.go b/state/state.go
@@ -339,6 +339,8 @@ func (ls *LedgerState) removeBlock(txn *database.Txn, block models.Block) error
 }
 
 func (ls *LedgerState) GetBlock(point ocommon.Point) (*models.Block, error) {
+	ls.RLock()
+	defer ls.RUnlock()
 	ret, err := models.BlockByPoint(ls.db, point)
 	if err != nil {
 		return nil, err
@@ -368,6 +370,8 @@ func (ls *LedgerState) RecentChainPoints(count int) ([]ocommon.Point, error) {
 
 // GetIntersectPoint returns the intersect between the specified points and the current chain
 func (ls *LedgerState) GetIntersectPoint(points []ocommon.Point) (*ocommon.Point, error) {
+	ls.RLock()
+	defer ls.RUnlock()
 	var ret ocommon.Point
 	for _, point := range points {
 		// Ignore points with a slot earlier than an existing match