Merge remote-tracking branch 'origin/drop-headtracker' into tree-states-hot-rebase

michaelsproul · michaelsproul · commit b49272f74232 · 2025-03-18T15:58:59.000+11:00
diff --git a/beacon_node/beacon_chain/src/schema_change/migration_schema_v23.rs b/beacon_node/beacon_chain/src/schema_change/migration_schema_v23.rs
@@ -7,7 +7,7 @@ use fork_choice::{ForkChoice, ResetPayloadStatuses};
 use ssz::{Decode, Encode};
 use ssz_derive::{Decode, Encode};
 use std::sync::Arc;
-use store::{DBColumn, Error, HotColdDB, KeyValueStoreOp, StoreItem};
+use store::{DBColumn, Error, HotColdDB, KeyValueStore, KeyValueStoreOp, StoreItem};
 use types::{Hash256, Slot};
 
 /// Dummy value to use for the canonical head block root, see below.
@@ -16,7 +16,7 @@ pub const DUMMY_CANONICAL_HEAD_BLOCK_ROOT: Hash256 = Hash256::repeat_byte(0xff);
 pub fn upgrade_to_v23<T: BeaconChainTypes>(
     db: Arc<HotColdDB<T::EthSpec, T::HotStore, T::ColdStore>>,
 ) -> Result<Vec<KeyValueStoreOp>, Error> {
-    // Set the head-tracker to empty
+    // 1) Set the head-tracker to empty
     let Some(persisted_beacon_chain_v22) =
         db.get_item::<PersistedBeaconChainV22>(&BEACON_CHAIN_DB_KEY)?
     else {
@@ -29,7 +29,19 @@ pub fn upgrade_to_v23<T: BeaconChainTypes>(
         genesis_block_root: persisted_beacon_chain_v22.genesis_block_root,
     };
 
-    let ops = vec![persisted_beacon_chain.as_kv_store_op(BEACON_CHAIN_DB_KEY)];
+    let mut ops = vec![persisted_beacon_chain.as_kv_store_op(BEACON_CHAIN_DB_KEY)];
+
+    // 2) Wipe out all state temporary flags. While un-used in V23, if there's a rollback we could
+    // end-up with an inconsistent DB.
+    for state_root_result in db
+        .hot_db
+        .iter_column_keys::<Hash256>(DBColumn::BeaconStateTemporary)
+    {
+        ops.push(KeyValueStoreOp::DeleteKey(
+            DBColumn::BeaconStateTemporary,
+            state_root_result?.as_slice().to_vec(),
+        ));
+    }
 
     Ok(ops)
 }
diff --git a/beacon_node/store/src/lib.rs b/beacon_node/store/src/lib.rs
@@ -49,12 +49,6 @@ pub use types::*;
 
 const DATA_COLUMN_DB_KEY_SIZE: usize = 32 + 8;
 
-// TODO: check that these are not used by any variant of `DBColumn` using strum_macros
-const _DEPRECATED_COLUMN_PREFIXES: &[&str] = &[
-    // BeaconStateTemporary, deleted in Mar 2025.
-    "bst",
-];
-
 pub type ColumnIter<'a, K> = Box<dyn Iterator<Item = Result<(K, Vec<u8>), Error>> + 'a>;
 pub type ColumnKeyIter<'a, K> = Box<dyn Iterator<Item = Result<K, Error>> + 'a>;
 
@@ -319,6 +313,12 @@ pub enum DBColumn {
     /// Mapping from state root to `ColdStateSummary` in the cold DB.
     #[strum(serialize = "bcs")]
     BeaconColdStateSummary,
+    /// DEPRECATED.
+    ///
+    /// Previously used for the list of temporary states stored during block import, and then made
+    /// non-temporary by the deletion of their state root from this column.
+    #[strum(serialize = "bst")]
+    BeaconStateTemporary,
     /// Execution payloads for blocks more recent than the finalized checkpoint.
     #[strum(serialize = "exp")]
     ExecPayload,
@@ -420,6 +420,7 @@ impl DBColumn {
             | Self::BeaconStateHotSnapshot
             | Self::BeaconStateHotSummary
             | Self::BeaconColdStateSummary
+            | Self::BeaconStateTemporary
             | Self::ExecPayload
             | Self::BeaconChain
             | Self::OpPool
