21
08

用Python代码模拟实现性能测试中的集合点

小伙伴们,大家好~ 今天的这篇文章来自蜗牛学院在线教育部卿淳俊老师。

小伙伴们,大家好~

今天的这篇文章来自蜗牛学院在线教育部卿淳俊老师。


蜗牛学院资深讲师,15年测试开发及项目管理经验,认证敏捷项目Scrum Master,精通各种测试及开发工具、框架的设计和实现,在移动互联网、敏捷项目管理、电商测试等领域有丰富的项目经验。

对大数据、机器学习、区块链等新兴技术均有深入研究。6年讲师经验铸就了深入浅出的授课风格,秉持着“授人以鱼不如授人以渔”的理念,让学员达到“知其然更要知其所以然”,真正掌握技术的本质



·  正  ·  文  ·  来  ·  啦  ·



做过性能测试的童鞋应该都知道集合点这个概念,也就是在多线程的环境中,我们通过设置集合点,使得这些创建好的线程并不立即执行,而是让所有线程等待到某一个点之后再统一一起执行,这样做的目的是对服务器造成真正的并发压力。在性能测试工具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()方法,通知所有的线程进入激活状态之后,所有的线程才开始真正执行。

代码运行后,结果如下。从结果中我们可以看到,所有的线程确实是先生成好后,然后再统一启动,成功实现了集合点的效果。


20190919_105548_380.jpg



为了答谢大家对蜗牛学院的支持,蜗牛学院将会定期对大家免费发放干货,敬请关注蜗牛学院的官方微信。


20190320_095757_834.jpg





版权所有,转载本站文章请注明出处:蜗牛学苑, https://www.woniuxy.cn/article/326