Merge pull request #46 from TheWisePigeon/44-retry-system-when-connec…

…ting-to-databases

retry 5 seconds later if failed to connect to database

internal/database/postgres.go

@@ -1,15 +1,30 @@
 package database
 
 import (
+	"fmt"
+	"os"
+	"time"
+
 	"github.com/jmoiron/sqlx"
 	_ "github.com/lib/pq"
-	"os"
 )
 
+func connect() (db *sqlx.DB, err error) {
+	db, err = sqlx.Connect("postgres", os.Getenv("PG_URL"))
+	return
+}
+
 func NewPostgresPool() *sqlx.DB {
-	db, err := sqlx.Connect("postgres", os.Getenv("PG_URL"))
+	var db *sqlx.DB
+	db, err := connect()
 	if err != nil {
-		panic(err)
+		fmt.Println("Failed to connect to database server. Retrying in 5 seconds")
+		time.Sleep(time.Second * 5)
+		db, err = connect()
+		if err != nil {
+			fmt.Println("Failed to connect to database")
+			panic(err)
+		}
 	}
 	db.SetConnMaxLifetime(0)
 	db.SetMaxIdleConns(50)