alloy_network_primitives/
block.rs1use alloy_primitives::B256;
2
3use crate::TransactionResponse;
4use alloc::{vec, vec::Vec};
5use alloy_consensus::error::ValueError;
6use alloy_eips::Encodable2718;
7use core::slice;
8
9#[derive(Clone, Debug, PartialEq, Eq)]
12#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
13#[cfg_attr(feature = "serde", serde(untagged))]
14pub enum BlockTransactions<T> {
15 Full(Vec<T>),
17 Hashes(Vec<B256>),
19 Uncle,
21}
22
23impl<T> Default for BlockTransactions<T> {
24 fn default() -> Self {
25 Self::Hashes(Vec::default())
26 }
27}
28
29impl<T> BlockTransactions<T> {
30 #[inline]
32 pub const fn is_hashes(&self) -> bool {
33 matches!(self, Self::Hashes(_))
34 }
35
36 pub fn as_hashes(&self) -> Option<&[B256]> {
38 match self {
39 Self::Hashes(hashes) => Some(hashes),
40 _ => None,
41 }
42 }
43
44 pub fn first_transaction(&self) -> Option<&T> {
46 self.as_transactions().and_then(|txs| txs.first())
47 }
48
49 #[inline]
51 pub const fn is_full(&self) -> bool {
52 matches!(self, Self::Full(_))
53 }
54
55 pub fn map<U>(self, f: impl FnMut(T) -> U) -> BlockTransactions<U> {
59 match self {
60 Self::Full(txs) => BlockTransactions::Full(txs.into_iter().map(f).collect()),
61 Self::Hashes(hashes) => BlockTransactions::Hashes(hashes),
62 Self::Uncle => BlockTransactions::Uncle,
63 }
64 }
65
66 pub fn try_map<U, E>(
70 self,
71 f: impl FnMut(T) -> Result<U, E>,
72 ) -> Result<BlockTransactions<U>, E> {
73 match self {
74 Self::Full(txs) => {
75 Ok(BlockTransactions::Full(txs.into_iter().map(f).collect::<Result<_, _>>()?))
76 }
77 Self::Hashes(hashes) => Ok(BlockTransactions::Hashes(hashes)),
78 Self::Uncle => Ok(BlockTransactions::Uncle),
79 }
80 }
81
82 pub fn as_transactions(&self) -> Option<&[T]> {
86 match self {
87 Self::Full(txs) => Some(txs),
88 _ => None,
89 }
90 }
91
92 pub fn calculate_transactions_root(&self) -> Option<B256>
96 where
97 T: Encodable2718,
98 {
99 self.as_transactions().map(alloy_consensus::proofs::calculate_transaction_root)
100 }
101
102 #[inline]
104 pub const fn is_uncle(&self) -> bool {
105 matches!(self, Self::Uncle)
106 }
107
108 #[doc(alias = "transactions")]
112 pub fn txns(&self) -> impl Iterator<Item = &T> {
113 self.as_transactions().map(|txs| txs.iter()).unwrap_or_else(|| [].iter())
114 }
115
116 pub fn into_transactions(self) -> vec::IntoIter<T> {
119 match self {
120 Self::Full(txs) => txs.into_iter(),
121 _ => vec::IntoIter::default(),
122 }
123 }
124
125 pub fn into_transactions_vec(self) -> Vec<T> {
129 match self {
130 Self::Full(txs) => txs,
131 _ => vec![],
132 }
133 }
134
135 pub fn try_into_transactions(self) -> Result<Vec<T>, ValueError<Self>> {
139 match self {
140 Self::Full(txs) => Ok(txs),
141 txs @ Self::Hashes(_) => Err(ValueError::new_static(txs, "Unexpected hashes variant")),
142 txs @ Self::Uncle => Err(ValueError::new_static(txs, "Unexpected uncle variant")),
143 }
144 }
145
146 #[inline]
148 pub const fn uncle() -> Self {
149 Self::Uncle
150 }
151
152 #[inline]
154 pub fn len(&self) -> usize {
155 match self {
156 Self::Hashes(h) => h.len(),
157 Self::Full(f) => f.len(),
158 Self::Uncle => 0,
159 }
160 }
161
162 #[inline]
164 pub fn is_empty(&self) -> bool {
165 self.len() == 0
166 }
167}
168
169impl<T: TransactionResponse> BlockTransactions<T> {
170 pub fn new_hashes(txs: impl IntoIterator<Item = impl AsRef<T>>) -> Self {
172 Self::Hashes(txs.into_iter().map(|tx| tx.as_ref().tx_hash()).collect())
173 }
174
175 #[inline]
177 pub fn convert_to_hashes(&mut self) {
178 if !self.is_hashes() {
179 *self = Self::Hashes(self.hashes().collect());
180 }
181 }
182
183 #[inline]
185 pub fn convert_to_hashes_if(&mut self, condition: bool) {
186 if !condition {
187 return;
188 }
189 self.convert_to_hashes();
190 }
191
192 #[inline]
194 pub fn into_hashes(mut self) -> Self {
195 self.convert_to_hashes();
196 self
197 }
198
199 #[inline]
201 pub fn into_hashes_if(self, condition: bool) -> Self {
202 if !condition {
203 return self;
204 }
205 self.into_hashes()
206 }
207
208 #[deprecated = "use `hashes` instead"]
210 #[inline]
211 pub fn iter(&self) -> BlockTransactionHashes<'_, T> {
212 self.hashes()
213 }
214
215 #[inline]
217 pub fn hashes(&self) -> BlockTransactionHashes<'_, T> {
218 BlockTransactionHashes::new(self)
219 }
220}
221
222impl<T> From<Vec<B256>> for BlockTransactions<T> {
223 fn from(hashes: Vec<B256>) -> Self {
224 Self::Hashes(hashes)
225 }
226}
227
228impl<T: TransactionResponse> From<Vec<T>> for BlockTransactions<T> {
229 fn from(transactions: Vec<T>) -> Self {
230 Self::Full(transactions)
231 }
232}
233
234#[derive(Clone, Debug)]
238pub struct BlockTransactionHashes<'a, T>(BlockTransactionHashesInner<'a, T>);
239
240#[derive(Clone, Debug)]
241enum BlockTransactionHashesInner<'a, T> {
242 Hashes(slice::Iter<'a, B256>),
243 Full(slice::Iter<'a, T>),
244 Uncle,
245}
246
247impl<'a, T> BlockTransactionHashes<'a, T> {
248 #[inline]
249 fn new(txs: &'a BlockTransactions<T>) -> Self {
250 Self(match txs {
251 BlockTransactions::Hashes(txs) => BlockTransactionHashesInner::Hashes(txs.iter()),
252 BlockTransactions::Full(txs) => BlockTransactionHashesInner::Full(txs.iter()),
253 BlockTransactions::Uncle => BlockTransactionHashesInner::Uncle,
254 })
255 }
256}
257
258impl<T: TransactionResponse> Iterator for BlockTransactionHashes<'_, T> {
259 type Item = B256;
260
261 #[inline]
262 fn next(&mut self) -> Option<Self::Item> {
263 match &mut self.0 {
264 BlockTransactionHashesInner::Hashes(txs) => txs.next().copied(),
265 BlockTransactionHashesInner::Full(txs) => txs.next().map(|tx| tx.tx_hash()),
266 BlockTransactionHashesInner::Uncle => None,
267 }
268 }
269
270 #[inline]
271 fn size_hint(&self) -> (usize, Option<usize>) {
272 match &self.0 {
273 BlockTransactionHashesInner::Full(txs) => txs.size_hint(),
274 BlockTransactionHashesInner::Hashes(txs) => txs.size_hint(),
275 BlockTransactionHashesInner::Uncle => (0, Some(0)),
276 }
277 }
278}
279
280impl<T: TransactionResponse> ExactSizeIterator for BlockTransactionHashes<'_, T> {
281 #[inline]
282 fn len(&self) -> usize {
283 match &self.0 {
284 BlockTransactionHashesInner::Full(txs) => txs.len(),
285 BlockTransactionHashesInner::Hashes(txs) => txs.len(),
286 BlockTransactionHashesInner::Uncle => 0,
287 }
288 }
289}
290
291impl<T: TransactionResponse> DoubleEndedIterator for BlockTransactionHashes<'_, T> {
292 #[inline]
293 fn next_back(&mut self) -> Option<Self::Item> {
294 match &mut self.0 {
295 BlockTransactionHashesInner::Full(txs) => txs.next_back().map(|tx| tx.tx_hash()),
296 BlockTransactionHashesInner::Hashes(txs) => txs.next_back().copied(),
297 BlockTransactionHashesInner::Uncle => None,
298 }
299 }
300}
301
302#[cfg(feature = "std")]
303impl<T: TransactionResponse> std::iter::FusedIterator for BlockTransactionHashes<'_, T> {}
304
305#[derive(Clone, Copy, Debug, PartialEq, Eq, Default)]
310#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
311pub enum BlockTransactionsKind {
312 #[default]
314 Hashes,
315 Full,
317}
318
319impl BlockTransactionsKind {
320 pub const fn is_hashes(&self) -> bool {
322 matches!(self, Self::Hashes)
323 }
324
325 pub const fn is_full(&self) -> bool {
327 matches!(self, Self::Full)
328 }
329}
330
331impl From<bool> for BlockTransactionsKind {
332 fn from(is_full: bool) -> Self {
333 if is_full {
334 Self::Full
335 } else {
336 Self::Hashes
337 }
338 }
339}
340
341impl From<BlockTransactionsKind> for bool {
342 fn from(kind: BlockTransactionsKind) -> Self {
343 match kind {
344 BlockTransactionsKind::Full => true,
345 BlockTransactionsKind::Hashes => false,
346 }
347 }
348}
349
350#[cfg(test)]
351mod tests {
352 use super::*;
353
354 #[test]
355 fn test_full_conversion() {
356 let full = true;
357 assert_eq!(BlockTransactionsKind::Full, full.into());
358
359 let full = false;
360 assert_eq!(BlockTransactionsKind::Hashes, full.into());
361 }
362
363 #[test]
364 fn test_block_transactions_default() {
365 let default: BlockTransactions<()> = BlockTransactions::default();
366 assert!(default.is_hashes());
367 assert_eq!(default.len(), 0);
368 }
369
370 #[test]
371 fn test_block_transactions_is_methods() {
372 let hashes: BlockTransactions<()> = BlockTransactions::Hashes(vec![B256::ZERO]);
373 let full: BlockTransactions<u32> = BlockTransactions::Full(vec![42]);
374 let uncle: BlockTransactions<()> = BlockTransactions::Uncle;
375
376 assert!(hashes.is_hashes());
377 assert!(!hashes.is_full());
378 assert!(!hashes.is_uncle());
379
380 assert!(full.is_full());
381 assert!(!full.is_hashes());
382 assert!(!full.is_uncle());
383
384 assert!(uncle.is_uncle());
385 assert!(!uncle.is_full());
386 assert!(!uncle.is_hashes());
387 }
388
389 #[test]
390 fn test_as_hashes() {
391 let hashes = vec![B256::ZERO, B256::repeat_byte(1)];
392 let tx_hashes: BlockTransactions<()> = BlockTransactions::Hashes(hashes.clone());
393
394 assert_eq!(tx_hashes.as_hashes(), Some(hashes.as_slice()));
395 }
396
397 #[test]
398 fn test_as_transactions() {
399 let transactions = vec![42, 43];
400 let txs = BlockTransactions::Full(transactions.clone());
401
402 assert_eq!(txs.as_transactions(), Some(transactions.as_slice()));
403 }
404
405 #[test]
406 fn test_block_transactions_len_and_is_empty() {
407 let hashes: BlockTransactions<()> = BlockTransactions::Hashes(vec![B256::ZERO]);
408 let full = BlockTransactions::Full(vec![42]);
409 let uncle: BlockTransactions<()> = BlockTransactions::Uncle;
410
411 assert_eq!(hashes.len(), 1);
412 assert_eq!(full.len(), 1);
413 assert_eq!(uncle.len(), 0);
414
415 assert!(!hashes.is_empty());
416 assert!(!full.is_empty());
417 assert!(uncle.is_empty());
418 }
419
420 #[test]
421 fn test_block_transactions_txns_iterator() {
422 let transactions = vec![42, 43];
423 let txs = BlockTransactions::Full(transactions);
424 let mut iter = txs.txns();
425
426 assert_eq!(iter.next(), Some(&42));
427 assert_eq!(iter.next(), Some(&43));
428 assert_eq!(iter.next(), None);
429 }
430
431 #[test]
432 fn test_block_transactions_into_transactions() {
433 let transactions = vec![42, 43];
434 let txs = BlockTransactions::Full(transactions.clone());
435 let collected: Vec<_> = txs.into_transactions().collect();
436
437 assert_eq!(collected, transactions);
438 }
439
440 #[test]
441 fn test_block_transactions_kind_conversion() {
442 let full: BlockTransactionsKind = true.into();
443 assert_eq!(full, BlockTransactionsKind::Full);
444
445 let hashes: BlockTransactionsKind = false.into();
446 assert_eq!(hashes, BlockTransactionsKind::Hashes);
447
448 let bool_full: bool = BlockTransactionsKind::Full.into();
449 assert!(bool_full);
450
451 let bool_hashes: bool = BlockTransactionsKind::Hashes.into();
452 assert!(!bool_hashes);
453 }
454}