浅谈locust 性能压测使用
1. 基本介绍
Locust是一个开源的负载测试工具，用于模拟大量用户并发访问一个系统或服务，以评估其性能和稳定性。编写语言为Python，可通过Python来自定义构建性能压测场景脚本。Locust支持分
布式负载测试，可以通过多个机器协同工作来模拟大量用户并发访问。并提供了一个Web界面，可以实时监控并查看测试的性能指标和统计数据，并支持常见的协议，如HTTP、HTTPS、
FTP、WebSocket等，更重要的是具有良好的扩展性，用户可以编写自定义的扩展来满足特定需求。它还提供了多个钩子函数，可以在测试运行过程中插入自定义代码，高度灵活，随心创造。
2. 基本组成
User 类（User class）：
在Locust脚本中，必须定义一个继承自HttpUser或User类的用户类。这个用户类代表了模拟的用户，并定义了用户的行为和任务。用户类中的每个方法都代表一个用户行为，例如发送请求、处理响应等。
Task 类（Task class）：
用户类中的方法通常被称为任务（task），每个任务定义了用户的一个具体行为。任务可以是一个函数、一个装饰器或一个类方法。任务通常用于发送请求、处理响应等操作。
TaskSet 类（TaskSet class）：
TaskSet类是用于组织任务的一种方式。它可以包含多个任务，并定义了用户在执行这些任务时的行为和逻辑。通过TaskSet，可以更好地组织和管理任务，使脚本的结构更加清晰。
Weight 属性（Weight attribute）：
在用户类中，可以为每个任务指定一个权重（weight）。权重用于控制每个任务在负载测试中的执行频率。较高权重的任务将更频繁地执行，而较低权重的任务将相对较少执行。
on_start() 方法：
用户类中的on_start()方法定义了每个用户在开始执行任务之前要执行的初始化操作。例如，可以在on_start()方法中进行用户登录或准备测试数据等操作。
on_stop() 方法：
用户类中的on_stop()方法定义了每个用户在完成所有任务后要执行的清理操作。例如，可以在on_stop()方法中进行用户退出或清理资源等操作。
Host 属性：
在用户类中，可以定义一个host属性，指定要测试的目标主机地址。这个属性告诉Locust在执行请求时要发送到哪个主机。
TaskSet 属性：
用户类中可以定义一个或多个TaskSet属性，用于组织和管理任务。通过TaskSet属性，可以将任务分组并按组执行，使脚本更加模块化和可维护。
3. 基本使用步骤
定义用户行为和任务：
首先，需要定义用户的行为和任务，即编写Locust脚本。在脚本中，定义一个继承自HttpUser或User类的用户类，并在其中定义任务和行为。例如，发送请求、处理响应等。
from locust import HttpUser, task, between
class MyUser(HttpUser):
wait_time = between(2, 5) # 定义用户等待时间范围
@task
def my_task(self):
self.client.get("/path") # 发送GET请求
@task(3) # 指定任务权重
def weighted_task(self):
response = self.client.post("/path", json={"key": "value"}) # 发送POST请求
if response.status_code == 200:
self.log_success("POST request succeeded", response.elapsed.total_seconds())
else:
self.log_failure("POST request failed", response.elapsed.total_seconds())
def log_success(self, message, response_time): # 定义请求成功和失败处理
self.environment.events.request_success.fire(
request_type="my_task",
name=message,
response_time=response_time,
response_length=0,
)
def log_failure(self, message, response_time):
self.environment.events.request_failure.fire(
request_type="my_task",
name=message,
response_time=response_time,
exception=None,
response_length=0,
)
设置负载测试参数：
在脚本中，可以设置一些负载测试的参数，如用户的并发数量、每秒的请求数等。
class MyUser(HttpUser):
tasks = [MyUser] # 定义要执行的用户类
wait_time = constant(1) # 设置每个用户的等待时间
host = "https://example.com" # 设置目标主机地址
min_wait = 5000 # 设置用户最小等待时间
max_wait = 10000 # 设置用户最大等待时间
users = 100 # 设置并发用户数
spawn_rate = 10 # 设置每秒新增用户数启动 Locust
locust -f my_script.py
访问 Locust Web 界面：
启动Locust后，可以通过访问Locust的Web界面进行监控和控制。默认情况下，Web界面监听在http://localhost:8089。
总之，Locust是一个功能强大、易用且灵活的负载测试工具，可用于模拟大规模用户并发访问，并评估系统的性能和稳定性。