Python怎么自动化测试Flask接口并发性能_使用Locust编写压测脚本与并发模拟

秋静姑娘_6714

秋静姑娘_6714

2026-05-19

657人浏览

原创

Locust压测Flask需将Flask作为独立服务调用,用HttpUser发起HTTP请求;合理配置--users和--spawn-rate以匹配WSGI服务器能力;手动检查JSON响应并调用response.failure标记业务失败;通过X-Request-ID和统一时间格式关联两端日志。

python怎么自动化测试flask接口并发性能_使用locust编写压测脚本与并发模拟

Locust怎么写Flask压测脚本:从启动到定义任务

Locust 不是 Flask 插件,它独立运行,通过 HTTP 请求模拟用户行为。关键在于把 Flask 应用当作普通 Web 服务来调用,而不是在 Locust 内部启动 Flask 实例(否则会干扰并发模型)。

  • locustfile.py 中直接使用 HttpUser 类,不 import Flask 或调用 app.run()
  • 所有接口地址必须指向已启动的 Flask 服务,例如 @#@#@#@#@#@#@#@#@#@0
  • 每个请求用 self.client.get()、self.client.post() 发起,Locust 自动管理连接池和超时
  • 如果 Flask 接口需要认证,可在 on_start() 中统一设置 self.client.headers
from locust import HttpUser, task, between
<p>class ApiUser(HttpUser):
wait_time = between(1, 3)</p><pre class="brush:php;toolbar:false;">def on_start(self):
    self.client.headers.update({"Authorization": "Bearer xyz"})

@task
def get_user_list(self):
    self.client.get("/api/users")

并发数、用户数和 spawn rate 怎么配才真实

Locust 的 --users(总用户数)、--spawn-rate(每秒启动用户数)不是简单“开越多越好”,它们和 Flask 的实际处理能力、Gunicorn/Uvicorn 工作进程数强相关。

  • 若 Flask 用默认开发服务器(单线程),--users 10 就可能排队阻塞,压测结果反映的是 WSGI 服务器瓶颈,而非业务逻辑
  • 生产部署常用 Gunicorn(如 -w 4 -k gevent),此时 --users 应 ≥ worker 数 × 并发连接上限(gevent 默认约 1000),但需配合 --spawn-rate 缓慢加压,避免瞬间打满连接
  • 常见误配:--users 1000 --spawn-rate 1000 → 所有用户 1 秒内启动,TCP 连接激增,可能触发本地端口耗尽或目标服务拒绝连接(ConnectionRefusedError)

怎么捕获 Flask 返回的 JSON 错误并计入失败统计

Locust 默认只将 HTTP 状态码 ≥ 400 视为失败,但 Flask 常返回 200 OK + JSON body 包含 {"error": "xxx"}。这类业务错误需手动标记失败,否则压测报告里的“成功率”严重失真。

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
  • 在请求后检查响应体,用 response.json() 解析,再判断字段
  • 调用 response.failure("reason") 主动上报失败,Locust 会统计进 Errors 表
  • 注意异常捕获:若响应非 JSON(如 502 Bad Gateway),response.json() 会抛 JSONDecodeError,需用 try/except 包裹
@task
def create_order(self):
    with self.client.post("/api/orders", json={"item_id": 123}, catch_response=True) as response:
        try:
            data = response.json()
            if data.get("code") != 0:
                response.failure(f"API error code: {data.get('code')}")
        except Exception as e:
            response.failure(f"Invalid JSON: {e}")

Flask 日志和 Locust 报告对不上?时间戳和请求 ID 怎么关联

Locust 压测时看不到 Flask 侧具体哪次请求对应哪个用户行为,尤其出错时难定位。核心办法是让两者共享唯一标识,并统一时间精度。

  • Flask 端在请求进入时生成 request id(如用 uuid.uuid4().hex[:8]),写入日志前缀,并通过响应头回传(如 X-Request-ID)
  • Locust 端在请求头中带上自定义 ID(self.client.get(..., headers={"X-Request-ID": "abc123"})),并在 catch_response=True 块中读取响应头做日志关联
  • Flask 默认日志时间是 time.time()(秒级),Locust 统计精确到毫秒,建议 Flask 日志格式中启用 %(asctime)s.%(msecs)03d,避免时间对齐偏差

压测脚本真正难的不是写几个 @task,而是让 Locust 的并发模型和 Flask 的部署形态对齐——比如用了 gevent 却没关掉 Flask 的调试模式,或者忽略了 Werkzeug 开发服务器根本不能扛并发。这些细节不调平,数据就只是数字游戏。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

数码产品性能查询
数码产品性能查询

该软件包括了市面上所有手机CPU,手机跑分情况,电脑CPU,电脑产品信息等等,方便需要大家查阅数码产品最新情况,了解产品特性,能够进行对比选择最具性价比的商品。

下载

相关标签:

python

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1851

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

5024

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1809

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

30177

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3587

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3607

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1243

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2803

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习