#coding=utf8
importtimeimportpsutil, gc, commands,osfrom logger_until importLoggerUntilfrom until importkeep_circulating
logger= LoggerUntil(name="Monitor").getlog(logfilename='Monitor.log', loglevel=2, add_StreamHandler=1)
need_monitor_procces_names= [ ##需要检测的进程的cmd命令放到这里,支持模糊匹配
'touna0627.py','dailiip.py','redis-server','mongod',
]classMonitor(object):def __init__(self):
self.specified_process_list=self.get_specified_process()
self.current_process=self.get_current_process()
@staticmethoddefprint_all_cmdlines():for pid inpsutil.pids():
p=psutil.Process(pid)printp.cmdline()
@staticmethoddefget_specified_process():
all_pids=psutil.pids()
process_list=[]for pid inall_pids:
p=psutil.Process(pid)
p_cmdline=p.cmdline()for argx inp_cmdline:for name inneed_monitor_procces_names:if argx.find(name) > -1:if p.status() != 'stopped':
process_list.append(p)
p_pid_set=set()
process_list2=[]for p inprocess_list:if p.pid not inp_pid_set:
process_list2.append(p)
p_pid_set.add(p.pid)returnprocess_list2
@staticmethoddefmonitor_system():
psutil.cpu_percent()
time.sleep(1)
mem=psutil.virtual_memory()
mem_total= mem.total / 1000000mem_available= mem.available / 1000000mem_percent_str= str(mem.percent) + '%'cpu_count=psutil.cpu_count()
cpu_percent_str= str(psutil.cpu_percent()) + '%'msg= '本机总内存是:{0}M , 本机可用内存是:{1}M, 本机内存使用率是:{2}, 本机cpu核数是:{3}, 本机cpu使用率是:{4}\n\n'.format(mem_total, mem_available, mem_percent_str, cpu_count, cpu_percent_str)
logger.info(msg)defmonitor_specified_process(self):for p inself.specified_process_list:
p.cpu_percent(None)
time.sleep(1)for p inself.specified_process_list:#p = psutil.Process(0)
""":type :psutil.Process"""cmdline_str= ' '.join(p.cmdline()).ljust(60, ' ')
p_cpu_percent_str= str(round(p.cpu_percent(), 2)) + '%'p_memory_percent_str= str(round(p.memory_percent(), 2)) + '%'p_strated_time= time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(p.create_time()))
p_pid_str=str(p.pid)
msg= '进程' + cmdline_str + '的pid是:' + p_pid_str + 'cpu使用率是:' + p_cpu_percent_str + '内存使用率是:' +p_memory_percent_str \+ '进程的启动时间是:' +p_strated_time
logger.info(msg)
@staticmethoddefget_current_process():returnpsutil.Process()defis_need_release(self,threshold):printself.current_process.memory_percent()if self.current_process.memory_percent()
logger.info('回收当前 %s 进程id的内存' %self.current_process.pid)return 1
deffree_current_process_memory(self, threshold):"""回收python所处的当前进程的内存"""
if self.is_need_release(threshold) == 1:
gc.collect()classMemoryReleaser():def __init__(self,threshold,cmd_name_str='touna0627.py'):""":type threshold:float"""self.threshold= threshold #内存使用率的阈值
self.cmd_name_str =cmd_name_str
self.stutus, self.output= self.__get_memory_available()def __get_memory_available(self):#status, output = commands.getstatusoutput("free -m | grep Mem | awk '{print $4}'") ##shell命令查询计算机可用内存
status, output = commands.getstatusoutput("ps aux | grep %s | sort -k4,4nr|head -1| awk '{print $4}'"%(self.cmd_name_str)) ##shell命令查询程序的内存使用率
returnstatus, outputdefrelease_memory(self):if float(self.output) >self.threshold:
logger.info('程序的内存使用率是 {}% ,程序需要释放内存'.format(self.output))
gc.collect()
@keep_circulating(10)defmonitoring():
MemoryReleaser(40).release_memory() ###这一行来释放内存
monitor =Monitor()
monitor.monitor_specified_process()
monitor.monitor_system()if __name__ == "__main__":passa= list(range(10000000))dela
time.sleep(30)
monitoring()