From b0a3e4389ef6218c5018e34713bd8a868bb70f6a Mon Sep 17 00:00:00 2001
From: Sivachandran Paramasivam <sivachandran.p@gmail.com>
Date: Wed, 20 Jan 2021 11:17:43 +0530
Subject: [PATCH 1/2] Ping DB at startup

Add Ping function to DBInterface and calls it at startup.
---
 main.go                | 8 ++++++++
 repository/db.go       | 1 +
 repository/postgres.go | 5 +++++
 3 files changed, 14 insertions(+)

diff --git a/main.go b/main.go
index d64c912..6b5037a 100644
--- a/main.go
+++ b/main.go
@@ -6,11 +6,19 @@ import (
 	"clamp-core/listeners"
 	"clamp-core/migrations"
 	"clamp-core/models"
+	"clamp-core/repository"
+
 	"log"
 	"os"
 )
 
 func main() {
+	log.Println("Pinging DB...")
+	err := repository.GetDB().Ping()
+	if err != nil {
+		log.Fatalf("DB ping failed: %s", err)
+	}
+
 	var cliArgs models.CLIArguments = os.Args[1:]
 	os.Setenv("PORT", config.ENV.PORT)
 	migrations.Migrate()
diff --git a/repository/db.go b/repository/db.go
index 5328cd0..8958ebf 100644
--- a/repository/db.go
+++ b/repository/db.go
@@ -21,6 +21,7 @@ type DBInterface interface {
 	GetWorkflows(pageNumber int, pageSize int, sortBy models.SortByFields) ([]models.Workflow, int, error)
 	FindServiceRequestsByWorkflowName(workflowName string, pageNumber int, pageSize int) ([]models.ServiceRequest, error)
 	DeleteWorkflowByName(string) error
+	Ping() error
 }
 
 var db DBInterface
diff --git a/repository/postgres.go b/repository/postgres.go
index 98b2544..d291025 100644
--- a/repository/postgres.go
+++ b/repository/postgres.go
@@ -196,6 +196,11 @@ func (p *postgres) GetWorkflows(pageNumber int, pageSize int, sortFields models.
 	return workflows, totalWorkflowsCount, err
 }
 
+func (p *postgres) Ping() error {
+	_, err := p.getDb().Exec("SELECT 1")
+	return err
+}
+
 func (p *postgres) getDb() *pg.DB {
 	singletonOnce.Do(func() {
 		log.Println("Connecting to DB")

From b2677a4c5c58e566ba9940e365cb600f40f5b617 Mon Sep 17 00:00:00 2001
From: Sivachandran Paramasivam <sivachandran.p@gmail.com>
Date: Wed, 20 Jan 2021 11:38:28 +0530
Subject: [PATCH 2/2] Add Ping to mockDB to fix test breakage

---
 services/abstract_test.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/services/abstract_test.go b/services/abstract_test.go
index 15b4b2a..26b4480 100644
--- a/services/abstract_test.go
+++ b/services/abstract_test.go
@@ -75,6 +75,10 @@ func (m mockDB) FindAllStepStatusByServiceRequestIDAndStepID(serviceRequestID uu
 	return findAllStepStatusByServiceRequestIDAndStepIDMock(serviceRequestID, stepID)
 }
 
+func (m mockDB) Ping() error {
+	return nil
+}
+
 func init() {
 	repository.SetDb(&mockDB{})
 }