fix(snapshot): block increment for Transactions segment (#6274)

paradigmxyz · Jan 29, 2024 · 674c8e0 · 674c8e0
1 parent 17d94cf
commit 674c8e0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/crates/snapshot/src/segments/transactions.rs b/crates/snapshot/src/segments/transactions.rs
@@ -34,6 +34,12 @@ impl<DB: Database> Segment<DB> for Transactions {
             snapshot_provider.get_writer(*block_range.start(), SnapshotSegment::Transactions)?;
 
         for block in block_range {
+            if block > 0 {
+                let _snapshot_block =
+                    snapshot_writer.increment_block(SnapshotSegment::Transactions)?;
+                debug_assert_eq!(_snapshot_block, block);
+            }
+
             let block_body_indices = provider
                 .block_body_indices(block)?
                 .ok_or(ProviderError::BlockBodyIndicesNotFound(block))?;
@@ -48,7 +54,6 @@ impl<DB: Database> Segment<DB> for Transactions {
 
                 snapshot_writer.append_transaction(tx_number, transaction)?;
             }
-            snapshot_writer.increment_block(SnapshotSegment::Transactions)?;
         }
 
         Ok(())