25 lines
772 B
Python
25 lines
772 B
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# @Time : 2025/3/18 13:00
|
|
# @Author : AngesZhu
|
|
# @File : timer_utils.py
|
|
# @Desc : 定义计时上下文管理器
|
|
import time
|
|
from contextlib import contextmanager
|
|
from utils.logger_utils import logger
|
|
|
|
|
|
@contextmanager
|
|
def timer():
|
|
start_time = time.time()
|
|
fmt_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(start_time))
|
|
logger.info(f"查询开始时间: {fmt_time} 秒")
|
|
try:
|
|
yield
|
|
finally:
|
|
end_time = time.time()
|
|
fmt_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(end_time))
|
|
logger.info(f"查询结束时间: {fmt_time} 秒")
|
|
execution_time = end_time - start_time
|
|
logger.info(f"查询执行时间: {execution_time:.3f} 秒")
|