Python locust 压测

Locust是纯运行在python语言下的测试工具，采用了gevent开发框架，底层利用greenlet微线程机制，单机并发可以支持非常高，直接HTTP等协议接口，特点就是协程，web可视化管理工具。

What is Locust? — Locust 2.10.3.dev11 documentation

from locust import task, TaskSet, HttpUser
 

class UserBehavior(TaskSet):

    @task
    def get_slots(self):
        header = {
            "Content-Type": "application/json"
        }
        data = {
            "query": "测试"
        }
        // /test根据自己情况而定
        response = self.client.post("/test", headers=header, json=data)
        print(response.json())
 
 
class WebsiteUser(HttpUser):
    tasks = [UserBehavior]
    min_wait = 1000 # 最小等待时间
    max_wait = 3000 # 最大等待时间
    host = "http://ip:port"  # 服务器地址，也可以在写命令时 指定host地址
 
 
if __name__ == '__main__':
    import os
    os.system('locust -f test/test_pressure.py')  # 此处导入os，可以在pycharm中直接运行此py文件