Fix metrics decorator

kysre · Feb 11, 2024 · 2d355dc · 2d355dc
1 parent 0c152d4
commit 2d355dc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/datanode/src/datanode_server.py b/datanode/src/datanode_server.py
@@ -48,15 +48,13 @@ def wrapper(*args, **kwargs):
 
 
 def get_disk_info_decorator(func):
-    def wrapper():
-        result = func()
+    def wrapper(*args, **kwargs):
         path = ConfigManager.get_prop('partition_home_path')
         st = os.statvfs(path)
         total = st.f_blocks * st.f_frsize
         used = (st.f_blocks - st.f_bfree) * st.f_frsize
         disk_total_size.labels(provider=ConfigManager.get_prop('datanode_name')).set(total)
         disk_total_size.labels(provider=ConfigManager.get_prop('datanode_name')).set(used)
-        return result
     return wrapper
 
 
@@ -173,6 +171,7 @@ def push_to_partition(partition_index: int,
 
 
 def serve():
+    # Start metrics server
     start_http_server(9000)
 
     port = ConfigManager.get_prop('server_port')