39 lines
657 B
Go
39 lines
657 B
Go
package postgres
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/aykhans/oh-my-chat/internal/adapter/config"
|
|
postgresDriver "gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func NewDB(config *config.PostgresConfig) (*gorm.DB, error) {
|
|
dsn := fmt.Sprintf(
|
|
"host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=%s",
|
|
config.Host,
|
|
config.User,
|
|
config.Password,
|
|
config.DBName,
|
|
config.Port,
|
|
"disable",
|
|
"UTC",
|
|
)
|
|
|
|
var db *gorm.DB
|
|
var err error
|
|
for range 3 {
|
|
db, err = gorm.Open(postgresDriver.Open(dsn), &gorm.Config{})
|
|
if err == nil {
|
|
break
|
|
}
|
|
time.Sleep(3 * time.Second)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, nil
|
|
}
|