photoprism/internal/entity/db_migrate.go

46 lines
859 B
Go
Raw Normal View History

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
}