FastAPI压力测试概述

FastAPI作为高性能Python框架，压力测试需关注并发处理能力、响应时间及资源利用率。Locust作为开源负载测试工具，能模拟真实用户行为，通过编写Python脚本定义用户场景。

Locust测试环境搭建

安装Locust最新版本：

pip install locust

创建测试脚本（如locustfile.py），导入FastAPI应用端点。示例脚本模拟用户访问API：

from locust import HttpUser, task, between

class FastAPIUser(HttpUser):
    wait_time = between(1, 3)  # 用户请求间隔时间（秒）

    @task
    def get_items(self):
        self.client.get("/items/1")

    @task(3)  # 权重为3，更高调用频率
    def create_item(self):
        self.client.post("/items/", json={"name": "test"})

并发测试执行

启动Locust服务并指定并发参数：

locust -f locustfile.py --host=http://localhost:8000

通过Web界面（默认http://localhost:8089）设置并发用户数（如1000）、每秒启动速率（如100），实时监控RPS（每秒请求数）、响应时间及失败率。

关键优化建议

数据库连接池配置
FastAPI依赖数据库时（如SQLAlchemy），确保连接池大小适配并发量：

SQLALCHEMY_DATABASE_URL = "postgresql://user:pass@localhost/db"
engine = create_engine(SQLALCHEMY_DATABASE_URL, pool_size=20, max_overflow=10)

异步端点设计
对IO密集型操作使用async/await，避免阻塞事件循环：

@app.get("/items/{item_id}")
async def read_item(item_id: int):
    return await fetch_from_db(item_id)

静态文件与中间件优化
使用StaticFiles处理静态资源，精简中间件链：

app.mount("/static", StaticFiles(directory="static"), name="static")

负载均衡与水平扩展
通过Nginx反向代理多实例，利用UVicorn工作进程：

uvicorn main:app --workers 4 --host 0.0.0.0 --port 8000

监控与结果分析

Locust测试报告需关注：

• 95%响应时间（应低于500ms）
• 错误率（目标<1%）
• 系统资源（CPU/RAM）使用率
  结合Prometheus+Grafana实现实时监控，定位瓶颈如慢查询或内存泄漏。

高级场景模拟

Locust支持CSV数据驱动测试，模拟差异化用户行为：

import csv

class DynamicUser(HttpUser):
    @task
    def login(self):
        with open("users.csv") as f:
            for row in csv.DictReader(f):
                self.client.post("/login", json=row)

通过上述方法，可系统化验证FastAPI应用的稳定性与扩展性。