mirror of
https://github.com/Dvorinka/SEEN.git
synced 2026-06-04 20:43:03 +00:00
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type HealthHandler struct {
|
|
db *pgxpool.Pool
|
|
cache *redis.Client
|
|
}
|
|
|
|
func NewHealthHandler(db *pgxpool.Pool, cache *redis.Client) *HealthHandler {
|
|
return &HealthHandler{db: db, cache: cache}
|
|
}
|
|
|
|
func (h *HealthHandler) Live(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{"status": "ok", "timestamp": time.Now().UTC()})
|
|
}
|
|
|
|
func (h *HealthHandler) Ready(c *gin.Context) {
|
|
ctx, cancel := context.WithTimeout(c.Request.Context(), 2*time.Second)
|
|
defer cancel()
|
|
|
|
health := gin.H{"status": "ready"}
|
|
|
|
if err := h.db.Ping(ctx); err != nil {
|
|
health["status"] = "degraded"
|
|
health["postgres"] = err.Error()
|
|
c.JSON(http.StatusServiceUnavailable, health)
|
|
return
|
|
}
|
|
|
|
if err := h.cache.Ping(ctx).Err(); err != nil {
|
|
health["status"] = "degraded"
|
|
health["dragonfly"] = err.Error()
|
|
c.JSON(http.StatusServiceUnavailable, health)
|
|
return
|
|
}
|
|
|
|
health["postgres"] = "ok"
|
|
health["dragonfly"] = "ok"
|
|
c.JSON(http.StatusOK, health)
|
|
}
|