locust 通过队列实现 task级集合点并发

最新推荐文章于 2024-09-12 10:36:54 发布

词穷墨尽

最新推荐文章于 2024-09-12 10:36:54 发布

阅读量1.1k

点赞数

分类专栏： locust python 文章标签： python locust

本文链接：https://blog.csdn.net/weixin_41167150/article/details/106369460

版权

在进行公司压测时，作者选择了 Locust 作为工具，但遇到一个需求：部分任务需要并发执行，而其他任务保持正常运行。通过查阅资料，作者发现官方提供的协程钩子方法在运行开始后无法实现精确控制。为解决这个问题，作者采用了队列锁的方法来实现这一功能，同时欢迎有其他解决方案的人通过私信或加群交流。

摘要由CSDN通过智能技术生成

最近公司在搞压测笔者也是查了很多工具

lr收费用不起 jm 测试机扛不住最后就选择了locust

使用中有个场景覆盖不到就是几个task中有一个需要并发提交，其余task正常运行即可

也是查了一些资料官网上给的钩子方法是基于协程的运行开始前可实现并发一旦运行开始即开始即不可控

话不多说直接上代码

from gevent._semaphore import Semaphore
from locust import HttpLocust, TaskSet, events, task, HttpUser

all_locusts_spawned = Semaphore()
all_locusts_spawned.acquire()


class TestUserTask(TaskSet):
    @task()
    def job(self):

        print("进来了{}".format(self.user))
        data = self.user.myflag.get(0)
        is_wait = data['is_wait']
        flag = data['flag']
        print("flag={}".format(flag))

        if flag >= 19:
            is_wait = True
        print("is_wait={}".format(is_wait))
        if is_wait:
            flag = flag - 1
            if flag == 0:
                is_wait = False
            data = {"flag": flag, "is_