frostfs-s3-gw/cmd/s3-gw/app-healthy.go
Roman Khimov e622f1987b Makefile: deduplicate binaries
Signed-off-by: Roman Khimov <roman@nspcc.ru>
2021-05-20 16:26:20 +03:00

45 lines
998 B
Go

package main
import (
"fmt"
"net/http"
"github.com/gorilla/mux"
)
// Healthy is a health check interface.
type Healthy interface {
Status() error
}
const (
healthyState = "NeoFS S3 Gateway is "
hdrContentType = "Content-Type"
defaultContentType = "text/plain; charset=utf-8"
)
func attachHealthy(r *mux.Router, h Healthy) {
healthy := r.PathPrefix(systemPath + "/-").
Subrouter().
StrictSlash(true)
healthy.HandleFunc("/ready", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set(hdrContentType, defaultContentType)
w.WriteHeader(http.StatusOK)
_, _ = fmt.Fprintln(w, healthyState+"ready")
})
healthy.HandleFunc("/healthy", func(w http.ResponseWriter, r *http.Request) {
code := http.StatusOK
msg := "healthy"
if err := h.Status(); err != nil {
msg = "unhealthy: " + err.Error()
code = http.StatusBadRequest
}
w.Header().Set(hdrContentType, defaultContentType)
w.WriteHeader(code)
_, _ = fmt.Fprintln(w, healthyState+msg)
})
}