70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package health
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
|
|
"hotgo/internal/library/queue"
|
|
)
|
|
|
|
type response struct {
|
|
Status string `json:"status"`
|
|
Checks map[string]string `json:"checks"`
|
|
}
|
|
|
|
// Serve exposes an unauthenticated dependency health endpoint for container probes.
|
|
func Serve(r *ghttp.Request) {
|
|
if r.Method != http.MethodGet && r.Method != http.MethodHead {
|
|
r.Response.Header().Set("Allow", http.MethodGet+", "+http.MethodHead)
|
|
r.Response.WriteStatusExit(http.StatusMethodNotAllowed, "method not allowed")
|
|
return
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(r.Context(), 2*time.Second)
|
|
defer cancel()
|
|
checks := map[string]string{}
|
|
statusCode := http.StatusOK
|
|
|
|
if err := g.DB().Ctx(ctx).Raw("SELECT 1").Scan(new(int)); err != nil {
|
|
checks["mysql"] = err.Error()
|
|
statusCode = http.StatusServiceUnavailable
|
|
} else {
|
|
checks["mysql"] = "ok"
|
|
}
|
|
|
|
if _, err := g.Redis().Do(ctx, "PING"); err != nil {
|
|
checks["redis"] = err.Error()
|
|
statusCode = http.StatusServiceUnavailable
|
|
} else {
|
|
checks["redis"] = "ok"
|
|
}
|
|
|
|
if strings.EqualFold(strings.TrimSpace(g.Cfg().MustGet(ctx, "queue.driver").String()), "kafka") {
|
|
if err := queue.KafkaHealthCheck(ctx); err != nil {
|
|
checks["kafka"] = err.Error()
|
|
statusCode = http.StatusServiceUnavailable
|
|
} else {
|
|
checks["kafka"] = "ok"
|
|
}
|
|
} else {
|
|
checks["kafka"] = "disabled"
|
|
}
|
|
|
|
body := response{Status: "ok", Checks: checks}
|
|
if statusCode != http.StatusOK {
|
|
body.Status = "error"
|
|
}
|
|
r.Response.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
r.Response.WriteHeader(statusCode)
|
|
if r.Method != http.MethodHead {
|
|
r.Response.WriteJsonExit(body)
|
|
return
|
|
}
|
|
r.Exit()
|
|
}
|