1. from gevent import monkey
    2. monkey.patch_all()
    3. import time
    4. from locust import TaskSet, task, constant_throughput, wait_time
    5. from locust.contrib.fasthttp import FastHttpUser
    6. from logzero import logger
    9. class NotifyMsgApi(TaskSet):
    12.     def on_start(self):
    13.         self.headers = {
    14.             "Adapter-data": 'xxxx'
    16.         }
    18.     @task
    19.     def notify_msg(self):
    20.         params = {
    21.             "moduleType": "SHARE",
    22.             "moduleType": "LIVE",
    23.             "moduleType": "ALIGN",
    24.         }
    25.         start = time.perf_counter()
    26.         assert_data = ""
    27.         with self.client.get(
    28.             "/msg", headers=self.headers, params=params, catch_response=True, name="msg") as response:
    30.             if response.status_code == 200:
    31.                 response.success()
    32.                 # logger.info(response.json())
    33.             else:
    34.                 response.failure()
    35.                 logger.info(response.text)
    36.             assert_data = response.json().get('logId')
    37.         all_time = (time.perf_counter() - start) * 1000
    38.         if int(all_time) > 1000:
    39.             logger.info(f"all_time: {all_time}-- {assert_data}")
    41. class UserTest(FastHttpUser):
    42.     host = "http://xxxx:8906"
    43.     wait_time = constant_throughput(1.1)
    44.     tasks = [NotifyMsgApi]
    47. if __name__ == "__main__":
    48.     """
    49.     ps -ef | grep locust | grep -v grep | grep msg_perf.py | awk '{print $2}' | xargs kill -9
    50.     """
    51.     import os
    52.     from threading import Timer
    53.     import requests
    54.     import json
    55.     file_name = os.path.abspath(__file__)
    57.     logger.info("##################启动新压测进程########################")
    58.     os.system(f"nohup locust -f {file_name} --web-host=0.0.0.0 --master -P 8086 &")
    59.     for i in range(5):
    60.         os.system(f"nohup locust -f {file_name} --web-host=0.0.0.0 --worker &")
    63.     def stop():
    64.         logger.info("停止运行压测任务")
    65.         response = requests.get("http://x.x.x.x:8086/stop")
    66.         logger.info(f"operate stop perf task api: {response.text}")
    67.         logger.info("##################kill 进程########################")
    68.         if json.loads(response.text)["success"]:
    69.             os.system(
    70.                 "nohup ps -ef | grep locust | grep -v grep | grep %s | awk '{print $2}' | xargs kill -9 &" % file_name
    71.                 )
    74.     time = "00:30:20"   # 设置压测时间
    75.     run_time = sum(x * int(t) for x, t in zip([3600, 60, 1], time.split(":")))
    76.     logger.info(f"perf task run_time: {run_time}")
    77.     t = Timer(run_time, stop)
    78.     t.start()