Add healtcheck endpoint

This commit is contained in:
2021-11-06 00:56:15 +01:00
parent db8372f6db
commit c952c8d3cf
2 changed files with 12 additions and 1 deletions

View File

@@ -96,6 +96,7 @@ func NewServer(cfg config.FrontendConfig, hs *ssh.HoneypotServer, store store.Lo
r.Get("/stats", s.HandlerStats)
r.Get("/stream", s.HandlerAttemptStream)
r.Get("/query", s.HandlerQuery)
r.Get("/health", s.HandlerHealth)
})
})
s.Handler = r
@@ -273,6 +274,15 @@ func (s *Server) HandlerQuery(w http.ResponseWriter, r *http.Request) {
}
}
func (s *Server) HandlerHealth(w http.ResponseWriter, r *http.Request) {
err := s.store.IsHealthy()
if err != nil {
s.WriteAPIError(w, r, http.StatusInternalServerError, fmt.Sprintf("Health error: %s", err))
return
}
w.Write([]byte(`{}`))
}
type APIErrorResponse struct {
Error string `json:"error"`
}