locust性能测试工具怎么在python中使用

发布时间：2020-12-29 09:14:05 作者：Leah
来源：亿速云阅读：222

本篇文章为大家展示了locust性能测试工具怎么在python中使用，内容简明扼要并且容易理解，绝对能使你眼前一亮，通过这篇文章的详细介绍希望你能有所收获。

一、简介

　　Locust 是一个易于使用，分布式，用户负载测试工具。它用于负载测试 web 站点(或其他系统)，并计算出一个系统可以处理多少并发用户。在测试期间，一大群虚拟用户访问你的网站。每个测试用户的行为由您定义，集群过程由 web UI 实时监控。这将帮助您在让真正的用户进入之前进行测试并识别代码中的瓶颈。

　　Locust 完全是基于事件的，因此在一台机器上支持数千个并发用户是可能的。与许多其他基于事件的应用程序不同，它不使用回调。相反它通过 gevent 使用轻量级协程。这允许您用 Python 编写非常有表现力的场景，而不用回调使代码复杂化。

二、安装

用 pip 管理工具安装：

pip3 install locust

检查是否安装成功，执行命令：locust --v

locust性能测试工具怎么在python中使用

三、压测过程

1.编写脚本

Locust 不同于 jmeter 可以用 GUI 来创建压测脚本。Locust 需要自己编写 python 脚本，压测负载脚本主要包含两个子类UserTask和WebsiteUser，分别继承TaskSet和Httplocust类，拥有这两个父类的公共属性和方法。

from locust import HttpLocust, TaskSet, task, between
import os,json

# 定义用户行为
class UserTask(TaskSet):
  def on_start(self):
    '''初始化数据，每个虚拟用户只执行一次'''
    self.client.post("/login",{"username":"test","password":"123456"})

  @task(2)
  def home_index(self):
    r = self.client.get("/sz/Home/DefaultHomeV2Request")
    assert json.loads(r.text)['Error'] == 0

  @task(1)
  def sale(self):
    self.client.get("/sz/Home/FlashSaleRequest")
    assert json.loads(r.text)['Error'] == 0

  def on_stop(self):
    '''销毁数据，每个虚拟用户只执行一次'''
    self.client.post("/SignOut",{"CustomerGuid":"c7d7e646-9ce2-499b-a22e-a3c98d4545fe"})

class WebsiteUser(HttpLocust):
  host = 'http://10.1.62.126'
  task_set = UserTask
  wait_time = between(3, 5)


if __name__ == "__main__":
  os.system('locust -f stress_test.py ')

locust 运行时：