45 lines
859 B
Go
45 lines
859 B
Go
package entity
|
|
|
|
import (
|
|
"time"
|
|
)
|
|
|
|
// MigrateDb creates database tables and inserts default fixtures as needed.
|
|
func MigrateDb(dropDeprecated, runFailed bool) {
|
|
start := time.Now()
|
|
|
|
if dropDeprecated {
|
|
DeprecatedTables.Drop(Db())
|
|
}
|
|
|
|
Entities.Migrate(Db(), runFailed)
|
|
Entities.WaitForMigration(Db())
|
|
|
|
CreateDefaultFixtures()
|
|
|
|
log.Debugf("entity: successfully initialized [%s]", time.Since(start))
|
|
}
|
|
|
|
// InitTestDb connects to and completely initializes the test database incl fixtures.
|
|
func InitTestDb(driver, dsn string) *Gorm {
|
|
if HasDbProvider() {
|
|
return nil
|
|
}
|
|
|
|
if driver == "test" || driver == "sqlite" || driver == "" || dsn == "" {
|
|
driver = "sqlite3"
|
|
dsn = ".test.db"
|
|
}
|
|
|
|
log.Infof("initializing %s test db in %s", driver, dsn)
|
|
|
|
db := &Gorm{
|
|
Driver: driver,
|
|
Dsn: dsn,
|
|
}
|
|
|
|
SetDbProvider(db)
|
|
ResetTestFixtures()
|
|
|
|
return db
|
|
}
|