locust自定义负载策略。

1.时间峰值策略

每运行一分钟启动100个用户，总运行时间为10分钟

class CustomShape(LoadTestShape):

        # 设置时限

        time_limit = 600

        # 设置产生率

        spawn_rate = 20

        def tick(self):

            '''

            设置 tick()函数

            并在tick()里面调用 get_run_time()方法

            '''

            # 调用get_run_time()方法

            run_time = self.get_run_time()

            # 运行时间在 10分钟之内，则继续执行

            if run_time < self.time_limit:

                uesr_count = round(run_time, -2)

                # 返回user_count,spawn_rate这两个参数

                return (uesr_count, self.spawn_rate)

            return None

2.时间阶段负载策略

class MyCustomShape(LoadTestShape):

    """

        ps:在不同的阶段 具有不同的用户数和 产生率的 图形形状

        time -- 持续时间经过多少秒后，进入到下个阶段

        users -- 总用户数

        spawn_rate -- 产生率，即每秒启动/停止的用户数

    """

    stages = [

        {"time": 10, "users": 10, "spawn_rate": 10},

        {"time": 30, "users": 30, "spawn_rate": 10},

        {"time": 60, "users": 60, "spawn_rate": 10},

        {"time": 200, "users": 120, "spawn_rate": 10},

    ]

    def tick(self):

        run_time = self.get_run_time()

        for stage in self.stages:

            if run_time < stage['time']:

                tick_data = (stage['users'],stage['spawn_rate'])

                return tick_data

        return None

3.逐步施压负载策略

class MyCustomShape(LoadTestShape):

    '''

        step_time -- 逐步加载时间长度

        step_load -- 用户每一步增加的量

        spawn_rate -- 用户在每一步的停止/启动的多少用户数

        time_limit -- 时间限制压测的执行时长

    '''

    # 逐步负载策略每隔30秒新增启动10个用户

    setp_time = 30

    setp_load = 10

    spawn_rate = 10

    time_limit = 300

    def tick(self):

        run_time = self.get_run_time()

        if run_time > self.time_limit:

            return None

        current_step = math.floor(run_time /self.setp_time) +1

        return(current_step * self.setp_load,self.spawn_rate)

4.双波峰负载策略

class MyCustomShape(LoadTestShape):

    '''

        step_time -- 逐步加载时间长度

        step_load -- 用户每一步增加的量

        spawn_rate -- 用户在每一步的停止/启动的多少用户数

        time_limit -- 时间限制压测的执行时长

    '''

    # 逐步负载策略每隔30秒新增启动10个用户

    setp_time = 30

    setp_load = 10

    spawn_rate = 10

    time_limit = 300

    def tick(self):

        run_time = self.get_run_time()

        if run_time > self.time_limit:

            return None

        current_step = math.floor(run_time /self.setp_time) +1

        return(current_step * self.setp_load,self.spawn_rate)

巴特西

locust自定义负载策略。

最新文章

热门文章