python如何給內(nèi)存和cpu使用量設(shè)置限制

更新時(shí)間：2022年05月16日 10:33:59 作者：小平愛(ài)吃肉

這篇文章主要介紹了python如何給內(nèi)存和cpu使用量設(shè)置限制，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

給內(nèi)存和cpu使用量設(shè)置限制

在linux系統(tǒng)中，使用Python對(duì)內(nèi)存和cpu使用量設(shè)置限制需要通過(guò)resource模塊來(lái)完成。

resource文檔地址：resource — Resource usage information

限制Python進(jìn)程cpu使用時(shí)間的樣例如下

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

運(yùn)行上述代碼，當(dāng)超時(shí)時(shí)會(huì)產(chǎn)生SIGXCPU信號(hào)。程序就會(huì)做清理工作然后退出。

要限制內(nèi)存的使用可以使用如下函數(shù)

def limit_memory(maxsize):
? ? soft, hard = resource.getrlimit(resource.RLIMIT_AS)
? ? resource.setrlimit(resource.RLIMIT_AS, (maxsize, hard))

當(dāng)設(shè)定了內(nèi)存限制后，如果沒(méi)有更多的內(nèi)存可用，程序就會(huì)開(kāi)始產(chǎn)生MemoryError異常。

注：以上示例代碼來(lái)源于：《Python Cookbook》P575 “給內(nèi)存和cpu使用量設(shè)置限制”。

查詢(xún)windows的cpu、內(nèi)存使用率

# -*- 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))
# 無(wú)法直接查出內(nèi)存使用率，總內(nèi)存單位是b，而剩余內(nèi)存單位是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的內(nèi)存使用率是{}%".format(mem_usage_rate))

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: