python如何给内存和cpu使用量设置限制_Python

python如何给内存和cpu使用量设置限制

2023-01-28 13:59小平爱吃肉 Python

这篇文章主要介绍了python如何给内存和cpu使用量设置限制，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

给内存和cpu使用量设置限制

在linux系统中，使用Python对内存和cpu使用量设置限制需要通过resource模块来完成。

resource文档地址：resource — Resource usage information

限制Python进程cpu使用时间的样例如下

				?

									import signal

									import resource

									import os

									def time_exceeded(signo, frame):

									    print("time's up")

									    raise SystemExit(1)

									def set_max_runtime(seconds):

									    soft,hard = resource.getrlimit(resource.RLIMIT_CPU)

									    resource.setrlimit(resource.RLIMIT_CPU, (seconds, hard))

									    signal.signal(signal.SIGXCPU, time_exceeded)

									if __name__ == '__main__':

									    set_max_runtime(5)

									    while True:

									        pass

运行上述代码，当超时时会产生SIGXCPU信号。程序就会做清理工作然后退出。

要限制内存的使用可以使用如下函数

				?

									def limit_memory(maxsize):

									    soft, hard = resource.getrlimit(resource.RLIMIT_AS)

									    resource.setrlimit(resource.RLIMIT_AS, (maxsize, hard))

当设定了内存限制后，如果没有更多的内存可用，程序就会开始产生MemoryError异常。

注：以上示例代码来源于：《Python Cookbook》P575 “给内存和cpu使用量设置限制”。

查询windows的cpu、内存使用率

				?

									# -*- coding: UTF-8 -*-

									import os

									def get_info(metric):

									    metric_cmd_map = {

									        "cpu_usage_rate": "wmic cpu get loadpercentage",

									        "mem_total": "wmic ComputerSystem get TotalPhysicalMemory",

									        "mem_free": "wmic OS get FreePhysicalMemory"

									    }

									    out = os.popen("{}".format(metric_cmd_map.get(metric)))

									    value = out.read().split("\n")[2]

									    out.close()

									    return float(value)

									# cpu使用率

									cpu_usage_rate = get_info('cpu_usage_rate')

									print("windows的CPU使用率是{}%".format(cpu_usage_rate))

									# 无法直接查出内存使用率，总内存单位是b，而剩余内存单位是kb

									mem_total = get_info('mem_total')/1024

									mem_free = get_info('mem_free')

									mem_usage_rate = (1 - mem_free/mem_total)*100

									print("windows的内存使用率是{}%".format(mem_usage_rate))