多线程运行Jmeter脚本

解决问题：之前对数据库用Jmeter要做一个压力测试的时候，由于有多个Jmeter脚本且希望所有脚本都是在同一时间运行。

解决方法：所以用python写了一个多线程运行的小脚本来执行多个jmk脚本。

一、框架结构以及各种方法的说明：

'''
Created on 2018年6月6日

@author: zww

'''

文件名|方法名 | 说明 |
|--------|--------|--------|
TxtOperating|write| 将内容写进文本文件 |
|--------|--------|--------|
CpuMonitor|change2G | 将磁盘单位转换成G|
CpuMonitor|getCpuInfo | 获取当前CPU的信息 |
CpuMonitor|getMemoryInfo | 获取当前内存的信息 |
CpuMonitor|getDiskInfo|获取当前磁盘的信息 |
CpuMonitor|formating|申请退款|
CpuMonitor|run|执行监控|
|--------|--------|--------|
StartJmeter|getJmeterFile|获取要执行的jmeter脚本列表|
StartJmeter|filterFile|筛选文件，只筛选出.jmx后缀的文件|
StartJmeter|startJmeter|执行.jmx脚本的系统命令组装|
StartJmeter|run|启动多线程运行|

bin|--|启动项目

二、各文件方法的具体代码：

（1）TxtOperating.py,文件操作：

（2）CpuMonitor.py,监控系统cpu、内存以及磁盘情况，提供屏幕输出以及写入文本

# -*- coding: utf-8 -*-
'''
Created on 2018年6月6日

@author: zww
监控系统cpu、内存以及磁盘情况，提供屏幕输出以及写入文本
'''
import time,psutil
from handler import TxtOperating as TxtOp

cpu = {'user':0,'system':0,'idle':0,'percent':0}
memory = {'total':0,'available':0,'percent':0,'used':0,'free':0}

#将磁盘单位转换成G
def change2G(disk_memory):
　　return str(round((disk_memory/1024/1024/1024)))

def getCpuInfo():
　　cpu_info = psutil.cpu_times()
　　cpu['user'] = cpu_info.user
　　cpu['system'] = cpu_info.system
　　cpu['idle'] = cpu_info.idle
　　cpu['percent'] = psutil.cpu_percent(interval=1)

def getMemoryInfo():
　　memory_info = psutil.virtual_memory()
　　memory['total'] = change2G(memory_info.total)
　　memory['available'] = change2G(memory_info.available)
　　memory['percent'] = memory_info.percent
　　memory['used'] = change2G(memory_info.used)
　　memory['free'] = change2G(memory_info.percent)

def getDiskInfo():
　　#磁盘名称
　　disk_id = []
　　#将每个磁盘的total used free percent 分别存入到相应的list
　　disk_total = []
　　disk_used = []
　　disk_free = []
　　disk_percent = []
　　for id in psutil.disk_partitions():
　　　　if 'cdrom' in id.opts or id.fstype == '':
　　　　　　continue
　　　　disk_name = id.device.split(':')
　　　　s = disk_name[0]
　　　　disk_id.append(s)
　　　　disk_info = psutil.disk_usage(id.device)

　　　　disk_total.append(change2G(disk_info.total))
　　　　disk_used.append(change2G(disk_info.used))
　　　　disk_free.append(change2G(disk_info.free))
　　　　disk_percent.append(disk_info.percent)
　　return disk_id,disk_total,disk_used,disk_free,disk_percent

def formating(disk_id,disk_total,disk_used,disk_free,disk_percent):
　　all_str = ''
　　length_of_disk = len(disk_id)
　　for i in range(length_of_disk):
　　　　str = '磁盘 %s:总内存量 %sG,已用 %sG,剩余 %sG,使用率为%s%%'%(disk_id[i],disk_total[i],disk_used[i],disk_free[i],disk_percent[i])
　　　　# print(str) #需要输出在命令台则取消注释
　　　　all_str = ''.join([all_str,str,'\n'])
　　return all_str

def run():
　　while True:
　　　　runtime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())

　　　　getCpuInfo()
　　　　getMemoryInfo()
　　　　disk_id,disk_total,disk_used,disk_free,disk_percent = getDiskInfo()

　　　　cpu_str = ''.join(['CPU使用率:',str(cpu['percent'])])
　　　　memory_str = '內存 :总内存量 %sG,已用 %sG,剩余 %sG,使用率为%s%%'%(memory['total'],memory['used'],memory['available'],memory['percent'])
　　　　disk_str = formating(disk_id,disk_total,disk_free,disk_used,disk_percent)
　　　　str_all = ''.join([runtime,'\n',cpu_str,'\n',memory_str,'\n',disk_str])
　　　　print(str_all)

　　　　#构造监控文本存放的路径和命名
　　　　cur_dir = os.path.abspath(os.path.dirname(__file__))
　　　　run_time = time.strftime('%Y%m%d_%H%M%S',time.localtime())
　　　　result_path = ''.join([cur_dir,'\\MonitorResult\\',run_time,'sysMonitor.txt'])
　　　　TxtOp.write(str_all,result_path)
　　　　time.sleep(3)

（3）StartJmete,多线程运行jmeter脚本：

# -*- coding: utf-8 -*-
'''
Created on 2018年6月6日

@author: zww
多线程启动jmeter脚本
'''
import os,time
import threading

#获取当前目录下的所有文件
def getJmeterFile(filepath='D:\\JmeterFiles'):
　　for root,dirs,files in os.walk(filepath):
　　　　return root,files

#过滤文件，只筛选‘.jmx’的文件
def filterFile(root,files):
　　after_files = []
　　for file in files:
　　　　if file.endswith('.jmx'):
　　　　after_files.append(''.join([root,'\\',file]))
　　return after_files
#运行jmeter脚本
def startJmeter(casename):
　　runtime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
　　comd = ''.join(['jmeter -n -t',casename,' -l',runtime,casename,'.jtl'])
　　os.system(comd)

#多线程运行
def run():
　　root,files = getJmeterFile()
　　excutefiles = filterFile(root,files)
　　for file in excutefiles:
　　　　t = threading.Thread(target=startJmeter,args=(file,))
　　　　#t.setDaemon() #后台运行
　　　　t.start()
（4）启动项目：

# -*- coding: utf-8 -*-
'''
Created on 2018年6月6日

@author: zww

'''
from interface import CpuMonitor,StartJmeter

if __name__ == '__main__':
　　# StartJmeter.run()
　　CpuMonitor.run()

多线程运行Jmeter脚本

猜你喜欢