解决问题:之前对数据库用Jmeter要做一个压力测试的时候,由于有多个Jmeter脚本且希望所有脚本都是在同一时间运行。
解决方法:所以用python写了一个多线程运行的小脚本来执行多个jmk脚本。
一、框架结构以及各种方法的说明:
'''
Created on 2018年6月6日
@author: zww
'''
多线程运行Jmeter的脚本(也可以用来多线程执行其他的命令)
------
目录结构以及作用:
AutoPressureTest/ 根目录
|-- handler/
| |-- __init__.py
| |-- TxtOperating.py 用来操作文件读写的(把监控结果写到文本里面,方便过后查看)
|
|-- interface/
|
| |-- __init__.py
| |-- CpuMonitor.py 监控系统CPU、内存的脚本
| |-- StartJmeter.py 多线程运行jmeter脚本
|
|-- MonitorResult/ 存放监控结果文本的路径
|
|-- bin.py 执行
|-- README
不同文件提供的对应的方法:
文件名|方法名 | 说明 |
|--------|--------|--------|
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()