Skip to content

Commit 6aa61f3

Browse files
authored
Add performance tracking decorator
Implement a performance decorator to track execution time and memory usage.
1 parent b60efee commit 6aa61f3

1 file changed

Lines changed: 32 additions & 0 deletions

File tree

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
import time
2+
import tracemalloc
3+
from functools import wraps
4+
5+
tracemalloc.start()
6+
7+
8+
def performance(fn):
9+
@wraps(fn)
10+
def wrapper(*args, **kwargs):
11+
12+
performance.counter += 1
13+
14+
tracemalloc.reset_peak()
15+
start_time = time.perf_counter()
16+
result = fn(*args, **kwargs)
17+
18+
end_time = time.perf_counter()
19+
20+
performance.total_time += end_time - start_time
21+
22+
current_memory, peak_memory = tracemalloc.get_traced_memory()
23+
performance.total_mem += peak_memory
24+
25+
return result
26+
27+
return wrapper
28+
29+
30+
performance.counter = 0
31+
performance.total_time = 0.0
32+
performance.total_mem = 0

0 commit comments

Comments
 (0)