diff --git a/crates/papyrus_p2p_sync/src/client/mod.rs b/crates/papyrus_p2p_sync/src/client/mod.rs index 9248d85f02..9b3e808d97 100644 --- a/crates/papyrus_p2p_sync/src/client/mod.rs +++ b/crates/papyrus_p2p_sync/src/client/mod.rs @@ -184,6 +184,7 @@ impl P2PSyncClientChannels { let header_stream = HeaderStreamBuilder::create_stream( self.header_sender, storage_reader.clone(), + None, config.wait_period_for_new_data, config.num_headers_per_query, config.stop_sync_at_block_number, @@ -192,6 +193,7 @@ impl P2PSyncClientChannels { let state_diff_stream = StateDiffStreamBuilder::create_stream( self.state_diff_sender, storage_reader.clone(), + None, config.wait_period_for_new_data, config.num_block_state_diffs_per_query, config.stop_sync_at_block_number, @@ -200,6 +202,7 @@ impl P2PSyncClientChannels { let transaction_stream = TransactionStreamFactory::create_stream( self.transaction_sender, storage_reader.clone(), + None, config.wait_period_for_new_data, config.num_block_transactions_per_query, config.stop_sync_at_block_number, @@ -208,6 +211,7 @@ impl P2PSyncClientChannels { let class_stream = ClassStreamBuilder::create_stream( self.class_sender, storage_reader.clone(), + None, config.wait_period_for_new_data, config.num_block_classes_per_query, config.stop_sync_at_block_number, diff --git a/crates/papyrus_p2p_sync/src/client/stream_builder.rs b/crates/papyrus_p2p_sync/src/client/stream_builder.rs index f566d50ecb..36718a939d 100644 --- a/crates/papyrus_p2p_sync/src/client/stream_builder.rs +++ b/crates/papyrus_p2p_sync/src/client/stream_builder.rs @@ -2,6 +2,7 @@ use std::cmp::min; use std::time::Duration; use async_stream::stream; +use futures::channel::mpsc::Receiver; use futures::future::BoxFuture; use futures::stream::BoxStream; use futures::StreamExt; @@ -55,6 +56,7 @@ where fn create_stream( mut sqmr_sender: SqmrClientSender>, storage_reader: StorageReader, + _internal_block_receiver: Option>, wait_period_for_new_data: Duration, num_blocks_per_query: u64, stop_sync_at_block_number: Option,