Adding Postgresql driver and basic config structures.

This commit is contained in:
🐙PiperYxzzy
2022-10-13 07:06:50 +02:00
parent 2ada2b5936
commit 0f93876c8b
5 changed files with 191 additions and 4 deletions

View File

@@ -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)
}