筆者最近在做的是虛擬化相關(guān)的一個(gè)demo，牽涉到虛擬化環(huán)境下的服務(wù)器負(fù)載均衡。我們用到一個(gè)簡單的case，就是在負(fù)載出現(xiàn)“不平衡”的時(shí)候，進(jìn)行 live migration。由于只是demo需要，我們的負(fù)載暫時(shí)只考慮到cpu的利用率。而實(shí)際準(zhǔn)備中，由于沒有客戶端的壓力，很難做到CPU利用率的精確控制。這里寫了個(gè)簡單的腳本程序，通過自適應(yīng)的調(diào)節(jié)，來達(dá)到將服務(wù)器CPU利用率控制在一定的范圍內(nèi)，從而為確保在某臺(tái)服務(wù)器CPU過高的時(shí)候，會(huì)自動(dòng)將虛擬機(jī)live migration到其他CPU利用率低的機(jī)器上的測(cè)試做了準(zhǔn)備。

python實(shí)現(xiàn)的腳本如下，這個(gè)腳本需要5個(gè)數(shù)字參數(shù)的輸入，分別為：

最低CPU利用率，最高CPU利用率，初始線程數(shù)量，每次調(diào)節(jié)的線程數(shù)量，每個(gè)線程睡眠的時(shí)間（毫秒）

當(dāng)然，很大程度上，由于硬件環(huán)境的不同，這些參數(shù)帶有很強(qiáng)的經(jīng)驗(yàn)性。

#!/usr/bin/python  
import threading  
import time  
import os  
import string  
import sys  
class ControlThread(threading.Thread):  
def __init__(self):  
threading.Thread.__init__(self)  
self.runflag = True #線程運(yùn)行標(biāo)示，用于將來減少線程時(shí)能夠正常結(jié)束  
def run(self):  
while self.runflag:  
os.popen('usleep ' sys.argv[5])  
#time.sleep(string.atof(sys.argv[5]))  
#這里使用的是linux下shell里面的usleep，而不是python自帶的sleep函數(shù)。  
#相比之下，usleep還是相當(dāng)強(qiáng)大的，而python的sleep單位為秒，雖然可以輸入浮點(diǎn)數(shù)，
但還是相對(duì)弱了些  
def stop(self):  
self.runflag = False 
#讓其正常終止循環(huán)  
threadList=[]  
print 'Start Thread Number:' sys.argv[3] '\tSleep Time(ms):' sys.argv[5]  
#初始化一定數(shù)量的線程，否則從零開始，可能需要很長的時(shí)間才能達(dá)到指定范圍  
for i in range(0,string.atoi(sys.argv[3])):  
thread = ControlThread()  
threadList.append(thread)  
thread.start()  
#這里使用sar來抓取cpu利用率，這里指的是總的cpu利用率。然后通過比較，進(jìn)行自適應(yīng)調(diào)整  
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 < string.atoi(sys.argv[1]):#增加線程  
for i in range(0,string.atoi(sys.argv[4])):  
thread = ControlThread()  
thread.start()  
threadList.append(thread)  
print " " 
if output > string.atoi(sys.argv[2]):#減少線程  
for i in range(0,string.atoi(sys.argv[4])):   
thread = threadList.pop()  
thread.stop()  
print "-----" 
 

總的來說，這個(gè)腳本比較簡單，而且在不同的機(jī)器上，需要操作人員的經(jīng)驗(yàn)來初始化才能達(dá)到最佳效果。但這畢竟是我第一個(gè)python程序，而且也達(dá)到了預(yù)期的目的。以上只是能運(yùn)行的核心代碼，至于usage，錯(cuò)誤處理，退出程序等，就沒有給出來了