python CPU利用率控制

# !/usr/bin/python
# coding:utf-8

import threading
import os
import string
import sys

# 最低CPU利用率，最高CPU利用率，初始线程数量，每次调节的线程数量，每个线程睡眠的时间（毫秒）
# python run.py 70 80 220 3 100

class ControlThread(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
        self.runflag = True  # 线程运行标示，用于将来减少线程时能够正常结束

    def run(self):
        while self.runflag:
            os.popen('usleep ' + sys.argv[5])
            # time.sleep(string.atof(sys.argv[5]))
            # 这里使用的是linux下shell里面的usleep，而不是python自带的sleep函数。
            # 相比之下，usleep还是相当强大的，而python的sleep单位为秒，虽然可以输入浮点数，但还是相对弱了些

    def stop(self):
        self.runflag = False
        # 让其正常终止循环


threadList = []

print('Start Thread Number:' + sys.argv[3] + '\tSleep Time(ms):' + sys.argv[5])

# 初始化一定数量的线程，否则从零开始，可能需要很长的时间才能达到指定范围
for i in range(0, string.atoi(sys.argv[3])):
    thread = ControlThread()
    threadList.append(thread)
    thread.start()

# 这里使用sar来抓取cpu利用率，这里指的是总的cpu利用率。然后通过比较，进行自适应调整
while True:
    output = 100 - string.atof(os.popen('sar 1 1 | grep ^Average | awk \'{print $8}\'').read())
    print('CPU Usage:' + str(output) + '\tCurrent Thread Number:' + str(len(threadList)))

    if output < int(sys.argv[1]):  # 增加线程
        for i in range(0, int(sys.argv[4])):
            thread = ControlThread()
            thread.start()
            threadList.append(thread)
        print("+++++")
    if output > int(sys.argv[2]):  # 减少线程
        for i in range(0, int(sys.argv[4])):
            thread = threadList.pop()
            thread.stop()
        print("-----")