diff --git a/chain.go b/chain.go index 8ccf4d7b..87c410bd 100644 --- a/chain.go +++ b/chain.go @@ -245,6 +245,11 @@ func (s *Chain) DropTable(value interface{}) *Chain { return s } +func (s *Chain) AutoMigrate(value interface{}) *Chain { + s.do(value).autoMigrate().exec() + return s +} + func (s *Chain) Unscoped() *Chain { s.unscoped = true return s diff --git a/do.go b/do.go index 44bedf22..b20c0841 100644 --- a/do.go +++ b/do.go @@ -726,6 +726,14 @@ func (s *Do) dropTable() *Do { return s } +func (s *Do) autoMigrate() *Do { + sql := fmt.Sprintf("SELECT column_name, data_type FROM information_schema.columns WHERE table_name = %v", s.tableName()) + for _, field := range s.model.fields("other") { + s.sql = fmt.Sprintf(sql+"and column_name = %v", field.DbName) + } + return s +} + func (s *Do) initializeWithSearchCondition() { m := Model{data: s.value, driver: s.driver} diff --git a/gorm_test.go b/gorm_test.go index f6697399..a1287afe 100644 --- a/gorm_test.go +++ b/gorm_test.go @@ -1235,3 +1235,7 @@ func TestTableName(t *testing.T) { t.Errorf("Cart's singular table name should be shopping_cart") } } + +func TestAutoMigration(t *testing.T) { + db.AutoMigrate(Address{}) +} diff --git a/main.go b/main.go index 8bf63aea..55efbf0e 100644 --- a/main.go +++ b/main.go @@ -114,3 +114,7 @@ func (s *DB) CreateTable(value interface{}) *Chain { func (s *DB) DropTable(value interface{}) *Chain { return s.buildChain().DropTable(value) } + +func (s *DB) AutoMigrate(value interface{}) *Chain { + return s.buildChain().AutoMigrate(value) +} diff --git a/model.go b/model.go index c522734c..9751908e 100644 --- a/model.go +++ b/model.go @@ -331,10 +331,6 @@ func (m *Model) returningStr() (str string) { return } -func (m *Model) missingColumns() (results []string) { - return -} - func (m *Model) setValueByColumn(name string, value interface{}, out interface{}) { data := reflect.Indirect(reflect.ValueOf(out)) setFieldValue(data.FieldByName(snakeToUpperCamel(name)), value)