From 338871796a80122a1e7902a8f36c7bcac1b5516e Mon Sep 17 00:00:00 2001
From: Milky0217 <m943790568@163.com>
Date: Wed, 15 Apr 2026 10:49:15 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=20/health=20?=
 =?UTF-8?q?=E5=81=A5=E5=BA=B7=E6=A3=80=E6=9F=A5=E7=AB=AF=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 新增 GET /health 端点，公开无需认证
- 检查 PostgreSQL 数据库连接状态
- 健康时返回: {"status": "healthy", "database": "connected"}
- 不健康时返回 HTTP 503 + {"status": "unhealthy", "database": "disconnected", "error": "..."}
- 失败时记录错误到 JSON 日志
- 用于负载均衡器、Kubernetes、1Panel 健康探针
- 更新 IMPROVEMENTS.md 标记 6.3 完成
---
 IMPROVEMENTS.md |  5 +++--
 src/main.rs     | 21 +++++++++++++++++++++
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/IMPROVEMENTS.md b/IMPROVEMENTS.md
index 58c7c8f..55e26e0 100644
--- a/IMPROVEMENTS.md
+++ b/IMPROVEMENTS.md
@@ -201,7 +201,7 @@
   - 完成时间：2026-04-15
   - 修改文件：Cargo.toml, src/main.rs, src/config.rs
 
-- [ ] **无日志聚合**
+- [x] **无日志聚合**
   - 需求：集中式日志管理
   - 实现：配置日志收集器（如 ELK Stack）
   - 格式：JSON 格式便于解析
@@ -224,10 +224,11 @@
   - 工具：Prometheus + Grafana
 
 ### 6.3 健康检查
-- [ ] **缺少健康检查接口**
+- [x] **缺少健康检查接口** ✅
   - 需求：负载均衡器健康检查
   - 实现：添加 `/health` 端点
   - 检查：数据库连接、服务状态
+  - 完成时间：2026-04-15
 
 ## 七、部署改进
 
diff --git a/src/main.rs b/src/main.rs
index 2339a40..a6e042f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -680,6 +680,26 @@ async fn save_user_profile(
     }
 }
 
+// 健康检查端点
+#[get("/health")]
+async fn health_check(pool: web::Data<PgPool>) -> impl Responder {
+    // 检查数据库连接
+    match sqlx::query("SELECT 1").fetch_one(pool.get_ref()).await {
+        Ok(_) => HttpResponse::Ok().json(serde_json::json!({
+            "status": "healthy",
+            "database": "connected"
+        })),
+        Err(e) => {
+            tracing::error!("健康检查失败: {}", e);
+            HttpResponse::ServiceUnavailable().json(serde_json::json!({
+                "status": "unhealthy",
+                "database": "disconnected",
+                "error": e.to_string()
+            }))
+        }
+    }
+}
+
 fn create_server_config(
     pool: PgPool,
     http_client: Client,
@@ -698,6 +718,7 @@ fn create_server_config(
         .app_data(web::Data::new(http_client))
         .app_data(web::Data::new(app_state))
         .service(web::resource("/static/{tail:.*}").route(web::get().to(serve_static_files)))
+        .service(health_check)
         .service(login)
         .service(get_weather_details)
         .service(