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

1. #!/usr/bin/python  
2. import threading  
3. import time  
4. import os  
5. import string  
6. import sys  
7. class ControlThread(threading.Thread):  
8. def __init__(self):  
9. threading.Thread.__init__(self)  
10. self.runflag = True #線程運(yùn)行標(biāo)示，用于將來(lái)減少線程時(shí)能夠正常結(jié)束  
11. def run(self):  
12. while self.runflag:  
13. os.popen('usleep ' sys.argv[5])  
14. #time.sleep(string.atof(sys.argv[5]))  
15. #這里使用的是linux下shell里面的usleep，而不是python自帶的sleep函數(shù)。  
16. #相比之下，usleep還是相當(dāng)強(qiáng)大的，而python的sleep單位為秒，雖然可以輸入浮點(diǎn)數(shù)， 但還是相對(duì)弱了些  
17. def stop(self):  
18. self.runflag = False 
19. #讓其正常終止循環(huán)  
20. threadList=[]  
21. print 'Start Thread Number:' sys.argv[3] '/tSleep Time(ms):' sys.argv[5]  
22. #初始化一定數(shù)量的線程，否則從零開(kāi)始，可能需要很長(zhǎng)的時(shí)間才能達(dá)到指定范圍  
23. for i in range(0,string.atoi(sys.argv[3])):  
24. thread = ControlThread()  
25. threadList.append(thread)  
26. thread.start()  
27. #這里使用sar來(lái)抓取cpu利用率，這里指的是總的cpu利用率。然后通過(guò)比較，進(jìn)行自適應(yīng)調(diào)整  
28. while True:  
29. output = 100 - string.atof(os.popen('sar 1 1 | grep ^Average | awk /' {print $8}/'').read())  
30. print 'CPU Usage:' str(output) '/tCurrent Thread Number:' str(len(threadList))  
31. if output < string.atoi(sys.argv[1]):#增加線程  
32. for i in range(0,string.atoi(sys.argv[4])):  
33. thread = ControlThread()  
34. thread.start()  
35. threadList.append(thread)  
36. print " " 
37. if output > string.atoi(sys.argv[2]):#減少線程  
38. for i in range(0,string.atoi(sys.argv[4])):   
39. thread = threadList.pop()  
40. thread.stop()  
41. print "-----" 
42.  

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