chainHead: Track announced blocks as finalized

Signed-off-by: Alexandru Vasile <[email protected]>
paritytech · Sep 27, 2024 · 35b6573 · 35b6573
1 parent 678e9bf
commit 35b6573
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/substrate/client/rpc-spec-v2/src/chain_head/chain_head_follow.rs b/substrate/client/rpc-spec-v2/src/chain_head/chain_head_follow.rs
@@ -599,6 +599,10 @@ where
 		let pruned_block_hashes =
 			self.get_pruned_hashes(&notification.stale_heads, last_finalized)?;
 
+		for finalized in &finalized_block_hashes {
+			self.announced_blocks.insert(*finalized, BlockType::FinalizedBlock);
+		}
+
 		let finalized_event = FollowEvent::Finalized(Finalized {
 			finalized_block_hashes,
 			pruned_block_hashes: pruned_block_hashes.clone(),