diff --git a/migrator.go b/migrator.go index ac06a144..d0b566b9 100644 --- a/migrator.go +++ b/migrator.go @@ -33,6 +33,7 @@ type ColumnType interface { type Migrator interface { // AutoMigrate AutoMigrate(dst ...interface{}) error + After(dst interface{}) error // Database CurrentDatabase() string diff --git a/migrator/migrator.go b/migrator/migrator.go index 016ebfc7..efa8ac7e 100644 --- a/migrator/migrator.go +++ b/migrator/migrator.go @@ -147,11 +147,18 @@ func (m Migrator) AutoMigrate(values ...interface{}) error { return err } } + if err := tx.Migrator().After(value); err != nil { + return err + } } return nil } +func (m Migrator) After(dst interface{}) error { + return nil +} + func (m Migrator) CreateTable(values ...interface{}) error { for _, value := range m.ReorderModels(values, false) { tx := m.DB.Session(&gorm.Session{})