Add DbDriver, with support for a few popular JDBC drivers.

[squelch.git] / src / main / java / net / jaekl / squelch / stmt / Select.java

diff --git a/src/main/java/net/jaekl/squelch/stmt/Select.java b/src/main/java/net/jaekl/squelch/stmt/Select.java
index 6d89af5c293c6dbf237e3f8dadf7b170013c7bb7..088181a50662c1015d0f1a936a29730e2bbfe531 100644 (file)
--- a/src/main/java/net/jaekl/squelch/stmt/Select.java
+++ b/src/main/java/net/jaekl/squelch/stmt/Select.java
@@ -61,20 +61,28 @@ public class Select extends Query {
                        ResultSetMetaData metaData = rs.getMetaData();
                        int rowCount = 0;
                        Column[] cols = getColumns(metaData);
-                       
-                       pw.println("----------------------------------");
+                       StringBuilder sb = new StringBuilder();
+                       sb.append("----------------------------------\n");
                        for (int idx = 1; idx <= cols.length; ++idx) {
-                               pw.print("|" + cols[idx - 1].getLabel());
+                               sb.append("|" + cols[idx - 1].getLabel());
                        }
-                       pw.println("|");
+                       sb.append("|\n");
+                       String header = sb.toString();
                        
                        while (rs.next()) {
+                               if (null != header) {
+                                       pw.print(header);
+                                       header = null;
+                               }
+                               
                                for (int idx = 1; idx <= cols.length; ++idx) {
                                        pw.print("|" + rs.getObject(idx));
                                }
                                pw.println("|");
                        }
-                       pw.println("----------------------------------");
+                       if (rowCount > 0) {
+                               pw.println("----------------------------------");
+                       }
                        pw.println("" + rowCount + " row(s) returned.");
                        
                        return rowCount;