FastAPI集成APsecheduler的BackgroundScheduler+mongodb(精简)

项目架构:

FastAPI(folder)

>app(folder)

>core(folder)

>models(folder)

>routers(folder)

>utils(folder)

main.py(file)

1 utils文件夹下新建schedulers.py

from apscheduler.schedulers.background import BackgroundScheduler

from apscheduler.jobstores.mongodb import MongoDBJobStore

#特殊说明:此处显示指定DB,会在DB里创建数据库apscheduler_db

jobstores={

'default':MongoDBJobStore(

database='apscheduler_db',

collection='custom_jobs',

host='localhost',

port=27017

)

}

#特殊说明:replace_existing=True会覆盖同名的JOB,但不影响数据库中的,仅处理job_id相同的冲突

scheduler=BackgroundScheduler(jobstores=jobstores,replace_existing=True)

2 main.py中在lifespan上下文初始化和关闭scheduler

import uvicorn

from contextlib import asynccontextmanager

from app.utils.schedulers import jobstores

scheduler=None

#特殊说明:yield中可以监控到正常结束比如ctrl+c,异常结束不能执行yield后代码

@asynccontextmanager

async def lifespan(app:FastAPI):

jobstores'default'.remove_all_jobs()

from app.utils.schedulers import scheduler

yield

scheduler.remove_all_jobs()

scheduler.shutdown(wait=False)

app=FastAPI(lifespan=lifespan)

3 models文件夹新建scheduler.py文件配置基础参数类和默认值

from pydantic import BaseModel

class job_config(BaseModel):

job_id:str="default"

job_name:str="default"

trigger_type:str="interval"

trigger_kwargs:dict={}

seconds:int=30

pass

4 api文件夹下添加sechedulers.py配置添加创建job方法

from app.utils.schedulers import scheduler

from app.models.schedulers import job_config

from fastapi import APIRouter

from typing import Coroutine,Callable

from datetime import datetime

@router.get("create_job")

def create_job(jobconfig,func):

try:

if jobconfig is None:

jobconfig=job_config()

scheduler.add_job(

func,

trigger=jobconfig.trigger_type,

kwargs=jobconfig.trigger_kwargs,#特殊说明:这里可以添加自定义参数

id=jobconfig.job_id,

name=jobconfig.job_name,

seconds=jobconfig.seconds

)

scheduler.start()

except Exception as e:

raise e

except (KeyboardInterrupt, SystemExit):

scheduler.shutdown()

5 测试,调用test方法

@router.get("/function")

def function1():

try:

with open("D:\\demo.txt", "a") as file:

print("写入文件"+ str(datetime.now()), file=file)

except:

pass

@router.get("/test")

def test():

try:

create_job(None,function1)

except:

pass

相关推荐
Json____10 小时前
用 FastAPI + Vue3 打造一套完整的在线考试系统:从组卷到阅卷的业务闭环实践
后端·node.js·fastapi·it学习·wwwoop.com
dyxal15 小时前
A2A 协议 v1.0 完全指南:方法名、Agent Card、三种绑定与 FastAPI 实战
开发语言·qt·fastapi
vHelios16 小时前
Python FastAPI 422 Unprocessable Entity报错排查实录 error:Invalid \\escape
fastapi
溪语流沙21 小时前
【Web全栈进阶】Nginx反向代理 + HTTPS:正式上线
前端·vue.js·python·nginx·https·fastapi
肖永威1 天前
python版微电网仿真工具的前端托管选型:从http.server到FastAPI StaticFiles
前端·python·http·fastapi
七点半.1 天前
FastAPI · SSE · async/await
fastapi·sse
张张123y3 天前
后端第十一课:RAG 落地实战(从 0 到 1 接上私有知识库,并设计一个好用的 RAG)
开发语言·人工智能·python·机器学习·fastapi·rag
云水初3 天前
【项目篇】基于LanGraph的自动处理客服邮件的 Agent汇总
开发语言·笔记·python·学习·fastapi·langgraph
溪语流沙3 天前
【Web全栈进阶】文件上传与图片展示:早报站的第一张图片
前端·vue.js·python·fastapi
今年下半年4 天前
AI智能体项目实战记录(02):初始化 Python 项目
开发语言·人工智能·python·fastapi