From 6bf60726e01f03e82c6543e14dfa88afd4720108 Mon Sep 17 00:00:00 2001 From: Samuel Pua Date: Wed, 19 Oct 2022 20:52:29 +0800 Subject: [PATCH] Fix(database): Retry test database too --- backend/internal/common/database.go | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/backend/internal/common/database.go b/backend/internal/common/database.go index fc45857..d064b41 100644 --- a/backend/internal/common/database.go +++ b/backend/internal/common/database.go @@ -96,11 +96,18 @@ func TestDBInit() *gorm.DB { ) dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=Asia/Singapore", host, user, pass, "postgres", port, sslMode) - db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{ - Logger: newLogger, - }) - if err != nil { - fmt.Println("db err: (Init) ", err) + + var db *gorm.DB + var err error + + for ok := true; ok; ok = err != nil { + db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{ + Logger: newLogger, + }) + if err != nil { + fmt.Println("db err: (Init) ", err) + time.Sleep(1 * time.Second) + } } // Create Database