50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package health
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"log/slog"
|
|
"net/http"
|
|
"time"
|
|
|
|
"dashboard-cdc-worker/internal/monitor"
|
|
)
|
|
|
|
type Server struct {
|
|
server *http.Server
|
|
}
|
|
|
|
func New(addr string, metrics *monitor.Metrics) *Server {
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/health", func(w http.ResponseWriter, _ *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
payload := map[string]interface{}{"code": "ok"}
|
|
if metrics != nil {
|
|
payload["metrics"] = metrics.Snapshot()
|
|
}
|
|
_ = json.NewEncoder(w).Encode(payload)
|
|
})
|
|
mux.HandleFunc("/metrics", func(w http.ResponseWriter, _ *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
if metrics == nil {
|
|
_ = json.NewEncoder(w).Encode(map[string]string{"code": "ok"})
|
|
return
|
|
}
|
|
_ = json.NewEncoder(w).Encode(metrics.Snapshot())
|
|
})
|
|
return &Server{server: &http.Server{Addr: addr, Handler: mux, ReadHeaderTimeout: 5 * time.Second}}
|
|
}
|
|
|
|
func (s *Server) Start(logger *slog.Logger) {
|
|
go func() {
|
|
if err := s.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
|
logger.Error("health server failed", "error", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (s *Server) Shutdown(ctx context.Context) error {
|
|
return s.server.Shutdown(ctx)
|
|
}
|