添加日志

mini-program/app/db.py

@@ -1,6 +1,8 @@
 from collections.abc import Generator
+import logging
+import time
 
-from sqlalchemy import create_engine, text
+from sqlalchemy import create_engine, event, text
 from sqlalchemy.orm import Session, sessionmaker
 
 try:
@@ -19,6 +21,47 @@ engine = create_engine(
     pool_recycle=1800,
     future=True,
 )
+logger = logging.getLogger("app.db")
+
+
+@event.listens_for(engine, "before_cursor_execute")
+def before_cursor_execute(
+    conn,
+    cursor,
+    statement,
+    parameters,
+    context,
+    executemany,
+):
+    conn.info.setdefault("query_start_time", []).append(time.perf_counter())
+
+
+@event.listens_for(engine, "after_cursor_execute")
+def after_cursor_execute(
+    conn,
+    cursor,
+    statement,
+    parameters,
+    context,
+    executemany,
+):
+    start_time = None
+    query_start_time = conn.info.get("query_start_time")
+    if query_start_time:
+        start_time = query_start_time.pop(-1)
+    if start_time is None:
+        return
+
+    elapsed_ms = (time.perf_counter() - start_time) * 1000
+    if elapsed_ms >= settings.slow_sql_ms:
+        logger.warning(
+            "slow sql detected",
+            extra={
+                "event": "slow_sql",
+                "sql_ms": round(elapsed_ms, 2),
+                "statement": " ".join(statement.split())[:280],
+            },
+        )
 
 SessionLocal = sessionmaker(
     bind=engine,
@@ -39,7 +82,9 @@ def get_db() -> Generator[Session, None, None]:
 def check_db_connection() -> None:
     with engine.connect() as conn:
         conn.execute(text("SELECT 1"))
+    logger.info("database connection check succeeded", extra={"event": "db_check"})
 
 
 def close_db_engine() -> None:
     engine.dispose()
+    logger.info("database engine disposed", extra={"event": "db_close"})