Add endpoint for changing health status

Author: jskswamy

pkg/handler/handler.go

@@ -12,7 +12,11 @@ var statusCode = 200
 
 // Health return the dobby health status
 func Health(c *gin.Context) {
-	c.JSON(statusCode, gin.H{"healthy": true})
+	healthy := true
+	if statusCode != 200 {
+		healthy = false
+	}
+	c.JSON(statusCode, gin.H{"healthy": healthy})
 }
 
 // Version return dobby version
@@ -30,6 +34,18 @@ func Crash(_ *gin.Context) {
 	log.Fatal("you asked me do so, killing myself :-)")
 }
 
+// Sick will make dobby sick
+func Sick(c *gin.Context) {
+	statusCode = 500
+	c.JSON(200, gin.H{"status": "success"})
+}
+
+// Healthy will make dobby healthy again
+func Healthy(c *gin.Context) {
+	statusCode = 200
+	c.JSON(200, gin.H{"status": "success"})
+}
+
 func init() {
 	healthy, err := strconv.ParseBool(os.Getenv("HEALTH"))
 

pkg/server.go

@@ -13,5 +13,7 @@ func Run(bindAddress, port string) error {
 	r.GET("/health", handler.Health)
 	r.GET("/version", handler.Version)
 	r.PUT("/state/crash", handler.Crash)
+	r.PUT("/state/sick", handler.Sick)
+	r.PUT("/state/healthy", handler.Healthy)
 	return r.Run(fmt.Sprintf("%s:%s", bindAddress, port))
 }