mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
import timeit
|
|
from collections import defaultdict
|
|
from contextlib import contextmanager
|
|
from datetime import timedelta
|
|
from types import TracebackType
|
|
from typing import Iterator, Optional, Type
|
|
|
|
|
|
@contextmanager
|
|
def logged_duration(event: str) -> Iterator[None]:
|
|
start = timeit.default_timer()
|
|
yield
|
|
end = timeit.default_timer()
|
|
logging.debug("%s took %s", event, timedelta(seconds=end - start))
|
|
|
|
|
|
class MultiEventTracer:
|
|
def __init__(self) -> None:
|
|
self.events: dict[str, timedelta] = defaultdict(timedelta)
|
|
|
|
def __enter__(self) -> MultiEventTracer:
|
|
return self
|
|
|
|
def __exit__(
|
|
self,
|
|
exc_type: Optional[Type[BaseException]],
|
|
exc_val: Optional[BaseException],
|
|
exc_tb: Optional[TracebackType],
|
|
) -> None:
|
|
for event, duration in self.events.items():
|
|
logging.debug("%s took %s", event, duration)
|
|
|
|
@contextmanager
|
|
def trace(self, event: str) -> Iterator[None]:
|
|
start = timeit.default_timer()
|
|
yield
|
|
end = timeit.default_timer()
|
|
self.events[event] += timedelta(seconds=end - start)
|