58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package health
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/redis/go-redis/v9"
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Handler struct {
|
|
db *gorm.DB
|
|
rdb *redis.Client
|
|
log *zap.Logger
|
|
}
|
|
|
|
func NewHandler(db *gorm.DB, rdb *redis.Client, log *zap.Logger) *Handler {
|
|
return &Handler{db: db, rdb: rdb, log: log}
|
|
}
|
|
|
|
// Handle godoc
|
|
// @Summary 健康检查
|
|
// @Description 检查 PostgreSQL 和 Redis 连接状态
|
|
// @Tags System
|
|
// @Produce json
|
|
// @Success 200 {object} object{status=string,database=string,redis=string,timestamp=integer,version=string,trace_id=string}
|
|
// @Router /health [get]
|
|
func (h *Handler) Handle(c *gin.Context) {
|
|
status := "ok"
|
|
dbStatus := "connected"
|
|
redisStatus := "connected"
|
|
|
|
sqlDB, err := h.db.DB()
|
|
if err != nil || sqlDB.Ping() != nil {
|
|
status = "degraded"
|
|
dbStatus = "disconnected"
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(c.Request.Context(), 2*time.Second)
|
|
defer cancel()
|
|
if h.rdb.Ping(ctx).Err() != nil {
|
|
status = "degraded"
|
|
redisStatus = "disconnected"
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"status": status,
|
|
"database": dbStatus,
|
|
"redis": redisStatus,
|
|
"timestamp": time.Now().Unix(),
|
|
"version": "0.1.0",
|
|
"trace_id": c.GetString("trace_id"),
|
|
})
|
|
}
|