batch transform headers

Author: frisitano

crates/net/network/src/fetch/mod.rs

@@ -6,7 +6,7 @@ pub use client::FetchClient;
 
 use crate::{message::BlockRequest, session::BlockRangeInfo, transform::header::HeaderTransform};
 use alloy_primitives::B256;
-use futures::{future::join_all, StreamExt};
+use futures::StreamExt;
 use reth_eth_wire::{EthNetworkPrimitives, GetBlockBodies, GetBlockHeaders, NetworkPrimitives};
 use reth_network_api::test_utils::PeersHandle;
 use reth_network_p2p::{
@@ -282,9 +282,7 @@ impl<N: NetworkPrimitives> StateFetcher<N> {
             let header_transform = self.header_transform.clone();
             tokio::spawn(async move {
                 let res = match res {
-                    Ok(headers) => {
-                        Ok(join_all(headers.into_iter().map(|h| header_transform.map(h))).await)
-                    }
+                    Ok(headers) => Ok(header_transform.map(headers).await),
                     Err(e) => Err(e),
                 };
 

crates/net/network/src/transform/header.rs

@@ -2,17 +2,17 @@
 
 use reth_primitives_traits::BlockHeader;
 
-/// An instance of the trait applies a mapping to the input header.
+/// An instance of the trait applies a mapping to the input headers.
 #[async_trait::async_trait]
 pub trait HeaderTransform<H: BlockHeader>: std::fmt::Debug + Send + Sync {
-    /// Applies a mapping to the input header.
-    async fn map(&self, header: H) -> H;
+    /// Applies a mapping to the input headers.
+    async fn map(&self, headers: Vec<H>) -> Vec<H>;
 }
 
 #[async_trait::async_trait]
 impl<H: BlockHeader> HeaderTransform<H> for () {
-    async fn map(&self, header: H) -> H {
-        header
+    async fn map(&self, headers: Vec<H>) -> Vec<H> {
+        headers
     }
 }