FastAPI BackgroundTasks简易使用和实现逻辑

最新推荐文章于 2025-01-16 13:04:49 发布

Finger_ebic

最新推荐文章于 2025-01-16 13:04:49 发布

阅读量955

点赞数 3

文章标签： fastapi python 开发语言

本文链接：https://blog.csdn.net/KQe397773106/article/details/137199489

版权

本文介绍了如何在FastAPI应用中利用BackgroundTasks和anyio库实现异步后台任务，如发送电子邮件通知。通过`/`路由将任务添加到后台并在`response.py`中的`image.pngResponse`类的`__call__`方法中使用anyio并发执行这些任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import uvicorn
from fastapi import FastAPI,BackgroundTasks
import asyncio
import time
import datetime

app = FastAPI()


def write_notification(email: str, message: str = ""):
    # 1、模拟和邮件服务器建立连接
    time.sleep(3)
    with open("text.txt", mode="a+") as f:
        # 2、模拟发送邮件
        content = f"message is {message}"+datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")+"\n"
        f.write(content)
        print(content)

@app.get("/")
async def root(background_task: BackgroundTasks):
    background_task.add_task(write_notification, "email", message="test_message")
    return {"message": "Hello, World!"}

def run_server():
    uvicorn.run(app, host="0.0.0.0", port=8000)

if __name__ == "__main__":
    run_server()

Fastapi BackgroundTasks实现

在starlette的response.py里

image.png
Response类的__call__方法，会调用self.background
self.background是一个BackgroundTasks类
这个类有一个add_task方法，就是官方文档里的把后台函数添加到BackgroundTasks里的
然后, 这个__call__方法循环取出每一个task

最后在这里真正的用心的线程来跑后台任务
用的是anyio来起的线程