reth_primitives_traits/
storage.rs

1use alloy_primitives::{B256, U256};
2
3/// Account storage entry.
4///
5/// `key` is the subkey when used as a value in the `StorageChangeSets` table.
6#[derive(Debug, Default, Copy, Clone, PartialEq, Eq, PartialOrd, Ord)]
7#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
8#[cfg_attr(any(test, feature = "arbitrary"), derive(arbitrary::Arbitrary))]
9#[cfg_attr(any(test, feature = "reth-codec"), reth_codecs::add_arbitrary_tests(compact))]
10pub struct StorageEntry {
11    /// Storage key.
12    pub key: B256,
13    /// Value on storage key.
14    pub value: U256,
15}
16
17impl StorageEntry {
18    /// Create a new `StorageEntry` with given key and value.
19    pub const fn new(key: B256, value: U256) -> Self {
20        Self { key, value }
21    }
22}
23
24impl From<(B256, U256)> for StorageEntry {
25    fn from((key, value): (B256, U256)) -> Self {
26        Self { key, value }
27    }
28}
29
30// NOTE: Removing reth_codec and manually encode subkey
31// and compress second part of the value. If we have compression
32// over whole value (Even SubKey) that would mess up fetching of values with seek_by_key_subkey
33#[cfg(any(test, feature = "reth-codec"))]
34impl reth_codecs::Compact for StorageEntry {
35    fn to_compact<B>(&self, buf: &mut B) -> usize
36    where
37        B: bytes::BufMut + AsMut<[u8]>,
38    {
39        // for now put full bytes and later compress it.
40        buf.put_slice(&self.key[..]);
41        self.value.to_compact(buf) + 32
42    }
43
44    fn from_compact(buf: &[u8], len: usize) -> (Self, &[u8]) {
45        let key = B256::from_slice(&buf[..32]);
46        let (value, out) = U256::from_compact(&buf[32..], len - 32);
47        (Self { key, value }, out)
48    }
49}