fix(papyrus_p2p_sync): connect dropped class internal block receiver

starkware-libs · Dec 26, 2024 · 38bd732 · 38bd732
1 parent 0421768
commit 38bd732
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/crates/papyrus_p2p_sync/src/client/mod.rs b/crates/papyrus_p2p_sync/src/client/mod.rs
@@ -201,7 +201,7 @@ impl P2PSyncClientChannels {
         let class_stream = ClassStreamBuilder::create_stream(
             self.class_sender,
             storage_reader.clone(),
-            None,
+            Some(internal_blocks_receivers.class_receiver),
             config.wait_period_for_new_data,
             config.num_block_classes_per_query,
         );
@@ -268,15 +268,13 @@ pub(crate) struct InternalBlocksReceivers {
     header_receiver: Receiver<(BlockNumber, SyncBlock)>,
     state_diff_receiver: Receiver<(BlockNumber, SyncBlock)>,
     transaction_receiver: Receiver<(BlockNumber, SyncBlock)>,
-    #[allow(dead_code)]
     class_receiver: Receiver<(BlockNumber, SyncBlock)>,
 }
 
 pub struct InternalBlocksSenders {
     header_sender: Sender<(BlockNumber, SyncBlock)>,
     state_diff_sender: Sender<(BlockNumber, SyncBlock)>,
     transaction_sender: Sender<(BlockNumber, SyncBlock)>,
-    #[allow(dead_code)]
     class_sender: Sender<(BlockNumber, SyncBlock)>,
 }
 impl InternalBlocksSenders {