108 lines
3.0 KiB
Python
108 lines
3.0 KiB
Python
|
"""T11.1 Locust 性能基准脚本。
|
|||
|
|
|
|||
|
|
目标：对 T6.1 FastAPI 管理后台骨架做最小真实压测，验证计划中的
|
|||
|
|
"10 并发"目标，并记录各端点的延迟/错误率基线。
|
|||
|
|
|
|||
|
|
压测模型：
|
|||
|
|
- AdminBrowseUser：模拟已登录后台管理员，覆盖 /me /meta /orders /stats
|
|||
|
|
- LoginBurstUser：模拟登录高峰，持续打 /api/auth/login
|
|||
|
|
|
|||
|
|
运行示例：
|
|||
|
|
GAOKAO_JWT_SECRET=$(python3 - <<'PY'
|
|||
|
|
print('x'*64)
|
|||
|
|
PY
|
|||
|
|
) \
|
|||
|
|
locust -f locustfile.py --host http://127.0.0.1:18080 \
|
|||
|
|
--headless -u 10 -r 2 -t 1m --csv reports/perf/t11_1
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from locust import HttpUser, between, task
|
|||
|
|
|
|||
|
|
|
|||
|
|
ADMIN_USER = os.getenv("GAOKAO_ADMIN_USER", "admin")
|
|||
|
|
ADMIN_PASS = os.getenv("GAOKAO_ADMIN_PASS", "admin123")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AuthenticatedHttpUser(HttpUser):
|
|||
|
|
abstract = True
|
|||
|
|
wait_time = between(0.3, 1.2)
|
|||
|
|
token: Optional[str] = None
|
|||
|
|
|
|||
|
|
def on_start(self) -> None:
|
|||
|
|
self.login()
|
|||
|
|
|
|||
|
|
def login(self) -> None:
|
|||
|
|
with self.client.post(
|
|||
|
|
"/api/auth/login",
|
|||
|
|
json={"username": ADMIN_USER, "password": ADMIN_PASS},
|
|||
|
|
name="POST /api/auth/login",
|
|||
|
|
catch_response=True,
|
|||
|
|
) as resp:
|
|||
|
|
if resp.status_code != 200:
|
|||
|
|
resp.failure(f"login failed: {resp.status_code} {resp.text[:200]}")
|
|||
|
|
self.token = None
|
|||
|
|
return
|
|||
|
|
data = resp.json()
|
|||
|
|
token = data.get("access_token")
|
|||
|
|
if not token:
|
|||
|
|
resp.failure("login response missing access_token")
|
|||
|
|
self.token = None
|
|||
|
|
return
|
|||
|
|
self.token = token
|
|||
|
|
self.client.headers.update({"Authorization": f"Bearer {token}"})
|
|||
|
|
resp.success()
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AdminBrowseUser(AuthenticatedHttpUser):
|
|||
|
|
weight = 4
|
|||
|
|
|
|||
|
|
@task(5)
|
|||
|
|
def me(self) -> None:
|
|||
|
|
self.client.get("/api/auth/me", name="GET /api/auth/me")
|
|||
|
|
|
|||
|
|
@task(4)
|
|||
|
|
def meta(self) -> None:
|
|||
|
|
self.client.get("/api/meta", name="GET /api/meta")
|
|||
|
|
|
|||
|
|
@task(6)
|
|||
|
|
def list_orders(self) -> None:
|
|||
|
|
self.client.get("/api/orders?limit=50&offset=0", name="GET /api/orders")
|
|||
|
|
|
|||
|
|
@task(3)
|
|||
|
|
def order_stats(self) -> None:
|
|||
|
|
self.client.get("/api/stats/orders", name="GET /api/stats/orders")
|
|||
|
|
|
|||
|
|
@task(1)
|
|||
|
|
def detail_missing(self) -> None:
|
|||
|
|
with self.client.get(
|
|||
|
|
"/api/orders/non-existent-order-id",
|
|||
|
|
name="GET /api/orders/{id} [404]",
|
|||
|
|
catch_response=True,
|
|||
|
|
) as resp:
|
|||
|
|
if resp.status_code == 404:
|
|||
|
|
resp.success()
|
|||
|
|
else:
|
|||
|
|
resp.failure(f"expected 404, got {resp.status_code}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LoginBurstUser(HttpUser):
|
|||
|
|
weight = 1
|
|||
|
|
wait_time = between(1.0, 3.0)
|
|||
|
|
|
|||
|
|
@task(4)
|
|||
|
|
def login(self) -> None:
|
|||
|
|
self.client.post(
|
|||
|
|
"/api/auth/login",
|
|||
|
|
json={"username": ADMIN_USER, "password": ADMIN_PASS},
|
|||
|
|
name="POST /api/auth/login",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@task(1)
|
|||
|
|
def health(self) -> None:
|
|||
|
|
self.client.get("/health", name="GET /health")
|