测试场景:某个客户端程序长时间运行后存在内存泄漏问题,现在开发解决了需要去验证这个问题是否还存在,并要求出具相应测试验证报告。
手段:需要有一个工具能够实时去获取该程序进程一直运行下占用内存,CPU使用率情况。
方法:python去实现这么个监控功能?
import sys
import time
import psutil
sys.argv
# get pid from args
#获取命令行输入的参数个数,sys.ary是一个列表
#如果列表参数<2,说明只输入了python 文件名称.py,则退出不继续运行
if len(sys.argv) < 2:
print ("没有输入要监控的进程编号")
sys.exit()
#获取进程
print("打印进程号:"+sys.argv[1])
pid = int(sys.argv[1])
p = psutil.Process(pid)
#监控进程并将获取CPU,内存使用情况写入csv文件中
interval = 60 # 获取CPU,内存使用情况轮询时间间隔
num=100
with open("process_monitor_" + p.name() + '_' + str(pid) + ".csv", "a+") as f:
f.write("时间,cpu使用率(%),内存使用率(%),内存使用值MB\n") # csv文件表头列名:time,cpu使用率,内存使用率,内存占用值MB
while num>0:
num=num-1
current_time = time.strftime('%Y%m%d-%H%M%S',time.localtime(time.time()))
cpu_percent = p.cpu_percent() # better set interval second to calculate like: p.cpu_percent(interval=0.5)
mem_percent = p.memory_percent()
mem_info=p.memory_info().rss
mem_MB=4096 / mem_percent
print('当前进程的内存使用:',mem_info)
print('当前进程的内存使用:%.4f MB' %mem_MB)
line = current_time + ',' + str(cpu_percent) + ',' + str(mem_percent)+','+str(mem_MB)
print (line)
f.write(line + "\n")
time.sleep(interval)
?
|