revm_database/states/
transition_state.rs

1use super::TransitionAccount;
2use primitives::{hash_map::Entry, Address, HashMap};
3use std::vec::Vec;
4
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(&mut self, transitions: Vec<(Address, TransitionAccount)>) {
32        for (address, account) in transitions {
33            match self.transitions.entry(address) {
34                Entry::Occupied(entry) => {
35                    let entry = entry.into_mut();
36                    entry.update(account);
37                }
38                Entry::Vacant(entry) => {
39                    entry.insert(account);
40                }
41            }
42        }
43    }
44}
revm_database/states/transition_state.rs

revm_database/states/
transition_state.rs
