ab工具测试flask 接口并发

ab工具安装（ubuntu）

apt-get install apache2-utils

先用python程序看请求时间（1个请求）

falsktest.py的内容可以看上一篇文章

python falsktest.py 1.png

返回时间0.1秒

使用ab工具 测试

先根据接口参数生成json文件

import json
import base64
file = "/ntt/smz_data/images/test/1.png"
with open(file, "rb") as f:
    img_read = f.read()
    base64_data = base64.b64encode(img_read)

a_dict = {'input':str(base64_data, encoding='utf-8')}
json.dump(a_dict, open('1.json', 'w'))

使用命令

 ab -n 10 -c 10  -T application/json -p 1.json  "http://127.0.0.1:5000/predict"

参数

-p 定义 post 方法需要传输的 file
file 默认是寻找 /root 路径下目标文件，可以通过 tab 确认
file 可能需要注意换行符等问题（目前我这边没有问题）
-T 是和-p 搭配使用，指定 header

返回