feat: add production health smoke

2026-07-18 01:22:35 +08:00
parent 42e4e29ac1
commit 8f0cafbb67
3 changed files with 173 additions and 0 deletions
@@ -0,0 +1,73 @@
+#!/usr/bin/env python3
+"""Production smoke check for Jyotisha web deployment."""
+from __future__ import annotations
+
+import argparse
+import json
+import sys
+import time
+from urllib.error import HTTPError, URLError
+from urllib.request import Request, urlopen
+
+
+def fetch(url: str, timeout: float) -> tuple[int, str, float]:
+    started = time.monotonic()
+    request = Request(url, headers={"User-Agent": "jyotisha-production-smoke/1.0"})
+    try:
+        with urlopen(request, timeout=timeout) as response:
+            return response.status, response.read().decode("utf-8", "replace"), time.monotonic() - started
+    except HTTPError as error:
+        return error.code, error.read().decode("utf-8", "replace"), time.monotonic() - started
+    except URLError as error:
+        raise RuntimeError(str(error.reason)) from error
+
+
+def check(base_url: str, timeout: float) -> dict:
+    base = base_url.rstrip("/")
+    checks: list[dict] = []
+
+    status, body, elapsed = fetch(f"{base}/", timeout)
+    checks.append(
+        {
+            "name": "homepage",
+            "ok": status == 200 and ("Jyotisha" in body or "账户与出生资料" in body),
+            "status": status,
+            "latency_ms": round(elapsed * 1000),
+        }
+    )
+
+    status, body, elapsed = fetch(f"{base}/api/health", timeout)
+    health = json.loads(body) if body.strip().startswith("{") else {}
+    checks.append(
+        {
+            "name": "health",
+            "ok": status in {200, 503} and health.get("status") in {"ok", "degraded", "blocked"},
+            "status": status,
+            "latency_ms": round(elapsed * 1000),
+            "health_status": health.get("status"),
+            "checks": sorted((health.get("checks") or {}).keys()),
+        }
+    )
+
+    return {
+        "base_url": base,
+        "ok": all(item["ok"] for item in checks),
+        "checks": checks,
+    }
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--base-url", default="https://jyotisha.chat")
+    parser.add_argument("--timeout", type=float, default=8.0)
+    args = parser.parse_args()
+    try:
+        report = check(args.base_url, args.timeout)
+    except Exception as error:  # noqa: BLE001 - CLI smoke should report compact failure.
+        report = {"base_url": args.base_url, "ok": False, "error": str(error)}
+    print(json.dumps(report, ensure_ascii=False, indent=2, sort_keys=True))
+    return 0 if report.get("ok") else 1
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())