From 1ac16185870b1b6f7bfab2b8917947d9f0f341d6 Mon Sep 17 00:00:00 2001
From: Alexey Shekhirin <a.shekhirin@gmail.com>
Date: Thu, 16 Nov 2023 20:12:26 +0000
Subject: [PATCH] save one last checkpoint on stage run

---
 bin/reth/src/stage/run.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/bin/reth/src/stage/run.rs b/bin/reth/src/stage/run.rs
index ec3c2a7c124f..0b6d0c2d045c 100644
--- 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()?;
         }