RUNJ__IT # locust创建压测脚本

RUN IT

于 2020-12-23 09:37:38 发布

阅读量236

点赞数

CC 4.0 BY-SA版权

分类专栏： test

本文链接：https://blog.youkuaiyun.com/RUN__IT/article/details/111572422

test 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种使用Locust工具进行网站压力测试的方法。通过编写Python脚本，模拟大量用户并发访问，验证系统的稳定性和性能。示例脚本中包括了设置请求头、禁用安全警告等操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

压测脚本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"