1.添加定时任务(每天凌晨四点处理redis记录的删除,存储入mysql持久化操作)
2.修复了prompt中没有用户输入信息的bug 3.优化了更新类接口代码(删除了db.add()代码)
This commit is contained in:
1 parent
de370f77ac
commit
065528f509
7 files changed
+65
-16
No files matched your search
+18
-1
@@ -1,12 +1,16 @@
|
||||
from fastapi import FastAPI, Depends
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from contextlib import asynccontextmanager
|
||||
from sqlalchemy import create_engine
|
||||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||
from apscheduler.triggers.cron import CronTrigger
|
||||
from .models import Base
|
||||
from .routes.character_route import router as character_router
|
||||
from .routes.user_route import router as user_router
|
||||
from .routes.session_route import router as session_router
|
||||
from .routes.chat_route import router as chat_router
|
||||
from .dependencies.logger import get_logger
|
||||
from .background_tasks import updating_redis_cache
|
||||
from config import get_config
|
||||
|
||||
|
||||
@@ -28,8 +32,20 @@ logger.info("数据库初始化完成")
|
||||
#------------------------------------------------------
|
||||
|
||||
|
||||
#--------------------设置定时任务-----------------------
|
||||
scheduler = AsyncIOScheduler()
|
||||
scheduler.add_job(updating_redis_cache, CronTrigger.from_crontab("0 4 * * *"))
|
||||
@asynccontextmanager
|
||||
async def lifespan(app:FastAPI):
|
||||
scheduler.start() #启动定时任务
|
||||
yield
|
||||
scheduler.shutdown() #关闭定时任务
|
||||
logger.info("定时任务初始化完成")
|
||||
#------------------------------------------------------
|
||||
|
||||
|
||||
#--------------------创建FastAPI实例--------------------
|
||||
app = FastAPI()
|
||||
app = FastAPI(lifespan=lifespan)
|
||||
logger.info("FastAPI实例创建完成")
|
||||
#------------------------------------------------------
|
||||
|
||||
@@ -42,6 +58,7 @@ app.include_router(chat_router)
|
||||
logger.info("路由初始化完成")
|
||||
#-------------------------------------------------------
|
||||
|
||||
|
||||
#-------------------设置跨域中间件-----------------------
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
|
||||
Reference in new issue
Block a user