30 lines
742 B
Go
30 lines
742 B
Go
package version_local
|
|
|
|
import (
|
|
"runtime"
|
|
|
|
goautomigrations "go-admin/app/goauto/migrations"
|
|
"go-admin/cmd/migrate/migration"
|
|
common "go-admin/common/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// #155 adds collection attempt auditing and the current attempt counter.
|
|
func init() {
|
|
_, fileName, _, _ := runtime.Caller(0)
|
|
migration.Migrate.SetVersion(migration.GetFilename(fileName), migrateCollectionTaskAttempt)
|
|
}
|
|
|
|
func migrateCollectionTaskAttempt(db *gorm.DB, version string) error {
|
|
return db.Transaction(func(tx *gorm.DB) error {
|
|
if err := goautomigrations.Migrate(tx); err != nil {
|
|
return err
|
|
}
|
|
if err := tx.Model(&common.Migration{}).Create(&common.Migration{Version: version}).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
}
|