Using 'fake' tablename; print drop and create table when first

line-print is requested

timbuctoo-instancev4/src/main/java/nl/knaw/huygens/timbuctoo/v5/serializable/serializations/SqlSerialization.java

@@ -5,17 +5,16 @@
 import java.io.PrintWriter;
 import java.util.ArrayList;
 import java.util.List;
-
 import nl.knaw.huygens.timbuctoo.v5.serializable.dto.Value;
-import nl.knaw.huygens.timbuctoo.v5.serializable.serializations.base.CollectionsOfEntitiesSerialization;
+import nl.knaw.huygens.timbuctoo.v5.serializable.serializations.base.FlatTableSerialization;
 
-public class SqlSerialization extends CollectionsOfEntitiesSerialization {
+public class SqlSerialization extends FlatTableSerialization {
 
-  //  private final CSVPrinter csvPrinter;
-  private String tableName;
+  private String tableName = "graphqlexport";
   private String columnHeaders;
   private List<String> columns;
   protected final PrintWriter writer;
+  protected boolean firstLine = true;
 
   public SqlSerialization(OutputStream outputStream) throws IOException {
     writer = new PrintWriter(outputStream);
@@ -38,21 +37,25 @@ protected void setTableName(String tableName) {
 
   protected void writeCreateTable() {
     if (tableName.isEmpty()) {
-      System.err.println("writeCreateTable: tableName is empty!");
+      // System.err.println("writeCreateTable: tableName is empty!");
       tableName = "tableName";
     }
-    System.out.println("DROP TABLE " + tableName + ";");
+    writer.println("DROP TABLE " + tableName + ";");
     String createTableString = "CREATE TABLE " + tableName + " (\n";
     for (String column: columns) {
       createTableString += column + " text,\n";
       createTableString += column + "_type text,\n";
     }
     createTableString = createTableString.substring(0, createTableString.length() - 2);
     createTableString += ");";
-    System.out.println(createTableString);
+    writer.println(createTableString);
   }
 
   protected void writeRow(List<Value> values) throws IOException {
+    if (firstLine) {
+      writeCreateTable();
+      firstLine = false;
+    }
     String columnValues = "";
     for (Value value : values) {
       if (value == null) {
@@ -63,8 +66,14 @@ protected void writeRow(List<Value> values) throws IOException {
         columnValues += ", '" + value.getType() + "'";
       }
     }
-    System.out.println("INSERT INTO " + this.tableName + "(" + columnHeaders +
+    writer.println("INSERT INTO " + this.tableName + " (" + columnHeaders +
             ") VALUES (" + columnValues.substring(2) + ");");
   }
+
+  @Override
+  protected void finish() throws IOException {
+    writer.flush();
+    writer.close();
+  }
 
 }