做过性能测试的童鞋应该都知道集合点这个概念,也就是在多线程的环境中,我们通过设置集合点,使得这些创建好的线程并不立即执行,而是让所有线程等待到某一个点之后再统一一起执行,这样做的目的是对服务器造成真正的并发压力。在性能测试工具Jmeter或LR中可以轻松地做到这点,但如果是编写纯Python脚本的话,该如何实现多线程的类似集合点这样的并发请求呢?
其实要在Python中实现多线程的集合也是比较容易的,我们主要要用到线程同步方法中的Event来实现。Event是在threading模块中的一个类,这个事件类可以通过设置一个事件信号,让所有设置了该事件的线程暂停等待或激活执行,这个类主要有三个方法,大家可以从代码注释中看到这些方法的作用,非常简单:
event = threading.Event()
# 重置event,使得所有该event事件都处于待命状态
event.clear()
# 等待接收event的指令,决定是否阻塞程序执行
event.wait()
# 发送event指令,使所有设置该event事件的线程执行
event.set()
知道了这些方法,我们就可以利用这些方法达到模拟集合点目的。废话不多说,直接上代码:
import threading
class MyThread(threading.Thread):
def __init__(self, event):
super().__init__()
self.event = event
def run(self):
print("线程{}已初始化完成,随时准备启动....".format(self.name))
# 设置线程等待
self.event.wait()
print("{}开始执行...".format(self.name))
if __name__ == '__main__':
event = threading.Event()
threads = []
[threads.append(MyThread(event)) for i in range(1, 11)]
# 必须在子线程start之前先清空所有的event设置,让子线程的event.wait生效
event.clear()
[t.start() for t in threads]
# 设置event事件,事件设置后将通知所有设置了事件对象的线程激活
event.set()
[t.join() for t in threads]
在上面这段中,我们在主线程中新建了10个子线程,并且在每个线程方法中都设置了线程等待语句,这样当线程完成初始化进入run方法后,首先会打印出“线程xx已初始化完毕,随时准备启动..."的输出语句,随后就进入等待状态,后面的语句将不会执行,直到在主线程中运行了event.set()方法,通知所有的线程进入激活状态之后,所有的线程才开始真正执行。
代码运行后,结果如下。从结果中我们可以看到,所有的线程确实是先生成好后,然后再统一启动,成功实现了集合点的效果。
注:本文为蜗牛学院资深讲师卿淳俊老师原创,切勿擅自盗用,如需转载请私聊我处获得授权并注明出处。