筆者最近在做的是虛擬化相關(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)性。
- #!/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)示,用于將來(lái)減少線程時(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ù)量的線程,否則從零開(kāi)始,可能需要很長(zhǎng)的時(shí)間才能達(dá)到指定范圍
- for i in range(0,string.atoi(sys.argv[3])):
- thread = ControlThread()
- threadList.append(thread)
- thread.start()
- #這里使用sar來(lái)抓取cpu利用率,這里指的是總的cpu利用率。然后通過(guò)比較,進(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 "-----"
總的來(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)了