Locust 脚本开发入门（2）

Locust 脚本开发入门（2）

 

脚本开发入门

现在我们对这个脚本进行一些修改，让它更像一个真实的用户访问行为

1） 任务等待时间：任务之间按照特定的等待时间进行间隔

Locust 的任务之间等待时间控制包括：

方法1：between 类，指定范围内随机（5~10秒）等待

wait_time = between(5, 10)

方法2：constant 类，从上一次响应结束后，等待特定时间（3秒），再发起下一次请求

wait_time = constant(3)

方法3：constant_pacing 类，从上一次请求发起后，等待特定时间（3秒），再发起下一次请求，如果上一次请求的响应时间大于指定的时间（3秒），则在响应后立即发起下一次请求

wait_time = constant_pacing(3)

2） 步骤等待时间：在任务内加入步骤之间的时间间隔

方法1：从 time 导入 sleep 类即可

至此，脚本实现如下

from time import sleep
from locust import HttpUser, task, constant

class cnblogUser(HttpUser):
    
    # open_blog和open_links任务之间等待固定的3秒
    wait_time = constant(3)

    @task(2)
    def open_blog(self):
        self.client.get("/huanghaopeng/")

    @task(1)
    def open_links(self):
        # 在open_links任务内，上下两个步骤之间，等待1秒
        self.client.get("/huanghaopeng/p/13100305.html")
        sleep(1)
        self.client.get("/huanghaopeng/p/13220749.html")
        sleep(1)
        self.client.get("/huanghaopeng/p/13187807.html")
        sleep(1)

    def on_start(self):
        self.client.get("/") 

3） 参数化：从页面解析博客文章 ID

由上面 open_links 任务的内容可知，13100305、13220749、13187807 是文章 ID，但是随着日后博客内容的增删改，这些博客链接可能随之失效、新增，因此为了让我们的脚本行为更符合一个真实用户的访问行为，现需要脚本在打开我博客的首页后从首页的页面源码中，解析出文章链接进行点击。所以在下面的脚本中，我将会：

1. 导入 re，在 open_blog 任务中对博客的页面html源码进行解析，获得文章列表的url
2. 在 open_links 任务中，对 open_blog 任务解析得出的文章url列表进行遍历访问

import re
from time import sleep
from locust import HttpUser, task, constant

class cnblogUser(HttpUser):

    wait_time = constant(3)

    @task(2)
    def open_blog(self):
        with self.client.get("/huanghaopeng/") as resp:
            if resp.status_code < 300:
                pattern = re.compile(r'<div class="c_b_p_desc">
.*<a href="(.*)" class="c_b_p_desc_readmore">')
                self.urlList = pattern.findall(resp.text)
            else:
                pass

    @task(1)
    def open_links(self):
        # 由于权重配置，无法确保 open_links 任务执行时 self.urlList 包含文章链接列表
        for url in self.urlList:
            self.client.get(url)

    def on_start(self):
        self.client.get("/")

但是，上面脚本存在一个问题，由于当前任务的执行是依据权重配置执行的，无法确保 open_blog 和 open_links 的先后执行顺序。
当然，你也可以把 open_blog 和 open_links 两个任务内的工作合并为一个任务就可以达到顺序执行的目的。

4） 任务控制：按照顺序的方式执行（SequentialTaskSet 类）

主要步骤：

• 导入 SequentialTaskSet 类
• 创建一个 任务类，继承自 SequentialTaskSet，在里面编排好任务执行的顺序
• 在 cnblogUser 类内部，通过 tasks 转载需要执行的任务

import re
from time import sleep
from locust import HttpUser, task, constant, SequentialTaskSet

# 继承 SequentialTaskSet 的一个任务类，内部编排好任务的执行顺序
class TaskCase(SequentialTaskSet):
    
    # 初始化
    def on_start(self):
        self.client.get("/")

    # @task 装饰器说明下面是一个任务
    @task
    def open_blog(self):
        with self.client.get("/huanghaopeng/") as resp:
            if resp.status_code < 300:
                pattern = re.compile(r'<div class="c_b_p_desc">
.*<a href="(.*)" class="c_b_p_desc_readmore">')
                self.urlList = pattern.findall(resp.text)
            else:
                pass
                
    # @task 装饰器说明下面是一个任务
    @task
    def open_links(self):
        for url in self.urlList:
            self.client.get(url)

# 继承 httpUser
class cnblogUser(HttpUser):
    tasks = [TaskCase]
    wait_time = constant(3)
    

至此，任务执行顺序将为：on_start -> open_blog -> open_links，执行压测场景看看

由上面的结果，可以看到，虚拟用户实现了 打开博客首页，进入我的博客并解析页面上的文章链接，再逐个访问，但是，浏览多篇文章本质上只是不同的文章id传参请求，没有在结果中分别统计的需要，因此我们给请求指定一下name，进行合并统计：

    def open_links(self):
        for url in self.urlList:
            # 对name进行传参为任务名称
            self.client.get(url，name='open_links')

再次执行，可见多篇文章遍历访问的请求响应已经被合并统计