feat: make getBeaconProposersNextEpoch async to await for shuffling c…

…alculation
ChainSafe · Oct 12, 2024 · dc7038c · dc7038c
1 parent d37bdb0
commit dc7038c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/beacon-node/src/api/impl/validator/index.ts b/packages/beacon-node/src/api/impl/validator/index.ts
@@ -959,7 +959,7 @@ export function getValidatorApi(
         case stateEpoch + 1:
           // Requesting duties for next epoch is allowed since they can be predicted with high probabilities.
           // @see `epochCtx.getBeaconProposersNextEpoch` JSDocs for rationale.
-          indexes = state.epochCtx.getBeaconProposersNextEpoch();
+          indexes = await state.epochCtx.getBeaconProposersNextEpoch();
           break;
 
         case stateEpoch - 1: {

diff --git a/packages/state-transition/src/cache/epochCache.ts b/packages/state-transition/src/cache/epochCache.ts
@@ -918,12 +918,12 @@ export class EpochCache {
    * balances are sampled to adjust the probability of the next selection (32 per epoch on average). So to invalidate
    * the prediction the effective of one of those 32 samples should change and change the random_byte inequality.
    */
-  getBeaconProposersNextEpoch(): ValidatorIndex[] {
+  async getBeaconProposersNextEpoch(): Promise<ValidatorIndex[]> {
     if (!this.proposersNextEpoch.computed) {
       const indexes = computeProposers(
         this.config.getForkSeqAtEpoch(this.epoch + 1),
         this.proposersNextEpoch.seed,
-        this.getShufflingAtEpoch(this.nextEpoch),
+        await this.shufflingCache?.get(this.nextEpoch, this.nextDecisionRoot);
         this.effectiveBalanceIncrements
       );
       this.proposersNextEpoch = {computed: true, indexes};