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

python给内存和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使用量设置限制”。