cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

Merge remote-tracking branch 'origin/feat/health-endpoints' into feat/health-endpoints

Browse Source
This commit is contained in:
Cory Schwartz 2022-05-23 11:14:34 -07:00
commit 0a7fbd0fce
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
node/health.go
View File

@ -13,15 +13,19 @@ import (
var healthlog = logging.Logger("healthcheck") var healthlog = logging.Logger("healthcheck")
type HealthHandler struct { type HealthHandler struct {
healthy bool healthy int32
} }
func (h *HealthHandler) SetHealthy(healthy bool) { func (h *HealthHandler) SetHealthy(healthy bool) {
h.healthy = healthy h := int32(0)
if healthy {
h = 1
}
atomic.StoreInt32(&h.healthy, h)
} }
func (h *HealthHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) { func (h *HealthHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if !h.healthy { if atomic.LoadInt32(&h.healthy) != 1 {
w.WriteHeader(http.StatusServiceUnavailable) w.WriteHeader(http.StatusServiceUnavailable)
return return
} }