mirror of
https://github.com/sigp/lighthouse.git
synced 2026-05-30 20:57:10 +00:00
Gloas spec v1.7.0-alpha.5 and beacon_chain tests (#8998)
Fix database pruning post-Gloas - Fix DB pruning logic (and state summaries DAG) - Get the `beacon_chain` tests running with `FORK_NAME=gloas` 🎉 Co-Authored-By: Michael Sproul <michael@sigmaprime.io> Co-Authored-By: Jimmy Chen <jchen.tc@gmail.com> Co-Authored-By: Eitan Seri- Levi <eserilev@gmail.com> Co-Authored-By: dapplion <35266934+dapplion@users.noreply.github.com> Co-Authored-By: Eitan Seri-Levi <eserilev@ucsc.edu>
This commit is contained in:
@@ -7,7 +7,7 @@ use lru::LruCache;
|
||||
use std::collections::{BTreeMap, HashMap, HashSet};
|
||||
use std::num::NonZeroUsize;
|
||||
use tracing::instrument;
|
||||
use types::{BeaconState, ChainSpec, Epoch, EthSpec, Hash256, Slot, execution::StatePayloadStatus};
|
||||
use types::{BeaconState, ChainSpec, Epoch, EthSpec, Hash256, Slot};
|
||||
|
||||
/// Fraction of the LRU cache to leave intact during culling.
|
||||
const CULL_EXEMPT_NUMERATOR: usize = 1;
|
||||
@@ -23,10 +23,10 @@ pub struct FinalizedState<E: EthSpec> {
|
||||
state: BeaconState<E>,
|
||||
}
|
||||
|
||||
/// Map from (block_root, payload_status) -> slot -> state_root.
|
||||
/// Map from block_root -> slot -> state_root.
|
||||
#[derive(Debug, Default)]
|
||||
pub struct BlockMap {
|
||||
blocks: HashMap<(Hash256, StatePayloadStatus), SlotMap>,
|
||||
blocks: HashMap<Hash256, SlotMap>,
|
||||
}
|
||||
|
||||
/// Map from slot -> state_root.
|
||||
@@ -143,11 +143,8 @@ impl<E: EthSpec> StateCache<E> {
|
||||
return Err(Error::FinalizedStateDecreasingSlot);
|
||||
}
|
||||
|
||||
let payload_status = state.payload_status();
|
||||
|
||||
// Add to block map.
|
||||
self.block_map
|
||||
.insert(block_root, payload_status, state.slot(), state_root);
|
||||
self.block_map.insert(block_root, state.slot(), state_root);
|
||||
|
||||
// Prune block map.
|
||||
let state_roots_to_prune = self.block_map.prune(state.slot());
|
||||
@@ -270,9 +267,7 @@ impl<E: EthSpec> StateCache<E> {
|
||||
|
||||
// Record the connection from block root and slot to this state.
|
||||
let slot = state.slot();
|
||||
let payload_status = state.payload_status();
|
||||
self.block_map
|
||||
.insert(block_root, payload_status, slot, state_root);
|
||||
self.block_map.insert(block_root, slot, state_root);
|
||||
|
||||
Ok(PutStateOutcome::New(deleted_states))
|
||||
}
|
||||
@@ -321,10 +316,9 @@ impl<E: EthSpec> StateCache<E> {
|
||||
pub fn get_by_block_root(
|
||||
&mut self,
|
||||
block_root: Hash256,
|
||||
payload_status: StatePayloadStatus,
|
||||
slot: Slot,
|
||||
) -> Option<(Hash256, BeaconState<E>)> {
|
||||
let slot_map = self.block_map.blocks.get(&(block_root, payload_status))?;
|
||||
let slot_map = self.block_map.blocks.get(&block_root)?;
|
||||
|
||||
// Find the state at `slot`, or failing that the most recent ancestor.
|
||||
let state_root = slot_map
|
||||
@@ -345,12 +339,7 @@ impl<E: EthSpec> StateCache<E> {
|
||||
}
|
||||
|
||||
pub fn delete_block_states(&mut self, block_root: &Hash256) {
|
||||
let (pending_state_roots, full_state_roots) =
|
||||
self.block_map.delete_block_states(block_root);
|
||||
for slot_map in [pending_state_roots, full_state_roots]
|
||||
.into_iter()
|
||||
.flatten()
|
||||
{
|
||||
if let Some(slot_map) = self.block_map.delete_block_states(block_root) {
|
||||
for state_root in slot_map.slots.values() {
|
||||
self.states.pop(state_root);
|
||||
}
|
||||
@@ -423,14 +412,8 @@ impl<E: EthSpec> StateCache<E> {
|
||||
}
|
||||
|
||||
impl BlockMap {
|
||||
fn insert(
|
||||
&mut self,
|
||||
block_root: Hash256,
|
||||
payload_status: StatePayloadStatus,
|
||||
slot: Slot,
|
||||
state_root: Hash256,
|
||||
) {
|
||||
let slot_map = self.blocks.entry((block_root, payload_status)).or_default();
|
||||
fn insert(&mut self, block_root: Hash256, slot: Slot, state_root: Hash256) {
|
||||
let slot_map = self.blocks.entry(block_root).or_default();
|
||||
slot_map.slots.insert(slot, state_root);
|
||||
}
|
||||
|
||||
@@ -461,12 +444,8 @@ impl BlockMap {
|
||||
});
|
||||
}
|
||||
|
||||
fn delete_block_states(&mut self, block_root: &Hash256) -> (Option<SlotMap>, Option<SlotMap>) {
|
||||
let pending_state_roots = self
|
||||
.blocks
|
||||
.remove(&(*block_root, StatePayloadStatus::Pending));
|
||||
let full_state_roots = self.blocks.remove(&(*block_root, StatePayloadStatus::Full));
|
||||
(pending_state_roots, full_state_roots)
|
||||
fn delete_block_states(&mut self, block_root: &Hash256) -> Option<SlotMap> {
|
||||
self.blocks.remove(block_root)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user