fix

covrom · Feb 2, 2021 · 95ac0a7 · 95ac0a7
1 parent 92a22ed
commit 95ac0a7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -21,3 +21,4 @@ sql/test_ddl_full.sql
 schema/yaml_test.go
 schema2.puml
 schema2.yaml
+.vscode/launch.json
diff --git a/schema/diff.go b/schema/diff.go
@@ -364,6 +364,19 @@ func (s *PatchSchema) Build(from, to *Schema) {
 			}
 			pt.columns = append(pt.columns, pc)
 		}
+		if rt != nil {
+			for _, c := range rt.Columns {
+				pc := &PatchColumn{
+					tableName: t.Name,
+					to:        c,
+				}
+				tc, err := t.FindColumnByName(c.Name)
+				if err == nil {
+					pc.from = tc
+				}
+				pt.columns = append(pt.columns, pc)
+			}
+		}
 		for _, idx := range t.Indexes {
 			pi := &PatchIndex{
 				from: idx,