21 lines
645 B
Python
21 lines
645 B
Python
|
from app.middleware.cors import add_cors_middleware
|
||
|
|
from app.routes import websocket, stream
|
||
|
|
from app.tasks.cleanup_task import cleanup_streams
|
||
|
|
from fastapi import FastAPI
|
||
|
|
from contextlib import asynccontextmanager
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
@asynccontextmanager
|
||
|
|
async def lifespan(app: FastAPI):
|
||
|
|
print("正在启动服务...")
|
||
|
|
# 启动定时清理任务
|
||
|
|
asyncio.create_task(cleanup_streams())
|
||
|
|
yield # FastAPI 会在这里执行应用关闭时的清理任务
|
||
|
|
print("停止服务")
|
||
|
|
|
||
|
|
app = FastAPI(lifespan=lifespan)
|
||
|
|
# 添加跨域中间件
|
||
|
|
add_cors_middleware(app)
|
||
|
|
app.include_router(websocket.router)
|
||
|
|
app.include_router(stream.router)
|