locust性能测试框架随笔
现在有很多的性能测试工具，比如说我们熟悉的loadrunner、jmeter、ab、webbench等等，这些工具如果对一个没用过的朋友来说，学习起来比较不容易，但是如果你能看懂python代码，会写就更好了，就可用尝试一下今天的主角Locust，一款基于python的性能测试工具，它的优点是学习起来比较简单，功能完全自定制，使用比较灵活，支持分布式。
所有的性能测试工具都至少包含这3块：
1、压力产生器，也就是可以指定产生多大的压力，多少并发；
2、数据统计，也就是结果的展示，要统计TPS是多少，响应时间多少等等，这些数据；
3、代理功能，代理功能呢说白了就一句话，分摊压力。比如说你压测的时候要用1000个并发，但是你的电脑（压力机）配置比较弱，只支持500并发，再大电脑就死掉了，完犊子，压测不了。那怎么办呢，就得分摊压力，再管你同事借一台电脑，俩电脑发压力，那就可以了，每个人的电脑上500个并发，人多好干活嘛，代理就是干这个的，把这个上面的压力分摊到别的电脑上。
当然Locust这3个功能都是有的。
Locust是python的一个第三方模块，安装很简单，直接pip install locust即可，或者自己下载安装包，手动安装。安装完成后，就有locust命令，在命令行里面输入 locust --help，有帮助信息就安装成功了。
locust的官网是 www.locust.io，上面也有怎么安装和例子。
怎么用呢，很简单，只需要几行代码就可以实现，并且有漂亮的web界面，可以设置并发数，和查看结果，首先我们写几行简单的代码
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
from locust import HttpLocust, TaskSet, task
#HttpLocust 这个类的作用是用来发送http请求的
#TaskSet 这个类是定义用户行为的，相当于loadrunnerhttp协议的脚本，jmeter里面的http请求一样，要去干嘛的
#task 这个task是一个装饰器，它用来把一个函数，装饰成一个任务，也可以指定他们的先后执行顺序
class BestTest(TaskSet):
#自己定义的类，继承TaskSet，也就是这个类是实现咱们要去请求什么的
@task#用task装饰器把这个函数装饰成一个咱们要执行的性能任务
def index(self):#这个函数里面定义的是咱们要具体做的操作
self.client.get('/')#请求这个url里面的哪个路径，如果是接口的话，就是哪个接口
class BestTestIndexUser(HttpLocust):
#这个类继承了HttpLocust，代表每个并发里面的每个用户
task_set = BestTest #这个是每个用户都去干什么，指定了BestTest这个类，它就会每个用户去运行besttest这个类里面的方法
|
代码写好了，下面运行一下，看看结果，但是运行的时候不是像一个普通的python脚本一样运行了， 得使用locust命令，在命令行里运行
|
1
2
3
|
locust -f besttest.py --host=http://www.besttest.cn
#-f是指定一个python文件 后面跟上咱们刚才写的python文件
#--host是你要访问哪个网站，后面跟网站的url
|
这样就可以运行，运行完之后，就可以在浏览器里面打开locust的控制台了，locust的默认端口号是8089，运行完之后，访问的时候用ip:8090就可以访问了，如下图：
启动locust：
locust控制台页面：
开始测试后的页面：
tps图和响应时间图：
但是locust不像loadrunner和jmeter一样可以设置开始时间和结束时间，只能你自己点stop的时候停止测试，这一点比较烦，所以在使用过程中还是需要自己注意一下测试时间。
刚才上面写的例子是单个接口压测， 或者更说单场景的，如果想做混合场景的压测，比如说是一个业务流程，先登录、然后下单、购物这样的怎么办呢？
只需要写多个task就可以了，也就是在类里面写多个函数，想谁想执行，标上数字就行了，1,2,3,4代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
from locust import HttpLocust, TaskSet, task
class BestTest(TaskSet):
@task(1)#给task装饰器传一个参数，代表先访问首页
def index(self):#首页
self.client.get('/')
#发get请求
@task(2)
def login(self):#登录
self.client.post('/login',{'username':'besttest','password':'123456'})
#发送post请求，第一个是路径，第二个这个接口的入参，账号和密码
class BestTestIndexUser(HttpLocust):
#这个类继承了HttpLocust，代表每个并发里面的每个用户
task_set = BestTest #这个是每个用户都去干什么，指定了BestTest这个类，它就会每个用户去运行besttest这个类里面的方法
|
