FastAPI（41）- Background Task 后台任务

后台任务

• 顾名思义，可以在返回响应后运行后台任务
• 这对于需要在请求后执行特定的操作很有用，且客户端并不需要在接收响应之前等待该操作完成

常见的栗子

• 发送电子邮件通知，由于连接到电子邮件服务器并发送电子邮件往往会比较“缓慢”（几秒钟），因此可以立即返回响应并在后台发送电子邮件通知
• 假设您到一个必须经过缓慢处理的文件，可以先返回“已接受”（HTTP 202）响应并在后台处理它

实际栗子

创建后台任务要用到的函数

• 创建一个作为后台任务运行的函数，就是一个普通函数
• 可以加 async 也可以不加，FastAPI 将会正确处理它

import time

def write_notification(email: str, message: str = ""):
    # 1、模拟和邮件服务器建立连接
    time.sleep(3)
    with open("text.txt", mode="w") as f:
        # 2、模拟发送邮件
        content = f"message is {message}"
        f.write(content)
        print(content)

  

添加后台任务

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠萝测试笔记
# blog:  https://www.cnblogs.com/poloyy/
# time: 2021/9/29 7:11 下午
# file: 35_background_task.py
"""
import time
import uvicorn
from fastapi import FastAPI, BackgroundTasks

app = FastAPI()


def write_notification(email: str, message: str = ""):
    # 1、模拟和邮件服务器建立连接
    time.sleep(3)
    with open("text.txt", mode="w") as f:
        # 2、模拟发送邮件
        content = f"message is {message}"
        f.write(content)
        print(content)


@app.post("/email/{email}")
async def send_email(
        email: str,
        # 指定参数类型为 BackgroundTasks
        background_task: BackgroundTasks
):
    # 添加后台任务
    # 第一个参数：可调用对象，一般就是函数
    # 后面的参数：write_notification 函数所需要的参数
    background_task.add_task(write_notification, email, message="test_message")
    return {"message": "Notification sent in the background"}


if __name__ == '__main__':
    uvicorn.run(app="35_background_task:app", reload=True, host="127.0.0.1", port=8080)

  

后台任务结合依赖项

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠萝测试笔记
# blog:  https://www.cnblogs.com/poloyy/
# time: 2021/9/29 7:11 下午
# file: 35_background_task.py
"""
import time
from typing import Optional

import uvicorn
from fastapi import FastAPI, BackgroundTasks, Depends

app = FastAPI()


# 后台任务函数
def write_log(message: str):
    with open("log.txt", mode="a") as log:
        log.write(message)


# 依赖项函数
async def get_query(
        background_task: BackgroundTasks,
        q: Optional[str] = None,
):
    # 如果 q 有值才执行后台任务
    if q:
        message = f"found query: {q}
"
        background_task.add_task(write_log, message)


@app.post("/email_depends/{email}")
async def send_email(
        email: str,
        background_task: BackgroundTasks,
        q: str = Depends(get_query)
):
    # 执行一次后台任务
    message = f"message to {email}
"
    background_task.add_task(write_log, message)
    return {"message": "Message sent"}


if __name__ == '__main__':
    uvicorn.run(app="35_background_task:app", reload=True, host="127.0.0.1", port=8080)

• 后台任务可以在任意地方使用，比如路径操作、依赖项、子依赖项...
• FastAPI 会将所有后台任务合并在一起，然后在后台会按 add_task 的顺序运行

查看 BackgroundTasks 源码

• BackgroundTasks 是继承 BackgroundTask，而 BackgroundTask 是直接来自 starlette.background
• add_task() 第一个参数 func 类型是 Callable，可调用对象，一般传函数就好啦
• 内部会声明一个 BackgroundTask 对象，自动调用它的 __call__ 方法
• 可以看到，最终会执行 func()
• func() 函数参数就是 add_task() 函数除第一个参数以外的参数

BackgroundTasks 注意事项

• 如果需要执行繁重的后台计算，且可能需要多个进程运行（例如，不需要共享内存、变量等），使用其他更大的工具，如：Celery，效果可能会更好
• 它们往往需要更复杂的配置、消息/作业队列管理器，如 RabbitMQ 或 Redis，它们允许在多个进程中运行后台任务，尤其是在多个服务器中
• 但是，如果需要从同一个 FastAPI 应用程序访问变量和对象，或者需要执行小型后台任务（例如发送电子邮件通知），只需使用 BackgroundTasks