87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
from fastapi import FastAPI
|
||
from contextlib import asynccontextmanager
|
||
from fastapi.middleware.gzip import GZipMiddleware
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
import logging
|
||
from datetime import datetime
|
||
|
||
import app.services.project_info as project_info
|
||
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.infra.db.dynamic_manager import project_connection_manager
|
||
from app.infra.db.metadata.database import close_metadata_engine
|
||
from app.services.tjnetwork import open_project
|
||
from app.core.config import settings
|
||
|
||
# 导入审计中间件
|
||
from app.infra.audit.middleware import AuditMiddleware
|
||
|
||
|
||
logger = logging.getLogger()
|
||
logger.setLevel(logging.INFO)
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
logger.info("**********************************************************")
|
||
logger.info(str(datetime.now()))
|
||
logger.info("TJWater CloudService is starting...")
|
||
logger.info("**********************************************************")
|
||
|
||
# 初始化数据库连接池
|
||
tsdb.init_pool()
|
||
pgdb.init_pool()
|
||
|
||
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)
|
||
open_project(project_info.name)
|
||
|
||
yield
|
||
# 清理资源
|
||
await tsdb.close()
|
||
await pgdb.close()
|
||
await project_connection_manager.close_all()
|
||
await close_metadata_engine()
|
||
logger.info("Database connections closed")
|
||
|
||
|
||
app = FastAPI(
|
||
lifespan=lifespan,
|
||
title=settings.PROJECT_NAME,
|
||
description="TJWater Server - 供水管网智能管理系统",
|
||
version="1.0.0",
|
||
docs_url="/docs",
|
||
redoc_url="/redoc",
|
||
)
|
||
|
||
|
||
# Include Routers
|
||
app.include_router(api_router, prefix="/api/v1")
|
||
# Legcy Routers without version prefix
|
||
app.include_router(api_router)
|
||
|
||
# 配置中间件
|
||
app.add_middleware(GZipMiddleware, minimum_size=1000)
|
||
# 添加审计中间件（可选，记录关键操作）
|
||
app.add_middleware(AuditMiddleware)
|
||
# 配置 CORS 中间件
|
||
# 确保这是你最后一个添加的 app.add_middleware
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=[
|
||
"http://localhost:3000", # 必须明确指定
|
||
"http://127.0.0.1:3000", # 建议同时加上这个
|
||
],
|
||
allow_credentials=True, # 既然这里是 True
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|