给内存和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))
以上为个人经验，希望能给大家一个参考，也希望大家多多支持。
查看更多关于python如何给内存和cpu使用量设置限制的详细内容...
声明：本文来自网络，不代表【好得很程序员自学网】立场，转载请注明出处：http://www.haodehen.cn/did17329