Adding Postgresql driver and basic config structures.
This commit is contained in:
@@ -1,8 +1,11 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/yxzzy-wtf/gin-gonic-prepack/config"
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
@@ -12,9 +15,24 @@ type Database struct {
|
||||
}
|
||||
|
||||
var Db *gorm.DB
|
||||
var Dialect gorm.Dialector
|
||||
|
||||
func InitDialect() gorm.Dialector {
|
||||
if config.Config().DbDialect == "sqlite" {
|
||||
return sqlite.Open(config.Config().DbUrl)
|
||||
} else if config.Config().DbDialect == "postgres" {
|
||||
return postgres.New(postgres.Config{
|
||||
DSN: fmt.Sprintf("user=%v password=%v dbname=%v port=%v sslmode=disable TimeZone=UTC",
|
||||
config.Config().DbUsername, config.Config().DbPasswordSecret, config.Config().DbName,
|
||||
config.Config().DbPort),
|
||||
})
|
||||
} else {
|
||||
panic("No valid DB config set up.")
|
||||
}
|
||||
}
|
||||
|
||||
func Init() *gorm.DB {
|
||||
db, err := gorm.Open(sqlite.Open("prepack.db"), &gorm.Config{})
|
||||
db, err := gorm.Open(InitDialect(), &gorm.Config{})
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user