forked from nixcn/nixcn-cms
40 lines
816 B
Go
40 lines
816 B
Go
package data
|
|
|
|
import (
|
|
"nixcn-cms/config"
|
|
"nixcn-cms/data/drivers"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var Database *drivers.DBClient
|
|
|
|
func Init() {
|
|
// Init database
|
|
dbType := config.Get("database.type").(string)
|
|
exDSN := drivers.ExternalDSN{
|
|
Host: config.Get("database.host").(string),
|
|
Name: config.Get("database.name").(string),
|
|
Username: config.Get("database.username").(string),
|
|
Password: config.Get("database.password").(string),
|
|
}
|
|
|
|
if dbType != "postgres" {
|
|
log.Fatal("[Database] Only support postgras db!")
|
|
}
|
|
|
|
// Conect to db
|
|
db, err := drivers.Postgres(exDSN)
|
|
if err != nil {
|
|
log.Fatal("[Database] Error connecting to db!")
|
|
}
|
|
|
|
// Auto migrate
|
|
err = db.DB.AutoMigrate(&User{})
|
|
if err != nil {
|
|
log.Error("[Database] Error migrating database: ", err)
|
|
}
|
|
|
|
Database = db
|
|
}
|