Add health check; pings the DB

movetokube · Nov 22, 2023 · 1d026f2 · 1d026f2
1 parent 366d4a2
commit 1d026f2
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 1 deletion.
diff --git a/cmd/main.go b/cmd/main.go
@@ -17,6 +17,7 @@ package main
 
 import (
 	"flag"
+	"net/http"
 	"os"
 
 	// Import all Kubernetes client auth plugins (e.g. Azure, GCP, OIDC, etc.)
@@ -125,7 +126,9 @@ func main() {
 	}
 	//+kubebuilder:scaffold:builder
 
-	if err := mgr.AddHealthzCheck("healthz", healthz.Ping); err != nil {
+	if err := mgr.AddHealthzCheck("healthz", func(req *http.Request) error {
+		return pg.Ping(req.Context())
+	}); err != nil {
 		setupLog.Error(err, "unable to set up health check")
 		os.Exit(1)
 	}

diff --git a/internal/postgres/mock/postgres.go b/internal/postgres/mock/postgres.go
diff --git a/internal/postgres/postgres.go b/internal/postgres/postgres.go
@@ -1,6 +1,7 @@
 package postgres
 
 import (
+	"context"
 	"database/sql"
 	"fmt"
 	"log"
@@ -25,6 +26,7 @@ type PG interface {
 	GetUser() string
 	GetHost() string
 	GetDefaultDatabase() string
+	Ping(ctx context.Context) error
 }
 
 type pg struct {
@@ -81,6 +83,10 @@ func (c *pg) GetDefaultDatabase() string {
 	return c.default_database
 }
 
+func (c *pg) Ping(ctx context.Context) error {
+	return c.db.PingContext(ctx)
+}
+
 func GetConnection(user, password, host, database, uri_args string) (*sql.DB, error) {
 	db, err := sql.Open("postgres", fmt.Sprintf("postgresql://%s:%s@%s/%s?%s", user, password, host, database, uri_args))
 	if err != nil {