locust压测入门

不羁的心 提交于 2020-12-10 19:35:59

一、安装

pip install locust


了解locust相关命令:locust --help


二、获取当前CPU核心数 (后面需要根据核心数起进程)

pip3 install psutil

```
import pustil
print(pustil.cpu_count(False))
```


三、准备压测脚本(lc.py):

```
# 这是一个压测的临时脚本.
import requests
import json
import uuid
import time
from locust import between, task, constant
from locust.contrib.fasthttp import FastHttpUser
import random

# from utils.data import data
data = {
    "header": [["query"]],
    "data": [
        {"query": "后海大鲨鱼"},
        {"query": "这个衣服宽松不"},
        {"query": "包装好点"},
    ],
}


class WebsiteUser(FastHttpUser):
    wait_time = between(1, 1)
    host = "http://nta-api-ks.leyanbot.com"
    
    def body(self):
        test_data = data["data"]
        _index = random.randint(0, len(test_data) - 1)
        q = test_data[_index]["query"]
        request_id = str(uuid.uuid4())
        text = {"text": q}
        # form user 随机
        # role 2、3 随机
        # to-user 跟坐席一致
        role_list = [2, 3]
        _role_list_index = random.randint(0, 1)
        
        body = {
            "requestId": request_id,
            "createTime": int(time.time()),
            "generator": 1,
            "type": 1,
            "fromUser": {"nickName": f"LcTestUser-{request_id}", "role": 1, "openId": "sth-ApiTest"},
            "toUser": {"nickName": "lc-test", "role": role_list[_role_list_index], "openId": ""},
            "message": {"contentType": 1, "content": json.dumps(text)},
            "context": {"sellerId": 1724795549, "itemId": 0},
            "assistantOnlineStatus": 1,
        }
        return body

    @task
    def ks(self):
        data = self.body()
        with self.client.post(
            "/agent/conversation/ks", catch_response=True, json=data
        ) as response:
            if response.status_code == 200:
                print(response.json())
                response.success()
            else:
                try:
                    _resp = response.json()
                except Exception:
                    _resp = {"msg": "get response json error"}
                print(f"nick_name:{data['fromUser']['nickName']},code:{response.status_code}. resp:{_resp}")
                response.failure(f"code is not 200:{response.status_code}")
    
    def on_stop(self):
        self.client.cookies.clear()

```


方式一、通过web页面方式进行locust分布式压测

1、新建运行分布在多个进程中的locust,通过指定--master以下内容来启动主进程:

locust -f lc.py --master --web-host=127.0.0.1


注:无需加上端口,加上也不管用

前提:进入到lc.py压测脚本所在的目录

2、打开网址:http://127.0.0.1:8089,设置用户数等,开始压测,一般从10,20,50开始压测,然后视情况可逐渐加压至100,150,200,250,300,350,400,450,500等,加至目标值后连续压测半小时


注:压测时需关注压测链路中各服务器情况,QPS是否达到目标值,重点是响应时间P99。


3、起任意数量的分布式worker(相当于agent)--相当于从属进程:
locust -f lc.py --worker --master-host=127.0.0.1

注:运行分布式locust,必须在启动时指定主机(单台机器不需要这样做,因为主机默认是127.0.0.1)

一台电脑上可以起多个worker,如果本地起不了多个,可以通过其他电脑连接起mater机器的IP,然后再起worker,达到同时起多个worker的效果


4、结果查看
locust web端可以查看并下载压测结果

注:压测过程中如有失败,需将对应的requestid copy出来,然后通过日志查看对应的问题。


方式二、通过非web方式进行locust压测:

1、新建locust压测任务

locust -f lc.py --headless -u 100 -r 10

非web方式启动Locust,每秒加载10个user,总共加载100个,即:10秒后可加载完所有user,10秒后即可开始观察压测结果。

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!