fixes

pkg/health/health.go

@@ -0,0 +1,178 @@
+package health
+
+import (
+	"context"
+	"net/http"
+	"sync"
+	"time"
+
+	json "github.com/goccy/go-json"
+	"github.com/lukaszraczylo/gohoarder/internal/version"
+)
+
+// Status represents component health status
+type Status string
+
+const (
+	StatusHealthy   Status = "healthy"
+	StatusUnhealthy Status = "unhealthy"
+	StatusDegraded  Status = "degraded"
+)
+
+// Check represents a single health check
+type Check struct {
+	Name   string                                 `json:"name"`
+	Status Status                                 `json:"status"`
+	Error  string                                 `json:"error,omitempty"`
+	Fn     func(context.Context) (Status, string) `json:"-"`
+}
+
+// Response is the health check response
+type Response struct {
+	Success  bool        `json:"success"`
+	Data     *HealthData `json:"data,omitempty"`
+	Metadata *Metadata   `json:"metadata,omitempty"`
+}
+
+// HealthData contains health check data
+type HealthData struct {
+	Status     Status                `json:"status"`
+	Version    string                `json:"version"`
+	Uptime     string                `json:"uptime"`
+	Components map[string]*Component `json:"components"`
+}
+
+// Component represents a system component
+type Component struct {
+	Status  Status                 `json:"status"`
+	Details map[string]interface{} `json:"details,omitempty"`
+	Error   string                 `json:"error,omitempty"`
+}
+
+// Metadata contains response metadata
+type Metadata struct {
+	RequestID string `json:"request_id"`
+	Timestamp string `json:"timestamp"`
+}
+
+// Checker manages health checks
+type Checker struct {
+	checks    []*Check
+	startTime time.Time
+	mu        sync.RWMutex
+}
+
+// New creates a new health checker
+func New() *Checker {
+	return &Checker{
+		checks:    make([]*Check, 0),
+		startTime: time.Now(),
+	}
+}
+
+// AddCheck adds a health check
+func (c *Checker) AddCheck(name string, fn func(context.Context) (Status, string)) {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+
+	c.checks = append(c.checks, &Check{
+		Name: name,
+		Fn:   fn,
+	})
+}
+
+// RunChecks runs all health checks
+func (c *Checker) RunChecks(ctx context.Context) *HealthData {
+	c.mu.RLock()
+	checks := make([]*Check, len(c.checks))
+	copy(checks, c.checks)
+	c.mu.RUnlock()
+
+	components := make(map[string]*Component)
+	overallStatus := StatusHealthy
+
+	for _, check := range checks {
+		status, errMsg := check.Fn(ctx)
+		components[check.Name] = &Component{
+			Status: status,
+			Error:  errMsg,
+		}
+
+		// Determine overall status
+		if status == StatusUnhealthy {
+			overallStatus = StatusUnhealthy
+		} else if status == StatusDegraded && overallStatus == StatusHealthy {
+			overallStatus = StatusDegraded
+		}
+	}
+
+	return &HealthData{
+		Status:     overallStatus,
+		Version:    version.Version,
+		Uptime:     time.Since(c.startTime).String(),
+		Components: components,
+	}
+}
+
+// HealthHandler returns an HTTP handler for health checks
+func (c *Checker) HealthHandler() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		ctx, cancel := context.WithTimeout(r.Context(), 5*time.Second)
+		defer cancel()
+
+		healthData := c.RunChecks(ctx)
+
+		response := Response{
+			Success: healthData.Status == StatusHealthy,
+			Data:    healthData,
+			Metadata: &Metadata{
+				RequestID: r.Header.Get("X-Request-ID"),
+				Timestamp: time.Now().UTC().Format(time.RFC3339),
+			},
+		}
+
+		statusCode := http.StatusOK
+		if healthData.Status == StatusUnhealthy {
+			statusCode = http.StatusServiceUnavailable
+		} else if healthData.Status == StatusDegraded {
+			statusCode = http.StatusOK // 200 but degraded
+		}
+
+		w.Header().Set("Content-Type", "application/json")
+		w.WriteHeader(statusCode)
+		json.NewEncoder(w).Encode(response)
+	}
+}
+
+// ReadyHandler returns an HTTP handler for readiness checks
+func (c *Checker) ReadyHandler() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		ctx, cancel := context.WithTimeout(r.Context(), 5*time.Second)
+		defer cancel()
+
+		healthData := c.RunChecks(ctx)
+
+		ready := healthData.Status != StatusUnhealthy
+
+		response := Response{
+			Success: ready,
+			Data: &HealthData{
+				Status:     healthData.Status,
+				Components: healthData.Components,
+			},
+			Metadata: &Metadata{
+				RequestID: r.Header.Get("X-Request-ID"),
+				Timestamp: time.Now().UTC().Format(time.RFC3339),
+			},
+		}
+
+		statusCode := http.StatusOK
+		if !ready {
+			statusCode = http.StatusServiceUnavailable
+		}
+
+		w.Header().Set("Content-Type", "application/json")
+		w.WriteHeader(statusCode)
+		json.NewEncoder(w).Encode(response)
+	}
+}