听说locust是个不错的性能测试服务，相比较傻瓜式的工具，这个需要自己手动编写脚本，对于有一定代码经验的人来说还是一个不错的选择，同时支持二次开发，今天尝试写一个小demo

1、先用flask写一个测试接口，同时支持post和get两种请求

from flask import Flask, request
import json
app = Flask(__name__)
@app.route('/sum', methods=['POST'])
def sum():
    a = json.loads(request.data).get('a')
    b = json.loads(request.data).get('b')
    res = {"status": 200, "data": a + b}
    return json.dumps(res)

@app.route('/mul', methods=['GET'])
def mul():
    a = json.loads(request.args.get('a'))
    b = json.loads(request.args.get('b'))
    res = {"status": 200, "data": a * b}
    return json.dumps(res)

if __name__ == '__main__':
    app.run(host='192.168.72.1',port=8009)#192.168.72.1,是我本机的一个ip地址

启动服务

2、简单测试一下，不然跑不通也是白搭

import requests, json

url1 = 'http://192.168.72.1:8009/sum'
res = requests.post(url=url1, data=json.dumps({"a": 10, "b": 2}))
print(json.loads(res.content))
url2 = 'http://192.168.72.1:8009/mul?a=2&b=20'
res2 = requests.get(url=url2, )
print(json.loads(res2.content))
#返回结果：C:\Python37\python.exe D:/locust/aaaa.py
{'status': 200, 'data': 12}
{'status': 200, 'data': 40}
Process finished with exit code 0

好了，没问题，可以开始愉快的进行下面的步骤了

3、windows下直接下载安装就行了，傻瓜式操作，pip install locust

其实locust的本质就是模拟用户对接口进行请求，话不说多直接上代码：

from locust import HttpUser, TaskSet, task
# 忽略一些异常的警告
import urllib3
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
import json
class MyLocust(TaskSet):
    @task(1)
    def get_test(self):
        # 当然，ab的参数可以自己随便给，我这边为了方便，暂时写死
        url = '/mul?a=2&b=20'
        with self.client.get(url,verify=False,catch_response=True) as resposne:
            if resposne.status_code == 200:
                #支持手动标记成功或者失败
                resposne.success()
            else:
                resposne.failure('Failed')
    @task(2) #后面的参数代表权重
    def post_test(self):
        url='/sum'
        data={"a":2,"b":3}
        res=self.client.post(url,json.dumps(data),verify=False)
        print(res)

class websitUser(HttpUser):
    tasks = [MyLocust]
    min_wait = 1000  # 单位为毫秒
    max_wait = 3000 

启动服务：locust -f test_locust.py --host=http://192.168.72.1:8009，后面的地址就是我之前启动的本地flask服务的地址，就是要测试这个url的相关接口
我没有指定locust的服务地址，就是默认本地开启的服务：直接浏览器输入：http://localhost:8089/，将会展示这样的一个页面

image.png

image.png

image.png