locust是一款性能测试框架，比JMeter好的地方在于能产生更多的请求，因为locust是基于协程的，而JMeter是基于线程的。

安装

1	pip install locust

安装成功后查看版本信息：

1 2	locust -V locust 1.4.3

编写代码

from locust import HttpUser, TaskSet, task, between, tag

# 模拟n个用户，其实就是创建了n个MyTask的实例
class MyTask(TaskSet):
    
    token = ''
    
    def on_start(self):
        self.login_or_register()
        
    def login_or_register(self):
        url = 'http://api'
        body = {'id': 1}
        headers = {'token': 'xxxx'}
        with self.client.post(url, json=body, headers=headers, name='login_or_register') as response:
            if response.text != '':
                res_dict = json.loads(response.text)
                if res_dict['code'] == 0:
                    pass
    
    @task
    @tag('view_page')                
    def view_page(self):
        pass
                            
        
# 这个类是一个配置类, 只被初始化一次, 在这里写一些配置相关信息
# 要访问哪些id, 要生成哪些用户的token, 可以在这里生成
class MyUser(HttpUser):

    # 测试的api的host信息
    host = ''
    # 每次访问之间间隔的时间
    wait_time = between(1, 2)

运行

执行命令

1	locust -f file.py

如果只执行带有tag的业务代码

1	locust -f file.py --tags view_page

如果要进行分布式测试，首先启动master，master是管理节点，在这个节点上面查看统计数据。

1	locust -f file.py --tags view_page --master

再开两个worker节点, 如果主节点和从节点不在一台机器上，那需要指定master-host参数。

1	locust -f file.py --tags view_page --worker --master-host=192.168.1.2

当worker节点启动后，master的console里面会显示：

1
2

[2021-03-05 18:42:39,333] /INFO/locust.runners: Client '1140b3a3ec57404194845264ee8dae1f' reported as ready. Currently 1 clients ready to swarm.
[2021-03-05 18:45:13,222] /INFO/locust.runners: Client '8f1e5206c38f4f99a55641f1cf0c44d1' reported as ready. Currently 2 clients ready to swarm.

打开压测页面，http://localhost:8089/

Finology 大数据金融

locust性能测试框架

安装

编写代码

运行