go/golang-base-project
Archived
Template
1
0
Fork 0
Code Activity

Fixes #7 (#8)

Browse Source
This commit is contained in:
Markus Tenghamn
2021-12-24 08:44:04 +01:00
committed by GitHub
parent b5ba3d0387
commit 559059f54e
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
database.go
View File

@ -8,9 +8,26 @@ import (
"gorm.io/driver/postgres" "gorm.io/driver/postgres"
"gorm.io/driver/sqlite" "gorm.io/driver/sqlite"
"gorm.io/gorm" "gorm.io/gorm"
"time"
) )
func connectToDatabase(c config.Config) (db *gorm.DB, err error) { func connectToDatabase(c config.Config) (db *gorm.DB, err error) {
return connectLoop(c, 0)
}
func connectLoop(c config.Config, count int) (db *gorm.DB, err error) {
db, err = attemptConnection(c)
if err != nil {
if count > 300 {
return db, fmt.Errorf("could not connect to database after 300 seconds")
}
time.Sleep(1 * time.Second)
return connectLoop(c, count+1)
}
return db, err
}
func attemptConnection(c config.Config) (db *gorm.DB, err error) {
if c.Database == "sqlite" { if c.Database == "sqlite" {
// In-memory sqlite if no database name is specified // In-memory sqlite if no database name is specified
dsn := "file::memory:?cache=shared" dsn := "file::memory:?cache=shared"