From a4382dbedec2e540f010fcf671a43fddcb590d47 Mon Sep 17 00:00:00 2001
From: kasra <kasra.hajian@divar.ir>
Date: Sun, 11 Feb 2024 17:46:48 +0330
Subject: [PATCH] Fix setting message count for metrics

---
 datanode/src/datanode_server.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/datanode/src/datanode_server.py b/datanode/src/datanode_server.py
index 2e4c0f5..83779f2 100644
--- a/datanode/src/datanode_server.py
+++ b/datanode/src/datanode_server.py
@@ -44,7 +44,7 @@ def wrapper(*args, **kwargs):
 def set_message_count(func):
     def wrapper(*args, **kwargs):
         result = func(*args, **kwargs)
-        message_count.set(result)
+        message_count.set(result.remaining_messages_count)
         return result
     return wrapper