Add Memory Usage Logging Functionality

function_to_log.py

@@ -0,0 +1,41 @@
+import resource
+import functools
+import time
+
+
+def log\_memory\_usage(func):
+    """
+    Decorator to log memory usage before and after the execution of a function.
+
+    Args:
+        func (callable): The function to decorate.
+
+    Returns:
+        callable: The decorated function with memory usage logging.
+
+    """
+
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        start\_memory = resource.getrusage(resource.RUSAGE\_SELF).ru\_maxrss
+        start\_time = time.time()
+
+        try:
+            result = func(*args, **kwargs)
+        except Exception as e:
+            end\_memory = resource.getrusage(resource.RUSAGE\_SELF).ru\_maxrss
+            end\_time = time.time()
+            print(f"Error in {func.__name__}: {e}")
+            print(f"Memory usage before: {start_memory} KB")
+            print(f"Memory usage after: {end_memory} KB")
+            print(f"Time taken: {end_time - start_time:.4f} seconds")
+            raise
+        else:
+            end\_memory = resource.getrusage(resource.RUSAGE\_SELF).ru\_maxrss
+            end\_time = time.time()
+            print(f"Memory usage before: {start_memory} KB")
+            print(f"Memory usage after: {end_memory} KB")
+            print(f"Time taken: {end_time - start_time:.4f} seconds")
+            return result
+
+    return wrapper

prometheus_utils.py

@@ -0,0 +1,41 @@
+import resource
+import functools
+import time
+
+
+def log\_memory\_usage(function):
+    """
+    Decorator to log memory usage before and after the execution of a function.
+
+    Args:
+        function (callable): The function to decorate.
+
+    Returns:
+        callable: The decorated function with memory usage logging.
+
+    """
+
+    @functools.wraps(function)
+    def wrapper(*args, **kwargs):
+        start\_memory = resource.getrusage(resource.RUSAGE\_SELF).ru\_maxrss
+        start\_time = time.time()
+
+        try:
+            result = function(*args, **kwargs)
+        except Exception as e:
+            end\_memory = resource.getrusage(resource.RUSAGE\_SELF).ru\_maxrss
+            end\_time = time.time()
+            print(f"Error in function {function.__name__}: {e}")
+            print(f"Memory usage before: {start_memory} KB")
+            print(f"Memory usage after: {end_memory} KB")
+            print(f"Time taken: {end_time - start_time:.4f} seconds")
+            raise
+        else:
+            end\_memory = resource.getrusage(resource.RUSAGE\_SELF).ru\_maxrss
+            end\_time = time.time()
+            print(f"Memory usage before: {start_memory} KB")
+            print(f"Memory usage after: {end_memory} KB")
+            print(f"Time taken: {end_time - start_time:.4f} seconds")
+            return result
+
+    return wrapper