27 lines
423 B
Go
27 lines
423 B
Go
package db
|
|
|
|
import (
|
|
"giter.top/smart/pkg/config"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func NewDB(cfg *config.Config) *gorm.DB {
|
|
driver := cfg.Data.Database.Driver
|
|
var db *gorm.DB
|
|
var err error
|
|
switch driver {
|
|
case "mysql":
|
|
// db, err = NewMySQLDB(cfg)
|
|
case "postgres":
|
|
db, err = NewPgSQLDB(cfg)
|
|
case "sqlite":
|
|
// return NewSQLiteDB(cfg)
|
|
default:
|
|
panic("unsupported driver")
|
|
}
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return db
|
|
} |