From 22b91606dd2a9afb11adcb1ed7a91562504f36fa Mon Sep 17 00:00:00 2001
From: magicalne <magicalne@gmail.com>
Date: Mon, 26 Sep 2022 15:49:11 +0800
Subject: [PATCH] fix: iterate on changed dist

---
 agent/app.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/agent/app.py b/agent/app.py
index c794639..72cc130 100644
--- a/agent/app.py
+++ b/agent/app.py
@@ -111,10 +111,16 @@ def reset_metrics():
         global WithdrawalDict
         if BlockNumber is not None:
             last_block_num = BlockNumber - 1
-            CommitTransacionCount.pop(last_block_num, None)
-            DepositDict.pop(last_block_num, None)
-            WithdrawalDict.pop(last_block_num, None)
-            logging.info(f"Reset metrics for block: {last_block_num}")
+            for k in list(CommitTransacionCount.keys()):
+                if k <= last_block_num:
+                    CommitTransacionCount.pop(k, None)
+            for k in list(DepositDict.keys()):
+                if k <= last_block_num:
+                    DepositDict.pop(last_block_num, None)
+            for k in list(WithdrawalDict.keys()):
+                if k <= last_block_num:
+                    WithdrawalDict.pop(last_block_num, None)
+            logging.info(f"Reset metrics for blocks before: {last_block_num}")
 
 """
 General metrics job.