fix 数据库会话管理

app/api/endpoints/webhook.py

@@ -1,24 +1,27 @@
 from typing import Any
 
-from fastapi import APIRouter, BackgroundTasks, Request
+from fastapi import APIRouter, BackgroundTasks, Request, Depends
+from sqlalchemy.orm import Session
 
 from app import schemas
 from app.chain.webhook import WebhookChain
 from app.core.config import settings
+from app.db import get_db
 
 router = APIRouter()
 
 
-def start_webhook_chain(body: Any, form: Any, args: Any):
+def start_webhook_chain(db: Session, body: Any, form: Any, args: Any):
     """
     启动链式任务
     """
-    WebhookChain().message(body=body, form=form, args=args)
+    WebhookChain(db).message(body=body, form=form, args=args)
 
 
 @router.post("/", summary="Webhook消息响应", response_model=schemas.Response)
 async def webhook_message(background_tasks: BackgroundTasks,
-                          token: str, request: Request) -> Any:
+                          token: str, request: Request,
+                          db: Session = Depends(get_db),) -> Any:
     """
     Webhook响应
     """
@@ -27,18 +30,19 @@ async def webhook_message(background_tasks: BackgroundTasks,
     body = await request.body()
     form = await request.form()
     args = request.query_params
-    background_tasks.add_task(start_webhook_chain, body, form, args)
+    background_tasks.add_task(start_webhook_chain, db, body, form, args)
     return schemas.Response(success=True)
 
 
 @router.get("/", summary="Webhook消息响应", response_model=schemas.Response)
 async def webhook_message(background_tasks: BackgroundTasks,
-                          token: str, request: Request) -> Any:
+                          token: str, request: Request,
+                          db: Session = Depends(get_db)) -> Any:
     """
     Webhook响应
     """
     if token != settings.API_TOKEN:
         return schemas.Response(success=False, message="token认证不通过")
     args = request.query_params
-    background_tasks.add_task(start_webhook_chain, None, None, args)
+    background_tasks.add_task(start_webhook_chain, db, None, None, args)
     return schemas.Response(success=True)