性能工具之Jmeter脚本python启动

高楼Zee

发布于 2019-12-24 11:25:37

1.5K0

发布于 2019-12-24 11:25:37

文章被收录于专栏：7DGroup7DGroup

背景

上一节是shell命令启动Jmeter,担心大家对shell脚本语法不是很熟悉，如果定制自己想要的恐怕不好弄，这次改用python启动脚本，在改造之前大家先了解下Jmeter线程组相关参数，这样对大家改造脚本有一定帮助。

知识点

python中os、sys、file模块函数
Jmeter 执行方式【jmeter -n -t [jmx file] -l [results file] -e -o [Path to web report folder]】

以下是线程组样式

解释

Name ：Thread Group 线程组的名称，可以根据项目名称定义
Comments ：备注信息，描述该计划的目的

Action to be taken after a Sampler error

假如请求失败后，接下来执行什么动作

continue 继续执行接下来的操作
start Next ThreadLoop 执行下一个线程循环
Stop Thead 停止该该线程，不在执行该线程的任何操作
Stop Test 等待该线程的采样结束后，结束整个测试，不会立即停止
Stop TestNow 停止整个测试计划

Thread Properites ：线程属性

Number of Threads(users) ：线程数，也就虚拟用户数
Ramp-uo Period(in seconds) ：控制虚拟用户启动时间
Loop Count：控制执行次数。Infinite 假如勾选表示无线执行，它主要与后面的 Duration 配合使用
Same user on each iteration : 每次迭代使用相同的用户
Delay Thread creation until needed 将线程创建延迟到需要的时候
Number of Threads * Loop Count 就是 Sampler 执行的次数，可以在 View Results Tree 中看到。

specify thread lifetime: 设置执行时间

Duration(seconnds): 请求的执行时间,它与（Loop Count 的 Infinite）配合使用
Startup delay(seconds)：延迟时间，工作中很少用。

python启动Jmeter脚本

python基础知识复习

os.sep: 取代操作系统特定的路径分隔符
os.name: 指示你正在使用的工作平台。
os.getcwd: 得到当前工作目录，即当前python脚本工作的目录路径。
os.getenv()和os.putenv: 分别用来读取和设置环境变量
os.listdir(): 返回指定目录下的所有文件和目录名
os.remove(file): 删除一个文件
os.stat（file）: 获得文件属性
os.chmod(file): 修改文件权限和时间戳
os.mkdir(name): 创建目录
os.rmdir(name): 删除目录
os.removedirs: 删除多个目录
os.system(): 运行shell命令
os.exit(): 终止当前进程
os.linesep: 给出当前平台的行终止符。
os.path.split(): 返回一个路径的目录名和文件名
os.path.isfile()和os.path.isdir()分别检验给出的路径是一个目录还是文件
os.path.existe(): 检验给出的路径是否真的存在
os.listdir(dirname): 列出dirname下的目录和文件
os.getcwd(): 获得当前工作目录
os.curdir: 返回当前目录（'.'）
os.chdir(dirname): 改变工作目录到dirname
os.path.isdir(name): 判断name是不是目录，不是目录就返回false
os.path.isfile(name): 判断name这个文件是否存在，不存在返回false
os.path.exists(name): 判断是否存在文件或目录name
os.path.getsize(name): 获得文件大小，如果name是目录返回0L
os.path.abspath(name): 获得绝对路径
os.path.isabs(): 判断是否为绝对路径
os.path.normpath(path): 规范path字符串形式
os.path.split(name): 分割文件名与目录
os.path.splitext(): 分离文件名和扩展名
os.path.join(path,name): 连接目录与文件名或目录
os.path.basename(path): 返回文件名
os.path.dirname(path): 返回文件路径

打开文件函数复习

with open( 文件名, '操作类型' ) as f: f.read()
open(文件名，操作类型) f.readlines()

读取方式有

rand()
readline()
readlines()
seek
....等等

有上面基础知识，现在开始分析python启动Jmeter脚本

获取脚本路径，通过命令打开文件
替换jmeter脚本中的线程数，步长、执行时间内容，并且重命名脚本
通过命令行执行jmeter脚本

脚本参考代码

# -*- coding: utf-8 -*-
# @Time    : 2019/12/11 22:07
# @Author  : liwen406
# @Email   :  Jmeter启动脚本
# @File    : startJmeter.py
import os, sys


