reth_trie_db/
hashed_cursor.rs

Help
1use alloy_primitives::{B256, U256};
2use reth_db_api::{
3    cursor::{DbCursorRO, DbDupCursorRO},
4    tables,
5    transaction::DbTx,
6    DatabaseError,
7};
8use reth_primitives_traits::Account;
9use reth_trie::hashed_cursor::{HashedCursor, HashedCursorFactory, HashedStorageCursor};
10
11/// A struct wrapping database transaction that implements [`HashedCursorFactory`].
12#[derive(Debug)]
13pub struct DatabaseHashedCursorFactory<'a, TX>(&'a TX);
14
15impl<TX> Clone for DatabaseHashedCursorFactory<'_, TX> {
16    fn clone(&self) -> Self {
17        Self(self.0)
18    }
19}
20
21impl<'a, TX> DatabaseHashedCursorFactory<'a, TX> {
22    /// Create new database hashed cursor factory.
23    pub const fn new(tx: &'a TX) -> Self {
24        Self(tx)
25    }
26}
27
28impl<TX: DbTx> HashedCursorFactory for DatabaseHashedCursorFactory<'_, TX> {
29    type AccountCursor = DatabaseHashedAccountCursor<<TX as DbTx>::Cursor<tables::HashedAccounts>>;
30    type StorageCursor =
31        DatabaseHashedStorageCursor<<TX as DbTx>::DupCursor<tables::HashedStorages>>;
32
33    fn hashed_account_cursor(&self) -> Result<Self::AccountCursor, DatabaseError> {
34        Ok(DatabaseHashedAccountCursor(self.0.cursor_read::<tables::HashedAccounts>()?))
35    }
36
37    fn hashed_storage_cursor(
38        &self,
39        hashed_address: B256,
40    ) -> Result<Self::StorageCursor, DatabaseError> {
41        Ok(DatabaseHashedStorageCursor::new(
42            self.0.cursor_dup_read::<tables::HashedStorages>()?,
43            hashed_address,
44        ))
45    }
46}
47
48/// A struct wrapping database cursor over hashed accounts implementing [`HashedCursor`] for
49/// iterating over accounts.
50#[derive(Debug)]
51pub struct DatabaseHashedAccountCursor<C>(C);
52
53impl<C> DatabaseHashedAccountCursor<C> {
54    /// Create new database hashed account cursor.
55    pub const fn new(cursor: C) -> Self {
56        Self(cursor)
57    }
58}
59
60impl<C> HashedCursor for DatabaseHashedAccountCursor<C>
61where
62    C: DbCursorRO<tables::HashedAccounts>,
63{
64    type Value = Account;
65
66    fn seek(&mut self, key: B256) -> Result<Option<(B256, Self::Value)>, DatabaseError> {
67        self.0.seek(key)
68    }
69
70    fn next(&mut self) -> Result<Option<(B256, Self::Value)>, DatabaseError> {
71        self.0.next()
72    }
73}
74
75/// The structure wrapping a database cursor for hashed storage and
76/// a target hashed address. Implements [`HashedCursor`] and [`HashedStorageCursor`]
77/// for iterating over hashed storage.
78#[derive(Debug)]
79pub struct DatabaseHashedStorageCursor<C> {
80    /// Database hashed storage cursor.
81    cursor: C,
82    /// Target hashed address of the account that the storage belongs to.
83    hashed_address: B256,
84}
85
86impl<C> DatabaseHashedStorageCursor<C> {
87    /// Create new [`DatabaseHashedStorageCursor`].
88    pub const fn new(cursor: C, hashed_address: B256) -> Self {
89        Self { cursor, hashed_address }
90    }
91}
92
93impl<C> HashedCursor for DatabaseHashedStorageCursor<C>
94where
95    C: DbCursorRO<tables::HashedStorages> + DbDupCursorRO<tables::HashedStorages>,
96{
97    type Value = U256;
98
99    fn seek(&mut self, subkey: B256) -> Result<Option<(B256, Self::Value)>, DatabaseError> {
100        Ok(self.cursor.seek_by_key_subkey(self.hashed_address, subkey)?.map(|e| (e.key, e.value)))
101    }
102
103    fn next(&mut self) -> Result<Option<(B256, Self::Value)>, DatabaseError> {
104        Ok(self.cursor.next_dup_val()?.map(|e| (e.key, e.value)))
105    }
106}
107
108impl<C> HashedStorageCursor for DatabaseHashedStorageCursor<C>
109where
110    C: DbCursorRO<tables::HashedStorages> + DbDupCursorRO<tables::HashedStorages>,
111{
112    fn is_storage_empty(&mut self) -> Result<bool, DatabaseError> {
113        Ok(self.cursor.seek_exact(self.hashed_address)?.is_none())
114    }
115}
reth_trie_db/hashed_cursor.rs

reth_trie_db/
hashed_cursor.rs
