工作需要，今天写了一个人员注册的压力测试程序，以前都是利用requests+threading来实现的，感觉还是有点麻烦。所以网上找了一下有没有其他轮子，果不其然，找到了 trip这个轮子，作者结合Tornado与Requests来实现。其中参考文档https://trip.readthedocs.io/zh_CN/latest/。
如下代码部分，伪数据产生用faker和random来实现。其中faker库用来造数据再好不过了，常见的接口字段都可以随机生成，顺带提一下fake2db库，是一个伪数据库生成库。
由于是测试人脸识别图片功能，需要post带人脸的图片，这里只好取巧，本地收集了一堆人脸照片到某个文件夹，利用random随机选取。其实如果只是普通图片的话，可以自己写个爬虫脚本在线随机爬取。
值得注意的是post需要传递 [{"tag": "姓名", "value": self.fake.name()}, {"tag": "身份证号", "value": self.fake.ssn()}, {"tag": "性别", "value": random.choice(["男", "女"])}] 这一堆，需要将其强制转成字符串，但是会导致其中的双引号变成单引号，不符合json格式，所以需要用replace替换回来。
# -*- coding: utf-8 -*-
import trip
import faker
import random
import os
class Regiser():
def __init__(self, num=1):
self.fake = faker.Faker("zh_CN")
self.url = "http://xxx.xxx.x.x:9180/seeta/register/enrollment"
self.base_image_url = "C:/Users/YuChou/Desktop/人员/"
self.image_href_list = os.listdir(self.base_image_url)
self.num = num # 同时注册数据量
# 生成伪数据表单
def product_formdata(self):
data_list = [{"tag": "姓名", "value": self.fake.name()},
{"tag": "身份证号", "value": self.fake.ssn()},
{"tag": "性别", "value": random.choice(["男", "女"])}]
formData = {
"secret_key": "b3a9b9bb20544f9eb0e42a5b93a73aa1",
"meeting_key": "6c16544d0a004774b5d21c85bd0322be",
"attributes": str(data_list).replace("\'", "\"") #传json 转字符串会把"转成' 需要重新替换回来 不然会报json错误
}
return formData
# 生成随机图片
def product_image(self):
files = {"image": open(self.base_image_url + random.choice(self.image_href_list), "rb")}
return files
# 并发
@trip.coroutine
def request(self):
req = yield [trip.post(url=self.url, data=self.product_formdata(), files=self.product_image()) for i in range(self.num)]
print(list(map(lambda x: x.text, req)))
# 执行协程
def main(self):
trip.run(self.request)
if __name__ == "__main__":
obj = Regiser(num=16)
obj.main()
这里设置了16个同时注册，16个总时间算下来小于2.5s，前面有试过单个注册就需要2s左右，基本实现并发功能，运行结果：
image.png
