65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
# -*- 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()
|