## 概述
通过计算linux系统的/proc文件中物理内存相关内容,获取指定进程所占的物理内存。
这里以获取apache进程的物理内存和所占比例为例。
### 脚本
* 此脚本可以作为模块使用
* apache会开启多个进程,因此这个脚本适合单进程和多进程。
* 在主函数外,对计算过程进程了封装,定义了`getProgTotalMem`函数,简化了主函数中的内容.
* 当进程不存在时,pidof不报错,输出为空,因此这里检测进程是否存在应特别注意
* 可在[github gist][1]中获取脚本,被墙的原因,多数情况网页开。
```python
#!/usr/bin/env python
# coding: utf-8
import os
import sys
from subprocess import Popen, PIPE
def getPidList(prog_name):
"""获取进程的所有PID
输入:进程名
返回:进程PID列表"""
p = Popen(['pidof', '%s' % prog_name], stdout=PIPE, stderr=PIPE)
std_out, std_err = p.communicate()
# 当进程不存在时,pidof也不报错,所以检测std_out如果为空,说明进程不存在。
if std_out:
pids = std_out.split()
return pids
else:
print 'the prog %s is not exist.' % prog_name
sys.exit()
def getProgMemDict(pids):
"""获取进程每个PID所占的物理内存,存入字典
输入:进程PID列表
返回:形如{PID:mem}的字典"""
mem_dict = {}
for pid in pids:
fn = os.path.join('/proc', pid, 'status')
with open(fn,'r') as fd:
for line in fd:
if line.startswith('VmRSS'):
mem = int(line.split()[1])
mem_dict[pid] = mem
break
return mem_dict
def getProgTotalMem(prog_name):
"""获取进程所占的物理内存
输入:进程名
输出:进程所占的物理内存"""
# 获取进程所有PID
pid_list = getPidList(prog_name)
# 获取进程物理内存字典
prog_mem_dict = getProgMemDict(pid_list)
# 计算进程总的的物理内存
prog_total_mem = sum([mem for mem in prog_mem_dict.itervalues()])
return prog_total_mem
def getSysTotalMem():
"""获取系统总的物理内存
输入:无
输出:总的物理内存"""
with open('/proc/meminfo') as fd:
for line in fd:
if line.startswith('MemTotal'):
mem = int(line.split()[1])
return mem
if __name__ == '__main__':
httpd_total_mem = getProgTotalMem('http')
sysmem = getSysTotalMem()
print 'apache memory is: %d kb, and %.2f%%' % (httpd_total_mem, httpd_total_mem/float(sysmem))
```
[1]:https://gist.github.com/PoplarYang/69daae1e2b11a309e0de5a6a52cca5c4