refactor: clean up stage mgmt (#8980)

paradigmxyz · Jun 20, 2024 · 7229f93 · 7229f93
1 parent b5b42cf
commit 7229f93
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/bin/reth/src/commands/stage/unwind.rs b/bin/reth/src/commands/stage/unwind.rs
@@ -108,13 +108,11 @@ impl Command {
         let executor = block_executor!(provider_factory.chain_spec());
 
         let builder = if self.offline {
-            let stages = OfflineStages::new(executor, config.stages, PruneModes::default());
-            let stages = stages.builder().disable(reth_stages::StageId::SenderRecovery).build();
-            let mut builder = Pipeline::builder();
-            for stage in stages {
-                builder = builder.add_stage(stage);
-            }
-            builder
+            Pipeline::builder().add_stages(
+                OfflineStages::new(executor, config.stages, PruneModes::default())
+                    .builder()
+                    .disable(reth_stages::StageId::SenderRecovery),
+            )
         } else {
             Pipeline::builder().with_tip_sender(tip_tx).add_stages(
                 DefaultStages::new(