fix(bootstrap): fix num_accounts query retry logic

openmina · Mar 21, 2024 · 76cdf13 · 76cdf13
1 parent 370b4a5
commit 76cdf13
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/...ransition_frontier/sync/ledger/snarked/transition_frontier_sync_ledger_snarked_actions.rs b/...ransition_frontier/sync/ledger/snarked/transition_frontier_sync_ledger_snarked_actions.rs
@@ -160,9 +160,9 @@ impl redux::EnablingCondition<crate::State> for TransitionFrontierSyncLedgerSnar
                     // True if the next queued query is NumAccounts
                     let check_num_accounts = match ledger {
                         TransitionFrontierSyncLedgerSnarkedState::Pending {
-                            pending_num_accounts,
+                            pending_num_accounts: None,
                             ..
-                        } => ledger.is_num_accounts_query_next() && pending_num_accounts.is_none(),
+                        } => ledger.is_num_accounts_query_next(),
                         _ => false,
                     };
 
@@ -188,7 +188,6 @@ impl redux::EnablingCondition<crate::State> for TransitionFrontierSyncLedgerSnar
                         .map(|peer_rpc_state| matches!(peer_rpc_state, PeerRpcState::Init { .. }))
                         .unwrap_or(false)
                 }),
-            // TODO(sync): do we need Retry for this one?
             TransitionFrontierSyncLedgerSnarkedAction::PeerQueryNumAccountsRetry { peer_id } => {
                 None.or_else(|| {
                     let target_best_tip = state.transition_frontier.sync.best_tip()?;
@@ -198,9 +197,9 @@ impl redux::EnablingCondition<crate::State> for TransitionFrontierSyncLedgerSnar
                     // True if the next queued query is NumAccounts
                     let check_num_accounts = match ledger {
                         TransitionFrontierSyncLedgerSnarkedState::Pending {
-                            pending_num_accounts,
+                            pending_num_accounts: Some(_),
                             ..
-                        } => ledger.is_num_accounts_query_next() && pending_num_accounts.is_none(),
+                        } => ledger.is_num_accounts_query_next(),
                         _ => false,
                     };
 

diff --git a/...ransition_frontier/sync/ledger/snarked/transition_frontier_sync_ledger_snarked_effects.rs b/...ransition_frontier/sync/ledger/snarked/transition_frontier_sync_ledger_snarked_effects.rs
@@ -108,6 +108,10 @@ impl TransitionFrontierSyncLedgerSnarkedAction {
                         TransitionFrontierSyncLedgerSnarkedAction::PeerQueryNumAccountsInit {
                             peer_id: *peer_id,
                         },
+                    ) || store.dispatch(
+                        TransitionFrontierSyncLedgerSnarkedAction::PeerQueryNumAccountsRetry {
+                            peer_id: *peer_id,
+                        },
                     ) {
                         return;
                     }