feat(task): 后台任务改为apscheduler执行

core/routers/auth/services.py

@@ -87,7 +87,7 @@ async def get_current_user_form_ws(
     """在ws headers中提取token，并验证再获取当前用户"""
     auth_header = websocket.headers.get("authorization") or websocket.headers.get("Authorization")
     if auth_header and auth_header.lower().startswith("bearer "):
-        token = auth_header.removeprefix("Bearer ").strip()
+        token = auth_header.removeprefix("bearer ").strip()
     else:
         token = ''
     user = await get_current_user(token, db_session, redis_conn)

core/routers/card/actions.py

@@ -7,8 +7,11 @@
 # @Desc    :
 import logging
 
+from sqlalchemy.ext.asyncio import AsyncSession
+
 from core.base.schema import ProductCategory
 from core.db.crud import CRUD
+from core.db.engine import async_db_session
 from core.db.models import User, Product
 from core.routers.auth.schema import LoginUser
 from core.routers.pay.actions import actions_handler
@@ -16,11 +19,13 @@ from core.routers.pay.actions import actions_handler
 logger = logging.getLogger(__name__)
 
 
+@async_db_session()
 async def reward_points_after_uploading_pictures(
-        db_crud: CRUD,
         gift_product_id: int,
-        login_user: LoginUser
+        login_user: LoginUser,
+        db_session: AsyncSession
 ):
+    db_crud = CRUD(db_session, Product)
     product: Product = await db_crud.get(
         model=Product, filters={
             'id': gift_product_id,

core/routers/card/routes.py

@@ -8,6 +8,7 @@ import logging
 from typing import List, Literal
 
 import ujson
+from asyncer import asyncify
 from fastapi import APIRouter, Depends, Query, HTTPException, status
 import redis.asyncio as aioredis
 from sqlalchemy import select, func
@@ -15,6 +16,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 
 from config import settings
 from core.base import schema as base_schema
+from core.base.scheduler import add_task
 from core.db.crud import CRUD
 from core.db.engine import get_async_session, get_cache
 from core.db.models import Card, SearchCardCache, UserCollection, User
@@ -243,11 +245,12 @@ async def created_card(
             )
             # 上传图片奖励
             if payload.reward and settings.UPLOAD_REWARD_POINTS_GIFT_ID > 0:
-                await reward_points_after_uploading_pictures(
-                    db_crud, gift_product_id=settings.UPLOAD_REWARD_POINTS_GIFT_ID, login_user=login_user
+                await asyncify(add_task)(
+                    reward_points_after_uploading_pictures,
+                    gift_product_id=settings.UPLOAD_REWARD_POINTS_GIFT_ID, login_user=login_user,
+                    task_id=f'{login_user.user_id}-{card.card_number}'
                 )
 
-
     else:
         if card is None:
             card: Card = await db_crud.create(

core/routers/pay/actions.py

@@ -11,11 +11,13 @@ from sqlalchemy.ext.asyncio import AsyncSession
 
 from core.base.schema import ProductCategory, ProductAction, OrderStatusType
 from core.db.crud import CRUD
+from core.db.engine import async_db_session
 from core.db.models import User, Order, Product, PointsTransaction
 
 logger = logging.getLogger(__name__)
 
 
+@async_db_session()
 async def actions_after_payment(
         notify_params: dict,
         db_session: AsyncSession

core/routers/pay/routes.py

@@ -8,10 +8,12 @@ import logging
 from xml.parsers.expat import ExpatError
 
 import xmltodict
-from fastapi import APIRouter, Request, HTTPException, status, Depends, BackgroundTasks
+from asyncer import asyncify
+from fastapi import APIRouter, Request, HTTPException, status, Depends
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from core.base import schema as base_schema
+from core.base.scheduler import add_task
 from core.db.crud import CRUD
 from core.db.engine import get_async_session
 from core.db.models import Order, Product
@@ -94,7 +96,6 @@ async def create_order(
 @pay_router.post('/wechat/notify', summary='处理支付结果通知', response_model=base_schema.BaseResponse)
 async def wechat_notify(
         request: Request,
-        background_tasks: BackgroundTasks,
         db_session: AsyncSession = Depends(get_async_session)
 ):
     body = await request.body()
@@ -129,7 +130,11 @@ async def wechat_notify(
             'status': base_schema.OrderStatusType.PAID
         })
         result = {'status': 200, 'message': 'OK'}
-        background_tasks.add_task(actions_after_payment, notify_params=params, db_session=db_session)
+        await asyncify(add_task)(
+            actions_after_payment,
+            notify_params=params,
+            task_id=record['order_id'],
+        )
     else:
         result = {'status': 400, 'message': '支付失败'}
         record.update({