Python FastAPI：如何只接收一次请求
FastAPI 是一个现代、快速（高性能）的 Web 框架，用于构建 API。在某些情况下，我们可能需要确保 API 只接收一次请求，例如在处理某些关键任务时，以避免重复执行相同的操作。本文将介绍如何在 FastAPI 中实现这一点。
问题背景
假设我们有一个 API 端点，用于处理用户注册。我们希望确保每个用户只能注册一次，避免重复注册。这就需要我们在 API 中实现一种机制，以确保只接收一次请求。
实现方法
我们可以通过以下步骤来实现只接收一次请求的功能：
- 使用全局变量或数据库来存储已经处理过的请求。
- 在接收请求之前，检查请求是否已经被处理过。
- 如果请求已经被处理过，返回相应的响应，否则继续处理请求。
示例代码
以下是一个简单的示例，演示如何在 FastAPI 中实现只接收一次请求的功能：
from fastapi import FastAPI, HTTPException
from typing import Set
app = FastAPI()
# 使用集合存储已经处理过的请求 ID
processed_requests: Set[str] = set()
@app.post("/register")
async def register_user(request_id: str):
# 检查请求是否已经被处理过
if request_id in processed_requests:
raise HTTPException(status_code=400, detail="Request already processed")
# 处理请求
# 这里可以添加实际的注册逻辑
# 将请求 ID 添加到已处理的集合中
processed_requests.add(request_id)
return {"message": "User registered successfully"}
在这个示例中，我们使用了一个全局集合 processed_requests 来存储已经处理过的请求 ID。在 register_user 函数中，我们首先检查请求 ID 是否已经在 processed_requests 中。如果已经存在，我们抛出一个 HTTP 400 异常，表示请求已经被处理过。否则，我们继续处理请求，并将请求 ID 添加到 processed_requests 中。
注意事项
- 在实际应用中，可能需要使用数据库或其他持久化存储来替代全局变量，以支持更高的并发和持久性。
- 根据具体需求，可以对请求 ID 的生成和验证进行更严格的控制，以确保请求的唯一性和安全性。
- 在处理请求时，应确保遵循幂等性原则，即使多次接收到相同的请求，也不会产生意外的副作用。
结语
通过使用全局变量或数据库来存储已经处理过的请求，并在接收请求之前进行检查，我们可以在 FastAPI 中实现只接收一次请求的功能。这在处理关键任务或避免重复操作时非常有用。希望本文的示例和说明能够帮助你更好地理解和应用这一功能。
