Add healthcheck

gaivota.go

@@ -201,3 +201,7 @@ type OrderStore interface {
 	// Update the Order in the store.
 	Update(context.Context, *Order) error
 }
+
+type HealthChecker interface {
+	Ping() (msg string, err error)
+}

pkg/mux/healthcheck.go

@@ -0,0 +1,34 @@
+package mux
+
+import (
+	"log"
+	"net/http"
+
+	"github.com/leoschet/gaivota"
+)
+
+type HealthCheck struct {
+	logger       *log.Logger
+	dependencies []gaivota.HealthChecker
+}
+
+func NewHealthCheck(logger *log.Logger, dependencies []gaivota.HealthChecker) *HealthCheck {
+	return &HealthCheck{
+		logger,
+		dependencies,
+	}
+}
+
+func (hc *HealthCheck) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
+	for _, dep := range hc.dependencies {
+		msg, err := dep.Ping()
+		if err != nil {
+			rw.WriteHeader(http.StatusInternalServerError)
+			rw.Write([]byte(msg))
+			return
+		}
+	}
+
+	rw.WriteHeader(http.StatusOK)
+	rw.Write([]byte("pong"))
+}

pkg/postgres/postgres.go

@@ -43,6 +43,16 @@ func (db *Database) Close() {
 	db.Pool.Close()
 }
 
+func (db *Database) Ping() (msg string, err error) {
+	err = db.Pool.Ping(context.Background())
+
+	if err != nil {
+		return "Could not connect to the Database", err
+	}
+
+	return "", nil
+}
+
 // func (db *Database) CreateTable(model interface{}) {
 // 	// Get how many properties are in the model
 // 	column_count := reflect.ValueOf(model).NumField()