06-Locust-设置集合点

最新推荐文章于 2024-02-29 16:13:42 发布

原创

最新推荐文章于 2024-02-29 16:13:42 发布 · 1.5k 阅读

8 ·

CC 4.0 BY-SA版权

-- 来着爱学习de测试小白

文章标签：

#locust #python #集合点

本文介绍如何使用Locust进行负载测试时设置集合点。通过示例代码展示如何控制所有Locust实例同步启动任务，确保并发请求场景下系统的稳定性和性能。

前言

代码示例

前言

一定数量的用户，同时并发请求某个接口，这时就会用到集合点，下面一起来学习下locust设置集合点

代码示例

# -*- coding: utf-8 -*-
# @Time    : 2021/4/19
# @Author  : 大海

import os
from locust import TaskSet, task, events, HttpUser
from gevent._semaphore import BoundedSemaphore

all_locusts_spawned = BoundedSemaphore()
all_locusts_spawned.acquire()


def on_hatch_complete(**kwargs):
    # 创建钩子方法
    all_locusts_spawned.release()


# 挂在到locust钩子函数（所有的Locust示例产生完成时触发）
# events.hatch_complete += on_hatch_complete  # 1.0之前的写法
events.spawning_complete.add_listener(on_hatch_complete)  # 1.0之后的写法


class TestTask(TaskSet):
    def on_start(self):
        """所有任务启动前执行"""
        all_locusts_spawned.wait()
        print('所有任务启动前执行')

    @task(1)
    def index(self):
        all_locusts_spawned.wait()

        url = 'https://www.baidu.com'
        self.client.get(url)
        print('