save one last checkpoint on stage run

paradigmxyz · Nov 16, 2023 · 1ac1618 · 1ac1618
1 parent 662da5c
commit 1ac1618
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bin/reth/src/stage/run.rs b/bin/reth/src/stage/run.rs
@@ -265,13 +265,16 @@ impl Command {
             input.checkpoint = Some(checkpoint);
 
             provider_rw.save_stage_checkpoint(exec_stage.id(), checkpoint)?;
-
             if self.commit {
                 provider_rw.commit()?;
                 provider_rw = factory.provider_rw().map_err(PipelineError::Interface)?;
             }
         }
 
+        if let Some(checkpoint) = input.checkpoint {
+            provider_rw.save_stage_checkpoint(exec_stage.id(), checkpoint)?;
+        }
+
         if self.commit {
             provider_rw.commit()?;
         }