计算指定进程占用内存大小

最新推荐文章于 2024-08-11 12:53:12 发布

原创

最新推荐文章于 2024-08-11 12:53:12 发布 · 470 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#运维

本文介绍了如何使用Python脚本来计算指定进程kubelet的内存占用大小，通过运行test.py脚本进行实际操作。

#!/usr/bin/env python
# _*_ coding:UTF-8 _*_
# 收集程序所占用的物理内存大小，占所有物理内存的比例
# OS: Centos 6.7  Python: 2.7.6
# __author__:Dahlhin

import sys
import os
from subprocess import Popen,PIPE

def get_pid(program):
    '获取目标程序的PID列表'
    p = Popen(['pidof',program],stdout=PIPE,stderr=PIPE)
    pids,stderrput = p.communicate()
#     pids = p.stdout.read()  #这种方法也是可以的
#     这里也可以对stderrput来进行判断
    if pids:
        return pids.split()
    else:
        raise ValueError

def mem_calc(pids):
    '计算PIDs占用的内存大小'
    mem_total = 0
    for pid in pids:
        os.chdir('/proc/%s' % pid)
        with open('status') as fd:
            for line in fd:
                if line.startswith('VmRSS'):
                    mem = line.strip().split()[1]
                    mem_total += int(mem)
                    break
    return mem_total


def mem_percent(mem):
    '计算程序内存占用物理内存的百分比'
    with open('/proc/meminfo') as fd:
        for line in fd:
            if line.startswith('MemTotal'):