Fix setting message count for metrics

kysre · Feb 11, 2024 · a4382db · a4382db
1 parent 5ac0a2d
commit a4382db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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