refactor: locks

Layr-Labs · Dec 18, 2024 · de8f697 · de8f697
1 parent 67e46b0
commit de8f697
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/core/meterer/onchain_state.go b/core/meterer/onchain_state.go
@@ -153,7 +153,9 @@ func (pcs *OnchainPaymentState) GetReservedPaymentByAccount(ctx context.Context,
 	if err != nil {
 		return nil, err
 	}
+	pcs.ReservationsLock.Lock()
 	(pcs.ReservedPayments)[accountID] = res
+	pcs.ReservationsLock.Unlock()
 
 	return res, nil
 }
@@ -172,7 +174,10 @@ func (pcs *OnchainPaymentState) GetOnDemandPaymentByAccount(ctx context.Context,
 	if err != nil {
 		return nil, err
 	}
+
+	pcs.OnDemandLocks.Lock()
 	(pcs.OnDemandPayments)[accountID] = res
+	pcs.OnDemandLocks.Unlock()
 	return res, nil
 }