Max Stocker suggested that I should add type information to the graphs

kuali · Apr 28, 2005 · e4ba98c · e4ba98c
1 parent 0501dee
commit e4ba98c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/src/net/sourceforge/schemaspy/model/TableColumn.java b/src/net/sourceforge/schemaspy/model/TableColumn.java
@@ -1,14 +1,14 @@
 package net.sourceforge.schemaspy.model;
 
 import java.io.Serializable;
+import java.sql.ResultSet;
+import java.sql.SQLException;
 import java.util.Comparator;
 import java.util.HashMap;
 import java.util.Iterator;
 import java.util.Map;
 import java.util.Set;
 import java.util.TreeMap;
-import java.sql.ResultSet;
-import java.sql.SQLException;
 
 public class TableColumn implements Serializable {
     private final Table table;
@@ -36,7 +36,7 @@ public class TableColumn implements Serializable {
             buf.append(decimalDigits);
         }
         detailedSize = buf.toString();
-        
+
         isNullable = rs.getString("NULLABLE").equalsIgnoreCase("YES");
         defaultValue = rs.getString("COLUMN_DEF");
         id = new Integer(rs.getInt("ORDINAL_POSITION"));
@@ -61,7 +61,7 @@ public String getType() {
     public int getLength() {
 	return length;
     }
-    
+
     public String getDetailedSize() {
         return detailedSize;
     }

diff --git a/src/net/sourceforge/schemaspy/view/DotFormatter.java b/src/net/sourceforge/schemaspy/view/DotFormatter.java
@@ -34,7 +34,7 @@ public void writeRelationships(Table table, LineWriter out) throws IOException {
 
         Set relatedTables = getImmediateRelatives(table);
 
-        formatter.writeNode(table, "", true, true, out);
+        formatter.writeNode(table, "", true, true, true, out);
         Set relationships = formatter.getRelationships(table);
         tablesWritten.add(table);
 
@@ -45,7 +45,7 @@ public void writeRelationships(Table table, LineWriter out) throws IOException {
             if (!tablesWritten.add(relatedTable))
                 continue; // already written
 
-            formatter.writeNode(relatedTable, "", true, false, out);
+            formatter.writeNode(relatedTable, "", true, false, false, out);
             relationships.addAll(formatter.getRelationships(relatedTable, table));
         }
 
@@ -61,7 +61,7 @@ public void writeRelationships(Table table, LineWriter out) throws IOException {
                 if (!tablesWritten.add(cousin))
                     continue; // already written
                 relationships.addAll(formatter.getRelationships(cousin, relatedTable));
-                formatter.writeNode(cousin, "", false, false, out);
+                formatter.writeNode(cousin, "", false, false, false, out);
             }
         }
 
@@ -126,7 +126,7 @@ private int write(Collection tables, boolean onlyOrphans, LineWriter out) throws
             Table table = (Table)iter.next();
             boolean isOrphan = table.getMaxParents() == 0 && table.getMaxChildren() == 0;
             if (onlyOrphans && isOrphan || !onlyOrphans && !isOrphan) {
-                formatter.writeNode(table, "tables/", true, false, out);
+                formatter.writeNode(table, "tables/", true, false, onlyOrphans && isOrphan, out);
                 ++numWritten;
             }
         }