add healthcheck service
This commit is contained in:
55
internal/api/http/healthhandlers.go
Normal file
55
internal/api/http/healthhandlers.go
Normal file
@ -0,0 +1,55 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"git.loyso.art/frx/devsim/internal/probe"
|
||||
)
|
||||
|
||||
// ListenAndServe runs server to accept incoming connections. This function blocks on
|
||||
// handling connections.
|
||||
func livenessHandler(reporter probe.Reporter) http.HandlerFunc {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
switch reporter.ReportLiveness() {
|
||||
case probe.LivenessOk:
|
||||
w.WriteHeader(http.StatusOK)
|
||||
case probe.LivenessTimeout:
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
case probe.LivenessUnknown:
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func readinessHandler(reporter probe.Reporter) http.HandlerFunc {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
var status string
|
||||
var code int
|
||||
switch reporter.ReportReadiness() {
|
||||
case probe.ReadinessOk:
|
||||
status = "ok"
|
||||
code = http.StatusOK
|
||||
case probe.ReadinessFailed:
|
||||
status = "failed"
|
||||
code = http.StatusInternalServerError
|
||||
case probe.ReadinessNotReady:
|
||||
status = "not-ready"
|
||||
code = http.StatusOK
|
||||
case probe.ReadinessUnknown:
|
||||
status = "unknown"
|
||||
code = http.StatusOK
|
||||
}
|
||||
w.Header().Set("X-Readiness-Status", status)
|
||||
w.WriteHeader(code)
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user