签入版本

2026-09-12 22:24:53 +08:00
commit 05cb22302c
863 changed files with 118186 additions and 0 deletions
@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*-
+"""
+健康检查脚本
+=============================================================
+
+验证应用可构建、健康端点可访问。
+
+用法：
+    python scripts/health_check.py [--host 127.0.0.1] [--port 8021]
+"""
+
+import os
+import sys
+
+import requests
+
+
+ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+if ROOT not in sys.path:
+    sys.path.insert(0, ROOT)
+
+
+def check_app_build() -> bool:
+    """验证应用工厂可构建。"""
+    try:
+        from src.wsgi import app  # noqa: F401
+        return True
+    except Exception as e:
+        print(f"[FAIL] 应用构建失败: {e}")
+        return False
+
+
+def check_health_endpoint(host: str, port: int) -> bool:
+    """验证 /health 端点可访问。"""
+    url = f"http://{host}:{port}/health"
+    try:
+        resp = requests.get(url, timeout=5)
+        ok = resp.status_code == 200 and resp.json().get("code") == 0
+        print(f"[{'OK' if ok else 'FAIL'}] {url} → {resp.status_code}")
+        return ok
+    except Exception as e:
+        print(f"[FAIL] {url} 访问失败: {e}")
+        return False
+
+
+def main():
+    host = "127.0.0.1"
+    port = 8021
+    args = sys.argv[1:]
+    for i, a in enumerate(args):
+        if a == "--host" and i + 1 < len(args):
+            host = args[i + 1]
+        if a == "--port" and i + 1 < len(args):
+            port = int(args[i + 1])
+
+    build_ok = check_app_build()
+    if build_ok:
+        # 端点检查（服务器未启动时优雅降级，不阻塞构建校验）
+        check_health_endpoint(host, port)
+    sys.exit(0 if build_ok else 1)
+
+
+if __name__ == "__main__":
+    main()