Now uses a more aggressive removal of recursive constraints.

kuali · Aug 30, 2006 · ee6c341 · ee6c341
1 parent 3540e6c
commit ee6c341
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/net/sourceforge/schemaspy/model/Table.java b/src/net/sourceforge/schemaspy/model/Table.java
@@ -405,10 +405,13 @@ public boolean isLeaf() {
     public ForeignKeyConstraint removeSelfReferencingConstraint() {
         ForeignKeyConstraint recursiveConstraint = getSelfReferencingConstraint();
         if (recursiveConstraint != null) {
-            TableColumn childColumn = (TableColumn)recursiveConstraint.getChildColumns().get(0);
-            TableColumn parentColumn = (TableColumn)recursiveConstraint.getParentColumns().get(0);
-            childColumn.removeParent(parentColumn);
-            parentColumn.removeChild(childColumn);
+            // more drastic removal solution by Remke Rutgers:
+            for (int i = 0; i < recursiveConstraint.getChildColumns().size(); i++) {
+                TableColumn childColumn = (TableColumn)recursiveConstraint.getChildColumns().get(i);
+                TableColumn parentColumn = (TableColumn)recursiveConstraint.getParentColumns().get(i);
+                childColumn.removeParent(parentColumn);
+                parentColumn.removeChild(childColumn);
+            }
             return recursiveConstraint;
         }