#!/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} 秒")