jriou/restockbot
Archived
1
0
Fork 0
Code Issues Pull requests Activity

chore: Disable low-level database error messages

Browse source 
Signed-off-by: Julien Riou <julien@riou.xyz>
This commit is contained in:
Julien Riou 2022-09-02 08:49:19 +02:00
parent 0c81963131
commit e7c18d048e
No known key found for this signature in database
GPG key ID: FF42D23B580C89F7
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
db.go
View file

@ -5,21 +5,26 @@ import (
"gorm.io/driver/postgres"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
"gorm.io/gorm/logger"
)
var gconfig = &gorm.Config{
Logger: logger.Default.LogMode(logger.Silent),
}
// NewDatabaseFromConfig creates a database object from configuration structure
func NewDatabaseFromConfig(config DatabaseConfig) (*gorm.DB, error) {
switch config.Type {
case "postgres":
return gorm.Open(postgres.New(postgres.Config{DSN: config.DSN}), &gorm.Config{})
return gorm.Open(postgres.New(postgres.Config{DSN: config.DSN}), gconfig)
case "mysql":
return gorm.Open(mysql.New(mysql.Config{DSN: config.DSN}), &gorm.Config{})
return gorm.Open(mysql.New(mysql.Config{DSN: config.DSN}), gconfig)
default:
return gorm.Open(sqlite.Open(config.DSN), &gorm.Config{})
return gorm.Open(sqlite.Open(config.DSN), gconfig)
}
}
// NewDatabaseFromFile creates a database object from path to file (passed as argument)
func NewDatabaseFromFile(path string) (*gorm.DB, error) {
return gorm.Open(sqlite.Open(path), &gorm.Config{})
return gorm.Open(sqlite.Open(path), gconfig)
}