Fix health check

echo/handler.go

@@ -0,0 +1,26 @@
+package echo
+
+import (
+	"github.com/core-go/mq/health"
+	"github.com/labstack/echo/v4"
+	"net/http"
+)
+
+type Handler struct {
+	Checkers []health.Checker
+}
+
+func NewHandler(checkers ...health.Checker) *Handler {
+	return &Handler{checkers}
+}
+
+func (c *Handler) Check() echo.HandlerFunc {
+	return func(ctx echo.Context) error {
+		result := health.Check(ctx.Request().Context(), c.Checkers)
+		if result.Status == health.StatusUp {
+			return ctx.JSON(http.StatusOK, result)
+		} else {
+			return ctx.JSON(http.StatusInternalServerError, result)
+		}
+	}
+}

echo_v3/handler.go

@@ -0,0 +1,26 @@
+package echo
+
+import (
+	"github.com/core-go/mq/health"
+	"github.com/labstack/echo"
+	"net/http"
+)
+
+type Handler struct {
+	Checkers []health.Checker
+}
+
+func NewHandler(checkers ...health.Checker) *Handler {
+	return &Handler{checkers}
+}
+
+func (c *Handler) Check() echo.HandlerFunc {
+	return func(ctx echo.Context) error {
+		result := health.Check(ctx.Request().Context(), c.Checkers)
+		if result.Status == health.StatusUp {
+			return ctx.JSON(http.StatusOK, result)
+		} else {
+			return ctx.JSON(http.StatusInternalServerError, result)
+		}
+	}
+}

gin/handler.go

@@ -0,0 +1,26 @@
+package gin
+
+import (
+	"github.com/core-go/mq/health"
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+type Handler struct {
+	Checkers []health.Checker
+}
+
+func NewHandler(checkers ...health.Checker) *Handler {
+	return &Handler{checkers}
+}
+
+func (c *Handler) Check() gin.HandlerFunc {
+	return func(ctx *gin.Context) {
+		result := health.Check(ctx.Request.Context(), c.Checkers)
+		if result.Status == health.StatusUp {
+			ctx.JSON(http.StatusOK, result)
+		} else {
+			ctx.JSON(http.StatusInternalServerError, result)
+		}
+	}
+}

health/check.go

@@ -7,7 +7,7 @@ const (
 	StatusDown = "DOWN"
 )
 
-func Check(ctx context.Context, services []HealthChecker) Health {
+func Check(ctx context.Context, services []Checker) Health {
 	health := Health{}
 	health.Status = StatusUp
 	healths := make(map[string]Health)

health/health_checker.go → health/checker.go

@@ -2,7 +2,7 @@ package health
 
 import "context"
 
-type HealthChecker interface {
+type Checker interface {
 	Name() string
 	Check(ctx context.Context) (map[string]interface{}, error)
 	Build(ctx context.Context, data map[string]interface{}, err error) map[string]interface{}

health/health_handler.go → health/handler.go

@@ -6,15 +6,15 @@ import (
 	"net/http"
 )
 
-type HealthHandler struct {
-	Checkers []HealthChecker
+type Handler struct {
+	Checkers []Checker
 }
 
-func NewHealthHandler(checkers ...HealthChecker) *HealthHandler {
-	return &HealthHandler{Checkers: checkers}
+func NewHandler(checkers ...Checker) *Handler {
+	return &Handler{Checkers: checkers}
 }
 
-func (c *HealthHandler) Check(w http.ResponseWriter, r *http.Request) {
+func (c *Handler) Check(w http.ResponseWriter, r *http.Request) {
 	ctx := context.Background()
 	h := Check(ctx, c.Checkers)
 	bytes, err := json.Marshal(h)