使用Locust进行接口性能测试：Locust and TaskSet类详细分析_locust 报告接口(2)

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

class MyUser(Locust):
# …

class MyTaskSet(TaskSet):
    tasks = {task1: 2, task2: 1} # 设置执行频率
      
    def task1(self):
        # 执行任务1
        pass
    
    def task2(self):
        # 执行任务2
        pass

`task1`的执行概率是`task2`的两倍。


#### 2.3 on\_start函数


`on_start`用于任务开始执行任务前的预处理操作。例如，登录或初始化数据。它主要是使用了requests.Session，所以执行一次就可以让后续所有任务执行过程中都有登录态。

from locust import Locust, TaskSet, task

class MyUser(Locust):
# …

class MyTaskSet(TaskSet):
    @task(1)
    def my_task(self):
        self.client.get("/list")
        # 其他逻辑
        pass

    def on_start(self):
    # 在用户开始执行任务前执行的操作，例如登录
    self.client.post("/login")
        pass

`on_start`函数使用非常频繁。


#### 2.4 控制任务的执行顺序


`@seq_task()`可以控制任务的执行顺序。  
 按顺序执行任务：

from locust import Locust, TaskSequence, seq_task, task, between

class MyUser(Locust):
wait_time = between(1, 3) # 设置任务执行的等待时间范围

# ...

class MyTaskSequence(TaskSequence):
    @seq_task(1)  # 顺序1
    def login(self):
        # 执行登录操作
        pass
    
    @seq_task(2)  # 顺序2
    def view_items(self):
        # 执行查看商品列表的操作
        pass

    @seq_task(3)  # 顺序3
    @task(3)  # 为该任务设置执行概率
    def add_to_cart(self):
        # 随机执行添加商品到购物车的操作，概率为3/4
        pass

    @seq_task(4)  # 顺序4
    def checkout(self):
        # 执行结账操作
        pass

#### 2.5 休眠等待（wait）


在任务之间插入休眠时间`wait_time`，可以模拟用户在执行任务时的等待行为，且Locust内置一些函数，返回`wait_time`方法


* `between(min, max)` 生成指定范围内的随机等待时间, `min_wait`和`max_wait`分别表示等待时间的最小值和最大值（以毫秒为单位）
* `constant(constant_wait)` 生成固定的等待时间,`constant_wait`表示等待的时间（以毫秒为单位）
* `constant_pacing(pacing_interval)` 指定任务执行的固定间隔，比如：`constant_pacing(2)` 表示用户执行任务2秒，然后再继续下一个任务。

from locust import Locust, between, constant_pacing

class MyUser(Locust):
# 使用between函数，每个用户等待1到3秒之间的随机时间
wait_time = between(1000, 3000)

-----------------分割线------------------------------------

class MyUser(Locust):
# 使用constant_pacing属性，每个用户任务之间固定间隔2秒
wait_time = constant_pacing(2)

#### 2.6 TaskSets 嵌套


可以在一个TaskSet中嵌套其他TaskSet，实现更复杂的用户行为模拟。例如：

from locust import Locust, TaskSet, task

class MyUser(Locust):
# …

class SubTaskSet(TaskSet):
    @task(1)
    def sub_task1(self):
        # 执行子任务1
        pass
    
    @task(2)
    def sub_task2(self):
        # 执行子任务2
        pass

class MyTaskSet(TaskSet):
    @task
    def task1(self):
        # 执行任务1
        pass

    @task
    def task2(self):
        # 执行任务2，并嵌套执行子任务集
        self.sub_task_set()

    tasks = {task1: 1, task2: 1}

    def sub_task_set(self):
        self.client.get("/otherurl")  
        self.SubTaskSet(self.locust).run()

在`MyTaskSet`中的`task2`任务中，我们嵌套执行了`SubTaskSet`。这样就可以实现一些非常复杂的业务流程的压测了，比如：用户打开网站首页，可能会直接离开，也可能会去流量商品，也可能加入商品到购物车下单等等场景一起模拟。


#### 2.7 中断控制(interrupt)


做压测的时候，有些场景下是要提前结束某些任务的，那么可以设置一个计数器，在计数到一定值后中断任务的执行，比上述的嵌套场景中也就刚刚好符合这点，嵌套子任务执行达到某一个阈值的时候，就中断回到父任务，这样就阻止了子任务一直执行的情况发生。

from locust import Locust, TaskSet, InterruptTaskSet, task, between

class MyUser(Locust):
wait_time = between(1, 3) # 设置任务执行的等待时间范围

class SubTaskSet(TaskSet):
    def no_start(self):
        # 计数
        self.counter = 0

    @task(1)
    def sub_task1(self):
        # 执行子任务1
        # 一些任务逻辑
        pass

    @task(2)
    def sub_task2(self):
        # 执行子任务2，达到计数后中断退出子任务
        if self.counter > 100:
            self.interrupt()
        self.counter += 1
        # 一些任务逻辑
        pass

class MyTaskSet(InterruptTaskSet):

    @task(1)
    def task1(self):
        # 执行任务1
        # 一些任务逻辑
        pass

    @task(2)

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

需要这份系统化的资料的朋友，可以戳这里获取

有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！**

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

需要这份系统化的资料的朋友，可以戳这里获取