94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
|
package handler
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"dianshang/pkg/response"
|
|||
|
|
"net/http"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// HealthHandler 健康检查处理器
|
|||
|
|
type HealthHandler struct {
|
|||
|
|
db *gorm.DB
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewHealthHandler 创建健康检查处理器
|
|||
|
|
func NewHealthHandler(db *gorm.DB) *HealthHandler {
|
|||
|
|
return &HealthHandler{
|
|||
|
|
db: db,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// HealthResponse 健康检查响应结构
|
|||
|
|
type HealthResponse struct {
|
|||
|
|
Status string `json:"status"`
|
|||
|
|
Timestamp time.Time `json:"timestamp"`
|
|||
|
|
Version string `json:"version"`
|
|||
|
|
Database string `json:"database"`
|
|||
|
|
Uptime string `json:"uptime"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var startTime = time.Now()
|
|||
|
|
|
|||
|
|
// Health 健康检查接口
|
|||
|
|
// @Summary 健康检查
|
|||
|
|
// @Description 检查服务是否正常运行，包括数据库连接状态
|
|||
|
|
// @Tags 系统
|
|||
|
|
// @Accept json
|
|||
|
|
// @Produce json
|
|||
|
|
// @Success 200 {object} response.Response{data=HealthResponse} "服务正常"
|
|||
|
|
// @Failure 503 {object} response.Response "服务异常"
|
|||
|
|
// @Router /health [get]
|
|||
|
|
func (h *HealthHandler) Health(c *gin.Context) {
|
|||
|
|
// 检查数据库连接
|
|||
|
|
dbStatus := "connected"
|
|||
|
|
if h.db != nil {
|
|||
|
|
sqlDB, err := h.db.DB()
|
|||
|
|
if err != nil || sqlDB.Ping() != nil {
|
|||
|
|
dbStatus = "disconnected"
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
dbStatus = "not_configured"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 计算运行时间
|
|||
|
|
uptime := time.Since(startTime).String()
|
|||
|
|
|
|||
|
|
healthData := HealthResponse{
|
|||
|
|
Status: "healthy",
|
|||
|
|
Timestamp: time.Now(),
|
|||
|
|
Version: "1.0.0",
|
|||
|
|
Database: dbStatus,
|
|||
|
|
Uptime: uptime,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果数据库连接异常，返回503状态码
|
|||
|
|
if dbStatus != "connected" {
|
|||
|
|
healthData.Status = "unhealthy"
|
|||
|
|
c.JSON(http.StatusServiceUnavailable, response.Response{
|
|||
|
|
Code: http.StatusServiceUnavailable,
|
|||
|
|
Message: "服务异常",
|
|||
|
|
Data: healthData,
|
|||
|
|
})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
response.SuccessWithMessage(c, "服务正常", healthData)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Ping 简单的ping接口
|
|||
|
|
// @Summary Ping检查
|
|||
|
|
// @Description 简单的服务可用性检查
|
|||
|
|
// @Tags 系统
|
|||
|
|
// @Accept json
|
|||
|
|
// @Produce json
|
|||
|
|
// @Success 200 {object} response.Response "pong"
|
|||
|
|
// @Router /ping [get]
|
|||
|
|
func (h *HealthHandler) Ping(c *gin.Context) {
|
|||
|
|
response.SuccessWithMessage(c, "pong", gin.H{
|
|||
|
|
"timestamp": time.Now(),
|
|||
|
|
"message": "服务运行正常",
|
|||
|
|
})
|
|||
|
|
}
|