Respect 'HEALTH' env variable when returning the health status

Author: jskswamy

pkg/handler/handler.go

@@ -4,11 +4,14 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/thecasualcoder/dobby/pkg/config"
 	"os"
+	"strconv"
 )
 
+var statusCode = 200
+
 // Health return the dobby health status
 func Health(c *gin.Context) {
-	c.JSON(200, gin.H{"healthy": true})
+	c.JSON(statusCode, gin.H{"healthy": true})
 }
 
 // Version return dobby version
@@ -20,3 +23,13 @@ func Version(c *gin.Context) {
 	}
 	c.JSON(200, gin.H{"version": version})
 }
+
+func init() {
+	healthy, err := strconv.ParseBool(os.Getenv("HEALTH"))
+
+	if err != nil {
+		statusCode = 200
+	} else if !healthy {
+		statusCode = 500
+	}
+}