Update node/health.go

Co-authored-by: Łukasz Magiera <magik6k@users.noreply.github.com>
filecoin-project · May 23, 2022 · b485203 · b485203
1 parent 444d0b1
commit b485203
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/node/health.go b/node/health.go
@@ -10,15 +10,19 @@ import (
 )
 
 type HealthHandler struct {
-	healthy bool
+	healthy int32
 }
 
 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) {
-	if !h.healthy {
+	if atomic.LoadInt32(&h.healthy) != 1 {
 		w.WriteHeader(http.StatusServiceUnavailable)
 		return
 	}