初步实现数据加密、权限管理、日志审计等功能

app/main.py

@@ -10,6 +10,10 @@ from app.api.v1.router import api_router
 from app.infra.db.timescaledb.database import db as tsdb
 from app.infra.db.postgresql.database import db as pgdb
 from app.services.tjnetwork import open_project
+from app.core.config import settings
+
+# 导入审计中间件
+from app.infra.audit.middleware import AuditMiddleware
 
 
 logger = logging.getLogger()
@@ -29,6 +33,10 @@ async def lifespan(app: FastAPI):
 
     await tsdb.open()
     await pgdb.open()
+    
+    # 将数据库实例存储到 app.state，供依赖项使用
+    app.state.db = pgdb
+    logger.info("Database connection pool initialized and stored in app.state")
 
     if project_info.name:
         print(project_info.name)
@@ -36,11 +44,19 @@ async def lifespan(app: FastAPI):
 
     yield
     # 清理资源
-    tsdb.close()
-    pgdb.close()
+    await tsdb.close()
+    await pgdb.close()
+    logger.info("Database connections closed")
 
 
-app = FastAPI(lifespan=lifespan)
+app = FastAPI(
+    lifespan=lifespan,
+    title=settings.PROJECT_NAME,
+    description="TJWater Server - 供水管网智能管理系统",
+    version="1.0.0",
+    docs_url="/docs",
+    redoc_url="/redoc",
+)
 
 # 配置 CORS 中间件
 app.add_middleware(
@@ -53,6 +69,10 @@ app.add_middleware(
 
 app.add_middleware(GZipMiddleware, minimum_size=1000)
 
+# 添加审计中间件（可选，记录关键操作）
+# 如果需要启用审计日志，取消下面的注释
+# app.add_middleware(AuditMiddleware)
+
 # Include Routers
 app.include_router(api_router, prefix="/api/v1")
 # Legcy Routers without version prefix