revm_database/states/
transition_state.rs

1use super::TransitionAccount;
2use primitives::{hash_map::Entry, Address, HashMap};
3
4/// State of accounts in transition between transaction executions.
5#[derive(Clone, Default, Debug, PartialEq, Eq)]
6pub struct TransitionState {
7    /// Block state account with account state
8    pub transitions: HashMap<Address, TransitionAccount>,
9}
10
11impl TransitionState {
12    /// Create new transition state containing one [`TransitionAccount`].
13    pub fn single(address: Address, transition: TransitionAccount) -> Self {
14        let mut transitions = HashMap::default();
15        transitions.insert(address, transition);
16        TransitionState { transitions }
17    }
18
19    /// Take the contents of this [`TransitionState`] and replace it with an
20    /// empty one.
21    ///
22    /// See [core::mem::take].
23    pub fn take(&mut self) -> TransitionState {
24        core::mem::take(self)
25    }
26
27    /// Add transitions to the transition state.
28    ///
29    /// This will insert new [`TransitionAccount`]s, or update existing ones via
30    /// [`update`][TransitionAccount::update].
31    pub fn add_transitions(
32        &mut self,
33        transitions: impl IntoIterator<Item = (Address, TransitionAccount)>,
34    ) {
35        for (address, account) in transitions {
36            match self.transitions.entry(address) {
37                Entry::Occupied(entry) => {
38                    let entry = entry.into_mut();
39                    entry.update(account);
40                }
41                Entry::Vacant(entry) => {
42                    entry.insert(account);
43                }
44            }
45        }
46    }
47}
revm_database/states/transition_state.rs

revm_database/states/
transition_state.rs
