Merge pull request #226 from k1LoW/fix-hide-excluded-table-link

Remove relations of excluded tables
k1LoW · May 1, 2020 · 2723fa2 · 2723fa2
2 parents 78da408 + 00dc30b
commit 2723fa2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/config/config.go b/config/config.go
@@ -346,7 +346,7 @@ func excludeTableFromSchema(name string, s *schema.Schema) error {
 			// ChildRelations
 			childRelations := []*schema.Relation{}
 			for _, r := range c.ChildRelations {
-				if r.Table.Name != name {
+				if r.Table.Name != name && r.ParentTable.Name != name {
 					childRelations = append(childRelations, r)
 				}
 			}
@@ -355,7 +355,7 @@ func excludeTableFromSchema(name string, s *schema.Schema) error {
 			// ParentRelations
 			parentRelations := []*schema.Relation{}
 			for _, r := range c.ParentRelations {
-				if r.Table.Name != name {
+				if r.Table.Name != name && r.ParentTable.Name != name {
 					parentRelations = append(parentRelations, r)
 				}
 			}