聊聊Jmeter如何并发执行Python脚本
来源:AirPython
作者:星安果
1. 前言
大家好,我是安果!
最近有小伙伴后台给我留言,说自己用 Django 写了一个大文件上传的 Api 接口,现在想本地检验一下接口并发的稳定性,问我有没有好的方案
本篇文章以文件上传为例,聊聊 Jmeter 并发执行 Python 脚本的完整流程2. Python 实现文件上传
大文件上传包含 3 个步骤,分别是:获取文件信息及切片数目分段切片,并上传 - API文件合并 - API文件路径参数化
2-1 获取文件信息及切片数目
首先,获取文件的大小
然后,利用预设的切片大小获取分段总数
最后,获取文件名及 md5 值
import os
import math
import hashlib
def get_file_md5(self, file_path):
"""获取文件的md5值"""
with open(file_path, "rb") as f:
data = f.read()
return hashlib.md5(data).hexdigest()
def get_filename(self, filepath):
"""获取文件原始名称"""
# 文件名带后缀
filename_with_suffix = os.path.basename(filepath)
# 文件名
filename = filename_with_suffix.split(".")[0]
# 后缀名
suffix = filename_with_suffix.split(".")[-1]
return filename_with_suffix, filename, suffix
def get_chunk_info(self, file_path):
"""获取分段信息"""
# 获取文件总大小(字节)
file_total_size = os.path.getsize(file_path)
print(file_total_size)
# 分段总数
total_chunks_num = math.ceil(file_total_size / self.chunk_size)
# 文件名(带后缀)
filename = self.get_filename(file_path)[0]
# 文件的md5值
file_md5 = self.get_file_md5(file_path)
return file_total_size, total_chunks_num, filename, file_md5
2-2 切片及分段上传
利用分段总数和分段大小,对文件进行切片,调用分段文件上传接口
import requests
def do_chunk_and_upload(self, file_path):
"""将文件分段处理,并上传"""
file_total_size, total_chunks_num, filename, file_md5 = self.get_chunk_info(file_path)
# 遍历
for index in range(total_chunks_num):
print("第{}次文件上传".format(index + 1))
if index + 1 == total_chunks_num:
partSize = file_total_size % chunk_size
else:
partSize = chunk_size
# 文件偏移量
offset = index * chunk_size
# 生成分片id,从1开始
chunk_id = index + 1
print("开始准备上传文件")
print("分片id:", chunk_id, "文件偏移量:", offset, ",当前分片大小:", partSize, )
# 分段上传文件
self.__upload(offset, chunk_id, file_path, file_md5, filename, partSize, total_chunks_num)
def __upload(self, offset, chunk_id, file_path, file_md5, filename, partSize, total):
"""分次上传文件"""
url = "http://**/file/brust/upload"
params = {"chunk": chunk_id,
"fileMD5": file_md5,
"fileName": filename,
"partSize": partSize,
"total": total
}
# 根据文件路径及偏移量,读取文件二进制数据
current_file = open(file_path, "rb")
current_file.seek(offset)
files = {"file": current_file.read(partSize)}
resp = requests.post(url, params=params, files=files).text
print(resp)
2-3 合并文件
最后调用合并文件的接口,将分段小文件合成大文件
def merge_file(self, filepath):
"""合并"""
url = "http://**/file/brust/merge"
file_total_size, total_chunks_num, filename, file_md5 = self.get_chunk_info(filepath)
payload = json.dumps(
{
"fileMD5": file_md5,
"chunkTotal": total_chunks_num,
"fileName": filename
}
)
print(payload)
headers = {
"Content-Type": "application/json"
}
resp = requests.post(url, headers=headers, data=payload).text
print(resp)
2-4 文件路径参数化
为了并发执行,将文件上传路径参数化
# fileupload.py
...
if __name__ == "__main__":
filepath = sys.argv[1]
# 每一段切片的大小(MB)
chunk_size = 2 * 1024 * 1024
fileApi = FileApi(chunk_size)
# 分段上传
fileApi.do_chunk_and_upload(filepath)
# 合并
fileApi.merge_file(filepath)3. Jmeter 并发执行
在使用 Jmeter 创建并发流程前,我们需要编写批处理脚本
其中,执行批处理脚本时,需要跟上文件路径一起执行
# cmd.bat
@echo off
set filepath=%1
python C:UsersxingagDesktoprpc_demofileupload.py %*
然后,在本地新建一个 CSV 文件,写入多个文件路径
# 准备多个文件路径(csv)
C:UsersxingagDesktopcharles-proxy-4.6.1-win64.msi
C:UsersxingagDesktopV2.0.pdf
C:UsersxingagDesktopHBuilder1.zip
C:UsersxingagDesktopHBuilder2.zip
接着,就可以使用 Jmeter 创建并发流程了
完整步骤如下:创建一个测试计划,下面添加一个线程组这里线程组数目与上面文件数目保持一致即可线程组下,添加「 同步定时器 」同步定时器中的「 模拟用户组的数量 」和上面参数数量保持一致添加 CSV 数据文件设置指向上面准备的 csv 数据文件,设置文件格式为 UTF-8,变量名称设置为 file_path,最后将线程共享模式设置为「 当前线程组 」添加调试取样器,方便调试添加 OS 进程取样器选择上面创建的批处理文件,命令行参数设置为「 ${file_path} 」添加查看结果数4. 最后
运行上面创建的 Jmeter 并发流程,在结果数中可以查看并发上传文件的结果
当然,我们可以增加并发数量去模拟真实的使用场景,只需要修改 CSV 数据源及 Jmeter 参数即可
颜值高的奇瑞新能源微小蚂蚁eQ冲击你的眼球神经微型新能源汽车产品小巧玲珑,特别适合日常通勤,尤其是女司机对其特别喜爱。在当今的自主品牌汽车中,也有不少可爱的微型代步车,这些汽车通过电力驱动,解决了加油以及上牌的烦恼,同时也满足
pyAIV64位双核AI开发板,兼容PyBoard摄像头大家好,我是小月月。今天给大家介绍一款由01Studio设计研发人工智能开发板pyAIK210。板卡介绍pyAIK210是由01Studio设计研发,基于嘉楠科技边缘计算芯片K21
2021中秋节高速公路免费吗?高速免费政策可用便签辅助记录2021年的中秋节,是很多人期待已久的节日,在这个洋溢着幸福和团圆的节日里,大家各有各的过法。很多人喜欢在中秋节带着家人一起自驾游,那么2021中秋节高速公路免费吗?高速免费政策可
2021国庆期间高速免费吗?用备忘录帮自己记的更牢固2021年的国庆节,很多人都会选择出行旅游的方式放松自己。而在出行旅游的时候,很多人都会选择驾驶自己家里的汽车去出行,这个时候就很有可能在高速公路上通行。2021国庆期间高速免费吗
2021国庆节期间加班工资怎么算?用便签软件帮自己牢记2021年的国庆节是很多人都期待的一个节日,随着国庆节一起到来的,还有7天的假期。大部分人的工作岗位在国庆节期间都是放假的,但是也有一部分人因为工作岗位的特殊性,是需要加班的。20
中秋节有三倍工资吗在便签软件里存一下中秋节是我们中华民族的一个比较重要的传统节日,每逢到了中秋节,一半都会伴随着有假期。人们可以在中秋节的假期中出去游玩,也可以利用中秋节的假期走亲访友和亲人相聚。但是也有的人是需要在
开学老师对家长说的话用便签编辑开学对于每一个正在上学的孩子来说,都是一件非常重要的事情,每逢开学的时候,老师为了能够和家长配合好,对孩子的教育起到更好的效果,都会进行一些沟通。开学老师对家长说的话,可以用便签编
开学介绍自己的一段话便签写下来记得更牢由于种种原因,今年很多地方的学习都是分批次开学,所以各地2021年的9月份开学时间都跟往年有着一些不一样的地方。进入9月一来,莘莘学子们和老师都在迎接开学季,为了给大家留下比较深刻
手机便签的东西怎样发到别人手机上?手机上有很多好用的辅助工具,便签类软件就是其中之一,在使用便签软件的时候,可以记录各种各样的东西。在工作中,有的时候,需要把自己在便签软件中记录的内容发送给别人,手机便签的东西怎样
如何改变桌面便签的颜色桌面便签是很多人平时都在用的一种辅助工具,之所以叫做桌面便签,是因为它可以在操作设备的桌面显示内容,比如手机电脑等。在使用桌面便签软件的时候,为了丰富视觉效果,如何改变它的颜色呢?
开学了怎么帮孩子收心用便签规划要做的事时间过得真快,转眼间2021年已经步入了九月份,九月份对很多学生和家长来说都是一个比较重要的月份,因为这个月会迎来开学季。为了更好的帮助自己的孩子进入学习状态,很多家长会在这个时间