reth_storage_api/
bal.rs

1use alloc::{sync::Arc, vec::Vec};
2use alloy_eip7928::bal::DecodedBal;
3pub use alloy_eip7928::bal::RawBal;
4use alloy_eips::NumHash;
5use alloy_primitives::{BlockHash, BlockNumber, Bytes};
6use reth_storage_errors::provider::ProviderResult;
7
8/// Store for Block Access Lists (BALs).
9///
10/// This abstraction intentionally does not prescribe where BALs live. Implementations may keep
11/// recent BALs in memory, read canonical BALs from static files, or compose multiple tiers behind
12/// a single interface.
13#[auto_impl::auto_impl(&, Arc, Box)]
14pub trait BalStore: Send + Sync + 'static {
15    /// Insert the BAL for the given block.
16    ///
17    /// Implementations may buffer inserts. Call [`Self::flush`] when pending BALs need to be made
18    /// durable.
19    fn insert(&self, num_hash: NumHash, bal: RawBal) -> ProviderResult<()>;
20
21    /// Insert multiple BALs.
22    ///
23    /// The default implementation preserves the behavior of repeated [`Self::insert`] calls.
24    fn insert_many(&self, entries: Vec<(NumHash, RawBal)>) -> ProviderResult<()> {
25        for (num_hash, bal) in entries {
26            self.insert(num_hash, bal)?;
27        }
28        Ok(())
29    }
30
31    /// Flushes pending BALs for the given canonical blocks to the backing store.
32    ///
33    /// In-memory implementations may treat this as a no-op.
34    fn flush(&self, _blocks: &[NumHash]) -> ProviderResult<()> {
35        Ok(())
36    }
37
38    /// Prunes expired BALs according to the store's retention policy and the given chain tip.
39    ///
40    /// Returns the number of BALs pruned.
41    fn prune(&self, tip: BlockNumber) -> ProviderResult<usize>;
42
43    /// Fetch BALs for the given block hashes.
44    ///
45    /// The returned vector must align with `block_hashes`.
46    fn get_by_hashes(&self, block_hashes: &[BlockHash]) -> ProviderResult<Vec<Option<Bytes>>>;
47
48    /// Fetches the BAL for the given block hash.
49    fn get_by_hash(&self, block_hash: BlockHash) -> ProviderResult<Option<Bytes>> {
50        Ok(self.get_by_hashes(&[block_hash])?.into_iter().next().flatten())
51    }
52
53    /// Fetches and decodes the BAL for the given block hash.
54    fn get_decoded_by_hash(&self, block_hash: BlockHash) -> ProviderResult<Option<DecodedBal>> {
55        self.get_by_hash(block_hash)?
56            .map(DecodedBal::from_rlp_bytes)
57            .transpose()
58            .map_err(Into::into)
59    }
60
61    /// Fetch BAL response entries for the given block hashes, stopping after the soft limit is
62    /// exceeded.
63    ///
64    /// Entries are returned in request order. Unavailable BALs are represented as `None`. The
65    /// limit is soft: the entry that exceeds the limit is included.
66    fn get_by_hashes_with_limit(
67        &self,
68        block_hashes: &[BlockHash],
69        limit: GetBlockAccessListLimit,
70    ) -> ProviderResult<Vec<Option<Bytes>>> {
71        let mut out = Vec::new();
72        self.append_by_hashes_with_limit(block_hashes, limit, &mut out)?;
73        out.shrink_to_fit();
74        Ok(out)
75    }
76
77    /// Extends the given vector with BAL response entries for the given hashes.
78    ///
79    /// This adheres to the expected behavior of [`Self::get_by_hashes_with_limit`].
80    fn append_by_hashes_with_limit(
81        &self,
82        block_hashes: &[BlockHash],
83        limit: GetBlockAccessListLimit,
84        out: &mut Vec<Option<Bytes>>,
85    ) -> ProviderResult<()> {
86        let mut size = 0;
87        for bal in self.get_by_hashes(block_hashes)? {
88            size += bal.as_ref().map_or(1, |bytes| bytes.len());
89            out.push(bal);
90
91            if limit.exceeds(size) {
92                break
93            }
94        }
95        Ok(())
96    }
97}
98
99/// The limit to enforce for [`BalStore::get_by_hashes_with_limit`].
100#[derive(Debug, Clone, Copy, Eq, PartialEq)]
101pub enum GetBlockAccessListLimit {
102    /// No limit, return all BALs.
103    None,
104    /// Enforce a size limit on the returned BALs, for example 2MB.
105    ResponseSizeSoftLimit(usize),
106}
107
108impl GetBlockAccessListLimit {
109    /// Returns true if the given size exceeds the limit.
110    #[inline]
111    pub const fn exceeds(&self, size: usize) -> bool {
112        match self {
113            Self::None => false,
114            Self::ResponseSizeSoftLimit(limit) => size > *limit,
115        }
116    }
117}
118
119/// Clone-friendly façade around a BAL store implementation.
120#[derive(Clone)]
121pub struct BalStoreHandle {
122    inner: Arc<dyn BalStore>,
123}
124
125impl BalStoreHandle {
126    /// Creates a new [`BalStoreHandle`] from the given implementation.
127    pub fn new(inner: impl BalStore) -> Self {
128        Self { inner: Arc::new(inner) }
129    }
130
131    /// Creates a [`BalStoreHandle`] backed by [`NoopBalStore`].
132    pub fn noop() -> Self {
133        Self::new(NoopBalStore)
134    }
135
136    /// Insert the BAL for the given block.
137    #[inline]
138    pub fn insert(&self, num_hash: NumHash, bal: RawBal) -> ProviderResult<()> {
139        self.inner.insert(num_hash, bal)
140    }
141
142    /// Insert multiple BALs.
143    #[inline]
144    pub fn insert_many(&self, entries: Vec<(NumHash, RawBal)>) -> ProviderResult<()> {
145        self.inner.insert_many(entries)
146    }
147
148    /// Flushes pending BALs for the given canonical blocks to the backing store.
149    #[inline]
150    pub fn flush(&self, blocks: &[NumHash]) -> ProviderResult<()> {
151        self.inner.flush(blocks)
152    }
153
154    /// Prunes expired BALs according to the store's retention policy and the given chain tip.
155    #[inline]
156    pub fn prune(&self, tip: BlockNumber) -> ProviderResult<usize> {
157        self.inner.prune(tip)
158    }
159
160    /// Fetch BALs for the given block hashes.
161    #[inline]
162    pub fn get_by_hashes(&self, block_hashes: &[BlockHash]) -> ProviderResult<Vec<Option<Bytes>>> {
163        self.inner.get_by_hashes(block_hashes)
164    }
165
166    /// Fetches the BAL for the given block hash.
167    #[inline]
168    pub fn get_by_hash(&self, block_hash: BlockHash) -> ProviderResult<Option<Bytes>> {
169        self.inner.get_by_hash(block_hash)
170    }
171
172    /// Fetches and decodes the BAL for the given block hash.
173    #[inline]
174    pub fn get_decoded_by_hash(&self, block_hash: BlockHash) -> ProviderResult<Option<DecodedBal>> {
175        self.inner.get_decoded_by_hash(block_hash)
176    }
177
178    /// Fetch BAL response entries for the given block hashes, stopping after the soft limit is
179    /// exceeded.
180    #[inline]
181    pub fn get_by_hashes_with_limit(
182        &self,
183        block_hashes: &[BlockHash],
184        limit: GetBlockAccessListLimit,
185    ) -> ProviderResult<Vec<Option<Bytes>>> {
186        self.inner.get_by_hashes_with_limit(block_hashes, limit)
187    }
188
189    /// Extends the given vector with BAL response entries for the given hashes.
190    #[inline]
191    pub fn append_by_hashes_with_limit(
192        &self,
193        block_hashes: &[BlockHash],
194        limit: GetBlockAccessListLimit,
195        out: &mut Vec<Option<Bytes>>,
196    ) -> ProviderResult<()> {
197        self.inner.append_by_hashes_with_limit(block_hashes, limit, out)
198    }
199}
200
201impl Default for BalStoreHandle {
202    fn default() -> Self {
203        Self::noop()
204    }
205}
206
207impl core::fmt::Debug for BalStoreHandle {
208    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
209        f.debug_struct("BalStoreHandle").finish_non_exhaustive()
210    }
211}
212
213/// Provider-side access to BAL storage.
214#[auto_impl::auto_impl(&, Arc)]
215pub trait BalProvider {
216    /// Returns the configured BAL store handle.
217    fn bal_store(&self) -> &BalStoreHandle;
218
219    /// Fetches the BAL for the given block hash.
220    fn get_bal_by_hash(&self, block_hash: BlockHash) -> ProviderResult<Option<Bytes>> {
221        self.bal_store().get_by_hash(block_hash)
222    }
223
224    /// Fetches BALs for the given block hashes.
225    fn get_bals_by_hashes(&self, block_hashes: &[BlockHash]) -> ProviderResult<Vec<Option<Bytes>>> {
226        self.bal_store().get_by_hashes(block_hashes)
227    }
228
229    /// Fetches BAL response entries for the given block hashes, stopping after the soft limit is
230    /// exceeded.
231    fn get_bals_by_hashes_with_limit(
232        &self,
233        block_hashes: &[BlockHash],
234        limit: GetBlockAccessListLimit,
235    ) -> ProviderResult<Vec<Option<Bytes>>> {
236        self.bal_store().get_by_hashes_with_limit(block_hashes, limit)
237    }
238}
239
240/// No-op BAL store used as the default wiring target until a concrete implementation is injected.
241#[derive(Debug, Default, Clone, Copy)]
242pub struct NoopBalStore;
243
244impl BalStore for NoopBalStore {
245    fn insert(&self, _num_hash: NumHash, _bal: RawBal) -> ProviderResult<()> {
246        Ok(())
247    }
248
249    fn insert_many(&self, _entries: Vec<(NumHash, RawBal)>) -> ProviderResult<()> {
250        Ok(())
251    }
252
253    fn prune(&self, _tip: BlockNumber) -> ProviderResult<usize> {
254        Ok(0)
255    }
256
257    fn get_by_hashes(&self, block_hashes: &[BlockHash]) -> ProviderResult<Vec<Option<Bytes>>> {
258        Ok(block_hashes.iter().map(|_| None).collect())
259    }
260
261    fn append_by_hashes_with_limit(
262        &self,
263        block_hashes: &[BlockHash],
264        limit: GetBlockAccessListLimit,
265        out: &mut Vec<Option<Bytes>>,
266    ) -> ProviderResult<()> {
267        let mut size = 0;
268        for _ in block_hashes {
269            size += 1;
270            out.push(None);
271
272            if limit.exceeds(size) {
273                break
274            }
275        }
276        Ok(())
277    }
278}
279
280#[cfg(test)]
281mod tests {
282    use super::*;
283    use alloy_primitives::B256;
284
285    const EMPTY_LIST_CODE: u8 = 0xc0;
286
287    #[test]
288    fn noop_store_returns_empty_results() {
289        let store = BalStoreHandle::default();
290        let hashes = [B256::random(), B256::random()];
291
292        let by_hash = store.get_by_hashes(&hashes).unwrap();
293
294        assert_eq!(by_hash, vec![None, None]);
295        assert!(store.get_by_hash(B256::random()).unwrap().is_none());
296        assert_eq!(store.prune(10).unwrap(), 0);
297    }
298
299    #[test]
300    fn noop_store_flush_is_noop() {
301        let store = BalStoreHandle::default();
302
303        store.flush(&[]).unwrap();
304    }
305
306    #[test]
307    fn noop_store_decoded_lookup_returns_none() {
308        let store = BalStoreHandle::default();
309
310        assert!(store.get_decoded_by_hash(B256::random()).unwrap().is_none());
311    }
312
313    #[test]
314    fn decoded_lookup_decodes_raw_bal() {
315        let hash = B256::random();
316        let raw_bal = Bytes::from_static(&[EMPTY_LIST_CODE]);
317        let store = BalStoreHandle::new(TestBalStore { hash, raw_bal: raw_bal.clone() });
318
319        assert_eq!(store.get_by_hash(hash).unwrap(), Some(raw_bal.clone()));
320
321        let decoded = store.get_decoded_by_hash(hash).unwrap().unwrap();
322
323        assert_eq!(decoded.as_raw(), &raw_bal);
324    }
325
326    #[test]
327    fn noop_store_limited_lookup_returns_prefix() {
328        let store = BalStoreHandle::default();
329        let hashes = [B256::random(), B256::random(), B256::random()];
330
331        let limited = store
332            .get_by_hashes_with_limit(&hashes, GetBlockAccessListLimit::ResponseSizeSoftLimit(1))
333            .unwrap();
334
335        assert_eq!(limited, vec![None, None]);
336    }
337
338    #[test]
339    fn block_access_list_limit() {
340        let limit_none = GetBlockAccessListLimit::None;
341        assert!(!limit_none.exceeds(usize::MAX));
342
343        let size_limit_2mb = GetBlockAccessListLimit::ResponseSizeSoftLimit(2 * 1024 * 1024);
344        assert!(!size_limit_2mb.exceeds(1024 * 1024));
345        assert!(!size_limit_2mb.exceeds(2 * 1024 * 1024));
346        assert!(size_limit_2mb.exceeds(3 * 1024 * 1024));
347    }
348
349    #[derive(Debug)]
350    struct TestBalStore {
351        hash: B256,
352        raw_bal: Bytes,
353    }
354
355    impl BalStore for TestBalStore {
356        fn insert(&self, _num_hash: NumHash, _bal: RawBal) -> ProviderResult<()> {
357            Ok(())
358        }
359
360        fn prune(&self, _tip: BlockNumber) -> ProviderResult<usize> {
361            Ok(0)
362        }
363
364        fn get_by_hashes(&self, block_hashes: &[BlockHash]) -> ProviderResult<Vec<Option<Bytes>>> {
365            Ok(block_hashes
366                .iter()
367                .map(|hash| (*hash == self.hash).then(|| self.raw_bal.clone()))
368                .collect())
369        }
370    }
371}
reth_storage_api/bal.rs

reth_storage_api/
bal.rs
