From 90b6d006d1fc660c4883a4335e42f8e760c4adbe Mon Sep 17 00:00:00 2001
From: Alexey Shekhirin <a.shekhirin@gmail.com>
Date: Sat, 15 Jul 2023 21:59:03 +0100
Subject: [PATCH] fix(stages): update entities metrics on `SyncHeight` event
 (#3796)

---
 crates/stages/src/metrics/listener.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/crates/stages/src/metrics/listener.rs b/crates/stages/src/metrics/listener.rs
index 8e8102d3f3..18bd454772 100644
--- a/crates/stages/src/metrics/listener.rs
+++ b/crates/stages/src/metrics/listener.rs
@@ -59,8 +59,14 @@ impl MetricsListener {
         match event {
             MetricEvent::SyncHeight { height } => {
                 for stage_id in StageId::ALL {
-                    let stage_metrics = self.sync_metrics.get_stage_metrics(stage_id);
-                    stage_metrics.checkpoint.set(height as f64);
+                    self.handle_event(MetricEvent::StageCheckpoint {
+                        stage_id,
+                        checkpoint: StageCheckpoint {
+                            block_number: height,
+                            stage_checkpoint: None,
+                        },
+                        max_block_number: Some(height),
+                    });
                 }
             }
             MetricEvent::StageCheckpoint { stage_id, checkpoint, max_block_number } => {