add healthz check (#286)

* add healthz check

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

* add testing

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

gorush/server.go

@@ -32,6 +32,10 @@ func rootHandler(c *gin.Context) {
 	})
 }
 
+func heartbeatHandler(c *gin.Context) {
+	c.AbortWithStatus(200)
+}
+
 func pushHandler(c *gin.Context) {
 	var form RequestPush
 	var msg string
@@ -107,6 +111,7 @@ func routerEngine() *gin.Engine {
 	r.GET(PushConf.API.SysStatURI, sysStatsHandler)
 	r.POST(PushConf.API.PushURI, pushHandler)
 	r.GET(PushConf.API.MetricURI, metricsHandler)
+	r.GET("/healthz", heartbeatHandler)
 	r.GET("/", rootHandler)
 
 	return r

gorush/server_test.go

@@ -258,6 +258,16 @@ func TestMetricsHandler(t *testing.T) {
 		})
 }
 
+func TestHeartbeatHandler(t *testing.T) {
+	initTest()
+
+	r := gofight.New()
+
+	r.GET("/healthz").
+		Run(routerEngine(), func(r gofight.HTTPResponse, rq gofight.HTTPRequest) {
+			assert.Equal(t, http.StatusOK, r.Code)
+		})
+}
 func TestDisabledHTTPServer(t *testing.T) {
 	initTest()
 	PushConf.Core.Enabled = false