聊聊Locust常见参数及常用代码

 - `-f` 或 `--locustfile`：指定locustfile的路径，默认为`locustfile.py`。 - `--host`：指定被测试应用的URL（例如：http://example.com）。 - `-u` 或 `--users`：设置模拟的最大用户数。 - `-r` 或 `--spawn-rate`：设置每秒启动的用户数。 - `-t` 或 `--run-time`：设置测试运行的时间（例如：10s, 1m, 1h）。 - `--headless`：无界面模式，即不启动Web界面，直接运行测试。 - `--web-host`：指定Web界面的主机地址（默认为0.0.0.0）。 - `--web-port`：指定Web界面的端口（默认为8089）。 - `--csv`：指定输出CSV文件的前缀（将生成多个CSV文件，如：_stats.csv, _distribution.csv等）。 - `--html`：生成HTML报告。

with self.client.get("/resource", catch_response=True) as response:    if "success" not in response.text:        response.failure("未找到关键字")    elif response.elapsed > 1000:  # 响应时间>1秒标记失败        response.failure("响应超时")

import csvdef read_data():    with open("users.csv") as f:        return list(csv.DictReader(f))class TestUser(HttpUser):    users = read_data()    current_user = 0    @task    def update_profile(self):        user = self.users[self.current_user % len(self.users)]        self.client.put(f"/user/{user['id']}", json={"name": user["name"]})        self.current_user += 1

# 阶梯式等待（前5秒快，之后慢）def custom_wait(user):    if user.run_time < 5:        return 0.5  # 前5秒等待0.5秒    return 3class CustomUser(HttpUser):    wait_time = custom_wait

locust -f locustfile.py --host http://your-api.com

locust -f locustfile.py --headless -u 1000 -r 100 -t 5m

from locust import HttpUser, task, betweenclass MyUser(HttpUser):    wait_time = between(1, 3)  # 用户任务之间的等待时间（1-3 秒）    @task    def index_page(self):        self.client.get("/")  # 访问首页    @task(2)  # 权重为 2，执行概率是其他任务的两倍    def about_page(self):        self.client.get("/about")  # 访问关于页面

# 示例：固定等待 2 秒wait_time = constant(2)# 示例：任务间隔至少 1 秒wait_time = constant_pacing(1)

@task(3)  # 权重为 3def search_product(self):    self.client.post("/search", json={"q": "example"})@task(1)  # 权重为 1def view_home(self):    self.client.get("/")

@taskdef login(self):    self.client.post("/login", json={"username": "user", "password": "123"})

@taskdef test_api(self):    response = self.client.get("/api/data")    if response.elapsed.total_seconds() > 0.5:        # 响应时间超过 500ms，标记为失败        events.request_failure.fire(            request_type="GET",            name="test_api",            response_time=response.elapsed.total_seconds(),            exception="Response time exceeded SLA"        )

locust -f my_test.py --master

locust -f my_test.py --slave --master-host=192.168.1.100