acquire a multilock

prysmaticlabs · Mar 16, 2024 · b96a52b · b96a52b
1 parent 0ddbdfe
commit b96a52b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/beacon-chain/blockchain/process_attestation_helpers.go b/beacon-chain/blockchain/process_attestation_helpers.go
@@ -48,6 +48,18 @@ func (s *Service) getRecentPreState(ctx context.Context, c *ethpb.Checkpoint) st
 	if err != nil {
 		return nil
 	}
+	// Try if we have already set the checkpoint cache
+	epochKey := strconv.FormatUint(uint64(c.Epoch), 10 /* base 10 */)
+	lock := async.NewMultilock(string(c.Root) + epochKey)
+	lock.Lock()
+	defer lock.Unlock()
+	cachedState, err := s.checkpointStateCache.StateByCheckpoint(c)
+	if err != nil {
+		return nil
+	}
+	if cachedState != nil && !cachedState.IsNil() {
+		return cachedState
+	}
 	st, err = transition.ProcessSlotsUsingNextSlotCache(ctx, st, c.Root, slot)
 	if err != nil {
 		return nil