RUNJ__IT # locust创建压测脚本

压测脚本locust的使用

# coding=utf-8
import requests
from locust import HttpUser as HttpLocust, TaskSet, task
from urllib3.exceptions import InsecureRequestWarning
# 禁用安全请求警告
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)


class MyBlogs(TaskSet):
    # 访问我的博客首页
    @task(1)
    def get_blog(self):
        # 定义请求头
        header = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36"
        }

        req = self.client.get("/imyalost", headers=header, verify=False)
        if req.status_code == 200:
            print("success")
        else:
            print("fails")


class websitUser(HttpLocust):
    tasks = [MyBlogs]
    min_wait = 3000  # 单位为毫秒
    max_wait = 6000  # 单位为毫秒


if __name__ == "__main__":
    import os
    # master启动
    print("locust -f {}\hj_test.py --master --host=https://www.cnblogs.com".format(os.getcwd()))
    # 节点机启动
    # "locust -f ./hj_test.py --worker --master-host=192.168.21.72"