test(#94): 性能压测套件 - Locust REST API梯度压测 + WebSocket长连接并发压测(100-5000) + 压测方案文档

2026-08-11 07:07:44 +08:00
parent 0f382ce2d0
commit d5f9efe766
6 changed files with 174 additions and 0 deletions
@@ -0,0 +1,59 @@
+"""REST API 压测脚本（Issue #94）- Locust
+
+覆盖完整用户路径：登录 -> 设备查询 -> 数据上报 -> 报表查询
+并发梯度：locust -u 10/50/100/500/1000 --spawn-rate 10
+
+运行：
+    pip install locust
+    locust -f tests/performance/locustfile.py --host http://127.0.0.1:8000 \
+           -u 100 --spawn-rate 10 --run-time 5m --headless \
+           --html reports/locust_$(date +%F_%H%M).html
+"""
+
+from locust import HttpUser, between, task
+
+
+class WaterManagementUser(HttpUser):
+    wait_time = between(0.5, 2.0)
+    token = None
+
+    def on_start(self):
+        """登录获取 token（若接口不存在则以匿名继续，保证压测可运行）"""
+        resp = self.client.post("/api/auth/login",
+                                json={"username": "perf", "password": "perf123"},
+                                catch_response=True)
+        if resp.status_code == 200:
+            try:
+                self.token = resp.json().get("data", {}).get("access_token")
+            except Exception:
+                self.token = None
+        else:
+            resp.success()  # 登录接口未实现时不计入失败
+
+    @property
+    def headers(self):
+        return {"Authorization": f"Bearer {self.token}"} if self.token else {}
+
+    @task(5)
+    def list_devices(self):
+        self.client.get("/api/devices?page=1&page_size=20", headers=self.headers,
+                        name="/api/devices")
+
+    @task(3)
+    def device_detail(self):
+        self.client.get("/api/devices/1", headers=self.headers, name="/api/devices/{id}")
+
+    @task(2)
+    def report_data(self):
+        self.client.post("/api/data/report",
+                         json={"device_id": 1, "metric": "flow", "value": 12.5},
+                         headers=self.headers, name="/api/data/report")
+
+    @task(2)
+    def billing_report(self):
+        self.client.get("/api/billing/report?month=2026-07", headers=self.headers,
+                        name="/api/billing/report")
+
+    @task(1)
+    def health(self):
+        self.client.get("/health", name="/health")