mirror of
https://github.com/paradigmxyz/reth.git
synced 2026-02-07 05:25:14 -05:00
68 lines
2.2 KiB
Rust
68 lines
2.2 KiB
Rust
use alloc::{
|
|
collections::{BTreeMap, BTreeSet},
|
|
vec::Vec,
|
|
};
|
|
use alloy_primitives::{Address, BlockNumber, B256};
|
|
use core::ops::RangeInclusive;
|
|
use reth_db_api::models::BlockNumberAddress;
|
|
use reth_primitives_traits::StorageEntry;
|
|
use reth_storage_errors::provider::ProviderResult;
|
|
|
|
/// Storage reader
|
|
#[auto_impl::auto_impl(&, Arc, Box)]
|
|
pub trait StorageReader: Send + Sync {
|
|
/// Get plainstate storages for addresses and storage keys.
|
|
fn plain_state_storages(
|
|
&self,
|
|
addresses_with_keys: impl IntoIterator<Item = (Address, impl IntoIterator<Item = B256>)>,
|
|
) -> ProviderResult<Vec<(Address, Vec<StorageEntry>)>>;
|
|
|
|
/// Iterate over storage changesets and return all storage slots that were changed.
|
|
fn changed_storages_with_range(
|
|
&self,
|
|
range: RangeInclusive<BlockNumber>,
|
|
) -> ProviderResult<BTreeMap<Address, BTreeSet<B256>>>;
|
|
|
|
/// Iterate over storage changesets and return all storage slots that were changed alongside
|
|
/// each specific set of blocks.
|
|
///
|
|
/// NOTE: Get inclusive range of blocks.
|
|
fn changed_storages_and_blocks_with_range(
|
|
&self,
|
|
range: RangeInclusive<BlockNumber>,
|
|
) -> ProviderResult<BTreeMap<(Address, B256), Vec<u64>>>;
|
|
}
|
|
|
|
/// Storage ChangeSet reader
|
|
#[auto_impl::auto_impl(&, Arc, Box)]
|
|
pub trait StorageChangeSetReader: Send + Sync {
|
|
/// Iterate over storage changesets and return the storage state from before this block.
|
|
fn storage_changeset(
|
|
&self,
|
|
block_number: BlockNumber,
|
|
) -> ProviderResult<Vec<(BlockNumberAddress, StorageEntry)>>;
|
|
}
|
|
|
|
/// An enum that represents the storage location for a piece of data.
|
|
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
|
|
pub enum StorageLocation {
|
|
/// Write only to static files.
|
|
StaticFiles,
|
|
/// Write only to the database.
|
|
Database,
|
|
/// Write to both the database and static files.
|
|
Both,
|
|
}
|
|
|
|
impl StorageLocation {
|
|
/// Returns true if the storage location includes static files.
|
|
pub const fn static_files(&self) -> bool {
|
|
matches!(self, Self::StaticFiles | Self::Both)
|
|
}
|
|
|
|
/// Returns true if the storage location includes the database.
|
|
pub const fn database(&self) -> bool {
|
|
matches!(self, Self::Database | Self::Both)
|
|
}
|
|
}
|