使用FastAPI框架构建异步任务队列

本示例展示如何使用FastAPI框架构建一个基于Redis的异步任务队列。当有新任务添加到队列中时，服务器会立即返回一个任务ID，任务会在后台被异步执行，执行结果可以在后续查询任务ID时获取到。通过这种方式，能够实现前后端分离，提高服务器响应速度。

from fastapi import FastAPI
from rq import Queue
from redis import Redis

app = FastAPI()

# 创建Redis连接池
redis_conn = Redis(host="localhost", port=6379)

# 创建任务队列
task_queue = Queue("my_queue", connection=redis_conn)


# 定义异步任务
def async_task():
    pass  # do some stuff


# 添加任务到队列中
@app.post("/tasks/")
async def add_task():
    task = task_queue.enqueue(async_task)
    return {"task_id": task.id}


# 获取任务结果
@app.get("/tasks/{task_id}/")
async def get_task(task_id: str):
    task = task_queue.fetch_job(task_id)
    if task:
        return {"status": task.get_status(), "result": task.result}
    else:
        return {"status": "unknown"}

免责申明：文章和图片全部来源于公开网络，如有侵权，请通知删除 server@dude6.com