31 lines
875 B
Python
31 lines
875 B
Python
|
||
from fastapi import FastAPI
|
||
from .api import router as api_router
|
||
|
||
from .database import database, create_tables
|
||
|
||
app = FastAPI(
|
||
title="人脸识别服务 V2 - PostgreSQL版",
|
||
description="一个使用PostgreSQL和Milvus的高性能人脸识别服务。",
|
||
version="2.1.0"
|
||
)
|
||
|
||
@app.on_event("startup")
|
||
async def on_startup():
|
||
"""应用启动时，连接数据库并创建表"""
|
||
await database.connect()
|
||
create_tables() # SQLAlchemy的create_all是同步的，所以这样调用没问题
|
||
|
||
@app.on_event("shutdown")
|
||
async def on_shutdown():
|
||
"""应用关闭时，断开数据库连接"""
|
||
await database.disconnect()
|
||
|
||
|
||
# 挂载API路由
|
||
app.include_router(api_router, prefix="/api", tags=["Face Recognition"])
|
||
|
||
@app.get("/", summary="服务健康检查")
|
||
def read_root():
|
||
return {"status": "ok", "message": "欢迎使用人脸识别服务 V2"}
|