def jmeterNumber(caseName, num_threads, ramp_time, duration, remark, hostIps='127.0.0.1'):
    '''
    :param caseName: 脚本名字
    :param num_threads: 线程数
    :param ramp_time: 控制线程步长
    :param duration: 执行时间
    :param remark: 标志
    :param hostIps: 负载参数
    :return: 启动JMeter成功
    '''
    if caseName is None:
        return "测试用例为空"
    if num_threads is None:
        return "虚拟并发数为空"
    if ramp_time is None:
        return "测试步骤为空"
    if duration is None:
        return "执行时间为空"
    # 执行脚本名字
    runJmeterFile = '%s_%s_%s_%s_%s' % (caseName, num_threads, ramp_time, duration, remark)
    print("执行名字脚本：%s" % runJmeterFile)
    thisdir = os.getcwd()
    # 原始脚本
    newdir = os.path.join(thisdir, "testscript", caseName + ".jmx")
    print("当前脚本路径: %s" % newdir)
    if not os.path.exists(newdir):
        print('脚本不存在！请检查脚本')
        return False
    # 保存测试结果路径
    resultFile = os.path.join(thisdir, 'result', runJmeterFile)
    print("脚本执行路径: ", resultFile)
    # 判断结果路径是否存在
    if not os.path.exists(resultFile):
        os.makedirs(resultFile)
    lines = open(newdir, encoding="utf-8").readlines()
    fp = open(os.path.join(thisdir, "result", resultFile, runJmeterFile) + '.jmx', 'w')  # 打开你要写得文件
    for s in lines:
        fp.write(s.replace('num_threads">1</stringProp>', 'num_threads">%s</stringProp>' % num_threads)  # 替换并发数
                 .replace('ramp_time">1</stringProp>', 'ramp_time">%s</stringProp>' % ramp_time)  # 替换步长
                 .replace('scheduler">false</boolProp>', 'scheduler">true</boolProp>')  # 勾选通过时间判断结束
                 .replace('duration"></stringProp>', 'duration">%s</stringProp>' % duration)  # 替换执行时间
                 .replace('name="LoopController.loops">1</stringProp>',
                          'name="LoopController.loops">-1</stringProp>'))  # 勾选通过时间判断结束
    fp.close()
    os.chdir(resultFile)
    print("当前路径: ", os.getcwd())
    # 检查环境变量
    if isEvn():
        # 判断分布式执行方式
        if len(hostIps.split(",")) > 2:
            # 根据自己需求添加执行类型
            Rcmd = 'jmeter -n -t %s.jmx -R %s -l %s.jtl -j %s.log' % (
                runJmeterFile, hostIps, runJmeterFile, runJmeterFile)
            # Rcmd = 'jmeter -n -t %s.jmx -R %s -l %s.jtl -j %s.log -e -o %s' % (runJmeterFile, hostIps, runJmeterFile, runJmeterFile, runJmeterFile)
            print('执行命令：%s' % Rcmd)
            # os.system(Rcmd)
        else:
            # 不生成html报告
            # cmd = 'jmeter -n -t %s.jmx -l %s.jtl -j %s.log' % (runJmeterFile, runJmeterFile, runJmeterFile, runJmeterFile)
            # 自动生成html报表
            cmd = 'jmeter  -n -t %s.jmx -l %s.jtl -j %s.log -e -o %s' % (
                runJmeterFile, runJmeterFile, runJmeterFile, runJmeterFile)
            print('执行命令：%s' % cmd)
            os.system(cmd)


def isEvn():
    '''
    检查环境变量
    :return: True/Fals
    '''
    cmd = 'jmeter -v'
    lin = os.popen(cmd)
    for i in lin:
        if 'The Apache Software Foundation' in i:
            print("Jmeter环境变量配置成功")
            return True
    else:
        print("Jmeter环境变量配置失败")
        return False


if __name__ == '__main__':
    # 分布式ip写法，多个使用逗号隔开
    hostIps = '127.0.0.1'
    if len(sys.argv[1:]) == 5:
        print('参数个数为:', len(sys.argv), '个参数。')
        print('可用参数列表:', str(sys.argv[1:]))
        param = sys.argv[1:]
        print("脚本名字: %s,并发数: %s,步长: %s,执行时间: %s,备注: %s" % (param[0], param[1], param[2], param[3], param[4]))
        jmeterNumber(param[0], param[1], param[2], param[3], param[4], hostIps)
    else:
        print("参数不对")
    pass