45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
|
import cProfile
|
||
|
|
import pstats
|
||
|
|
import io
|
||
|
|
from functools import wraps
|
||
|
|
from collections import defaultdict
|
||
|
|
|
||
|
|
class Metrics:
|
||
|
|
def __init__(self):
|
||
|
|
self.call_count = defaultdict(int)
|
||
|
|
self.total_time = defaultdict(float)
|
||
|
|
|
||
|
|
def measure(self, func):
|
||
|
|
@wraps(func)
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
self.call_count[func.__name__] += 1
|
||
|
|
|
||
|
|
pr = cProfile.Profile()
|
||
|
|
pr.enable()
|
||
|
|
|
||
|
|
result = func(*args, **kwargs)
|
||
|
|
|
||
|
|
pr.disable()
|
||
|
|
s = io.StringIO()
|
||
|
|
ps = pstats.Stats(pr, stream=s).sort_stats('cumulative')
|
||
|
|
ps.print_stats()
|
||
|
|
|
||
|
|
# Extract the total time spent in the function
|
||
|
|
time_spent = float(s.getvalue().split('\n')[0].split()[-2])
|
||
|
|
self.total_time[func.__name__] += time_spent
|
||
|
|
|
||
|
|
return result
|
||
|
|
return wrapper
|
||
|
|
|
||
|
|
def get_metrics(self):
|
||
|
|
metrics = {}
|
||
|
|
for func_name in self.call_count:
|
||
|
|
metrics[func_name] = {
|
||
|
|
'call_count': self.call_count[func_name],
|
||
|
|
'total_time': self.total_time[func_name],
|
||
|
|
'average_time': self.total_time[func_name] / self.call_count[func_name] if self.call_count[func_name] > 0 else 0
|
||
|
|
}
|
||
|
|
return metrics
|
||
|
|
|
||
|
|
# Create a global instance of Metrics
|
||
|
|
metrics = Metrics()
|