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

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

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

注:本文为蜗牛学院资深讲师卿淳俊老师原创,切勿擅自盗用,如需转载请私聊我处获得授权并注明出处。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值