health.go RawHistoryFolder

1// SPDX-License-Identifier: AGPL-3.0-or-later
2
3package httpserver
4
5import (
6	"net/http"
7
8	"github.com/bnema/zerowrap"
9)
10
11// healthHandler answers 200 "ok" when every check passes and 503 otherwise.
12// The failure is logged, never sent: the body stays the same for any cause.
13func healthHandler(checks []Check) http.HandlerFunc {
14	return func(w http.ResponseWriter, r *http.Request) {
15		w.Header().Set("Cache-Control", "no-store")
16		w.Header().Set("Content-Type", "text/plain; charset=utf-8")
17		for _, check := range checks {
18			if err := check(r.Context()); err != nil {
19				log := zerowrap.FromCtx(r.Context())
20				log.Warn().Err(err).Msg("health check failed")
21				w.WriteHeader(http.StatusServiceUnavailable)
22				_, _ = w.Write([]byte("unavailable\n"))
23				return
24			}
25		}
26		_, _ = w.Write([]byte("ok\n"))
27	}
28}