Jmeter并發(fā)執(zhí)行Python腳本的示例分析

這篇文章將為大家詳細(xì)講解有關(guān)Jmeter并發(fā)執(zhí)行Python腳本的示例分析，小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

Python 實(shí)現(xiàn)文件上傳

大文件上傳包含 3 個(gè)步驟，分別是：

• 獲取文件信息及切片數(shù)目

• 分段切片，并上傳- API

• 文件合并- API

• 文件路徑參數(shù)化

2-1獲取文件信息及切片數(shù)目

首先，獲取文件的大小

然后，利用預(yù)設(shè)的切片大小獲取分段總數(shù)

最后，獲取文件名及 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):
    """獲取分段信息"""
    # 獲取文件總大小(字節(jié))
    file_total_size = os.path.getsize(file_path)
    print(file_total_size)

    # 分段總數(shù)
    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切片及分段上傳

利用分段總數(shù)和分段大小，對(duì)文件進(jìn)行切片，調(diào)用分段文件上傳接口

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('開始準(zhǔn)備上傳文件')
        print("分片id:", chunk_id, "文件偏移量：", offset, ",當(dāng)前分片大小:", 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
                }
    # 根據(jù)文件路徑及偏移量，讀取文件二進(jìn)制數(shù)據(jù)
    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合并文件

最后調(diào)用合并文件的接口，將分段小文件合成大文件

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文件路徑參數(shù)化

為了并發(fā)執(zhí)行，將文件上傳路徑參數(shù)化

# 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)

 Jmeter 并發(fā)執(zhí)行

在使用 Jmeter 創(chuàng)建并發(fā)流程前，我們需要編寫批處理腳本

其中，執(zhí)行批處理腳本時(shí)，需要跟上文件路徑一起執(zhí)行

# cmd.bat

@echo off
set filepath=%1

python  C:\Users\xingag\Desktop\rpc_demo\fileupload.py %*

然后，在本地新建一個(gè) CSV 文件，寫入多個(gè)文件路徑

# 準(zhǔn)備多個(gè)文件路徑（csv）
C:\\Users\\xingag\\Desktop\\charles-proxy-4.6.1-win64.msi
C:\\Users\\xingag\\Desktop\\V2.0.pdf
C:\\Users\\xingag\\Desktop\\HBuilder1.zip
C:\\Users\\xingag\\Desktop\\HBuilder2.zip

接著，就可以使用 Jmeter 創(chuàng)建并發(fā)流程了

完整步驟如下：

• 創(chuàng)建一個(gè)測(cè)試計(jì)劃，下面添加一個(gè)線程組

這里線程組數(shù)目與上面文件數(shù)目保持一致即可

• 線程組下，添加「同步定時(shí)器」

同步定時(shí)器中的「模擬用戶組的數(shù)量」和上面參數(shù)數(shù)量保持一致

• 添加 CSV 數(shù)據(jù)文件設(shè)置

指向上面準(zhǔn)備的 csv 數(shù)據(jù)文件，設(shè)置文件格式為 UTF-8，變量名稱設(shè)置為file_path，最后將線程共享模式設(shè)置為「當(dāng)前線程組」

• 添加調(diào)試取樣器，方便調(diào)試

• 添加 OS 進(jìn)程取樣器

選擇上面創(chuàng)建的批處理文件，命令行參數(shù)設(shè)置為「${file_path}」

• 添加查看結(jié)果數(shù)

 最后

運(yùn)行上面創(chuàng)建的 Jmeter 并發(fā)流程，在結(jié)果數(shù)中可以查看并發(fā)上傳文件的結(jié)果

當(dāng)然，我們可以增加并發(fā)數(shù)量去模擬真實(shí)的使用場(chǎng)景，只需要修改 CSV 數(shù)據(jù)源及 Jmeter 參數(shù)即可

關(guān)于“Jmeter并發(fā)執(zhí)行Python腳本的示例分析”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。