You've already forked orderservice
64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type HealthHandler struct {
|
|
DB *sqlx.DB
|
|
Redis *redis.Client
|
|
}
|
|
|
|
func NewHealthHandler(db *sqlx.DB, redisDB *redis.Client) *HealthHandler {
|
|
return &HealthHandler{
|
|
DB: db,
|
|
Redis: redisDB,
|
|
}
|
|
}
|
|
|
|
type HealthResponse struct {
|
|
Status string `json:"status"`
|
|
Details map[string]string `json:"details"`
|
|
}
|
|
|
|
func (h *HealthHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
ctx, cancel := context.WithTimeout(r.Context(), 2*time.Second)
|
|
defer cancel()
|
|
|
|
details := map[string]string{}
|
|
|
|
if err := h.DB.PingContext(ctx); err != nil {
|
|
details["postgres"] = "unhealthy: " + err.Error()
|
|
} else {
|
|
details["postgres"] = "ok"
|
|
}
|
|
|
|
if err := h.Redis.Ping(ctx).Err(); err != nil {
|
|
details["redis"] = "unhealthy: " + err.Error()
|
|
} else {
|
|
details["redis"] = "ok"
|
|
}
|
|
|
|
status := "ok"
|
|
for _, v := range details {
|
|
if v != "ok" {
|
|
status = "unhealthy"
|
|
break
|
|
}
|
|
}
|
|
|
|
resp := HealthResponse{Status: status, Details: details}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
if status != "ok" {
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
}
|
|
_ = json.NewEncoder(w).Encode(resp)
|
|
}
|