44 lines
1000 B
Python
44 lines
1000 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
|
|
from linxyun.core import Linxyun
|
|
from linxyun.config import Config, Upload
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
# 启动定时清理任务
|
|
asyncio.create_task(cleanup_streams())
|
|
yield # FastAPI 会在这里执行应用关闭时的清理任务
|
|
print("停止服务")
|
|
|
|
# 创建 config
|
|
config = Config(
|
|
entCode="57",
|
|
project="smartroadlamp",
|
|
role={
|
|
"1101": ["/**"],
|
|
"1102": ["/**"]
|
|
},
|
|
white_list=[
|
|
"/**"
|
|
],
|
|
upload=Upload(img_quality=0.5)
|
|
)
|
|
|
|
linxyun = Linxyun(config)
|
|
|
|
app = FastAPI(lifespan=lifespan)
|
|
# 添加跨域中间件
|
|
add_cors_middleware(app)
|
|
# 添加鉴权中间件
|
|
linxyun.add_security_middleware(app)
|
|
|
|
app.include_router(websocket.router)
|
|
app.include_router(stream.router)
|
|
|
|
|
|
|