27 lines
462 B
Go
27 lines
462 B
Go
|
package services
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"git.klink.asia/paul/certman/models"
|
||
|
|
"git.klink.asia/paul/certman/settings"
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
var DB *gorm.DB
|
||
|
|
|
||
|
|
func InitDB() *gorm.DB {
|
||
|
|
dsn := settings.Get("DATABASE_URL", "db.sqlite3")
|
||
|
|
|
||
|
|
// Establish connection
|
||
|
|
db, err := gorm.Open("sqlite3", dsn)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("Could not open database: %s", err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Migrate models
|
||
|
|
db.AutoMigrate(models.User{}, models.ClientConf{})
|
||
|
|
|
||
|
|
return db
|
||
|
|
}
|