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

Fixed database connection

Browse Source
This commit is contained in:
uberswe
2021-12-12 18:52:37 +01:00
parent 0511f8fbca
commit 18493611ec
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
database.go
View File

@ -8,7 +8,6 @@ import (
"gorm.io/driver/postgres" "gorm.io/driver/postgres"
"gorm.io/driver/sqlite" "gorm.io/driver/sqlite"
"gorm.io/gorm" "gorm.io/gorm"
"gorm.io/gorm/logger"
) )
func connectToDatabase(c config.Config) (db *gorm.DB, err error) { func connectToDatabase(c config.Config) (db *gorm.DB, err error) {
@ -21,9 +20,7 @@ func connectToDatabase(c config.Config) (db *gorm.DB, err error) {
db, err = gorm.Open(sqlite.Open(dsn), &gorm.Config{}) db, err = gorm.Open(sqlite.Open(dsn), &gorm.Config{})
} else if c.Database == "mysql" { } else if c.Database == "mysql" {
dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", c.DatabaseUsername, c.DatabasePassword, c.DatabaseHost, c.DatabasePort, c.DatabaseName) dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", c.DatabaseUsername, c.DatabasePassword, c.DatabaseHost, c.DatabasePort, c.DatabaseName)
db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{ db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
Logger: logger.Default.LogMode(logger.Info),
})
} else if c.Database == "postgres" { } else if c.Database == "postgres" {
dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=UTC", c.DatabaseHost, c.DatabaseUsername, c.DatabasePassword, c.DatabaseName, c.DatabasePort) dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=UTC", c.DatabaseHost, c.DatabaseUsername, c.DatabasePassword, c.DatabaseName, c.DatabasePort)
db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{}) db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
@ -34,7 +31,7 @@ func connectToDatabase(c config.Config) (db *gorm.DB, err error) {
} }
func migrateDatabase(db *gorm.DB) error { func migrateDatabase(db *gorm.DB) error {
err := db.AutoMigrate(&models.Token{}, &models.Session{}, &models.User{}, &models.Website{}) err := db.AutoMigrate(&models.User{}, &models.Token{}, &models.Session{}, &models.Website{})
seed(db) seed(db)
return err return err
} }