多線程運行Jmeter腳本

解決問題：以前對數據庫用Jmeter要作一個壓力測試的時候，因爲有多個Jmeter腳本且但願全部腳本都是在同一時間運行。

解決方法：因此用python寫了一個多線程運行的小腳原本執行多個jmk腳本。

1、框架結構以及各類方法的說明：

 

2、各文件的方法：

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

（2）CpuMonitor.py,監控系統cpu、內存以及磁盤狀況

(3)StartJmeter.py：

 

 

 

3、具體代碼：

（1）CpuMonitor.py

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

（2）StartJmeter.py：

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

(3)TxtOperating.py:

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

@author: zww

讀取或者寫入文本文件
'''

def write(filestr,filepath='D:\\AutoPressMonitor\\sysMonitor.txt'):
　　str = ''.join([filestr,'\n'])
　　with open(filepath,'a') as fp:
　　　　fp.write(str)
　　　　fp.close()

（4）bin.py：

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

@author: zww

'''
from interface import CpuMonitor,StartJmeter

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