mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2026-03-13 09:31:31 +08:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import asyncio
|
|
import functools
|
|
import time
|
|
|
|
from collections.abc import Callable
|
|
from typing import Any
|
|
|
|
from backend.common.log import log
|
|
|
|
|
|
def timer(func) -> Callable: # noqa: ANN001
|
|
"""函数耗时计时装饰器"""
|
|
|
|
@functools.wraps(func)
|
|
async def async_wrapper(*args, **kwargs) -> Any:
|
|
start_time = time.perf_counter()
|
|
result = await func(*args, **kwargs)
|
|
elapsed_seconds = time.perf_counter() - start_time
|
|
_log_time(func, elapsed_seconds)
|
|
return result
|
|
|
|
@functools.wraps(func)
|
|
def sync_wrapper(*args, **kwargs) -> Any:
|
|
start_time = time.perf_counter()
|
|
result = func(*args, **kwargs)
|
|
elapsed_seconds = time.perf_counter() - start_time
|
|
_log_time(func, elapsed_seconds)
|
|
return result
|
|
|
|
def _log_time(func, elapsed: float) -> None: # noqa: ANN001
|
|
# 智能选择单位（秒、毫秒、微秒、纳秒）
|
|
if elapsed >= 1:
|
|
unit, factor = 's', 1
|
|
else:
|
|
unit, factor = 'ms', 1e3
|
|
|
|
log.info(f'{func.__module__}.{func.__name__} | {elapsed * factor:.3f} {unit}')
|
|
|
|
return async_wrapper if asyncio.iscoroutinefunction(func) else sync_wrapper
|