issue-128 : Fix #128 by writing header during close if necessary

Signed-off-by: Peter Ansell <[email protected]>
FasterXML · Aug 22, 2016 · 6637c03 · 6637c03
1 parent feda1a2
commit 6637c03
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/com/fasterxml/jackson/dataformat/csv/CsvGenerator.java b/src/main/java/com/fasterxml/jackson/dataformat/csv/CsvGenerator.java
@@ -428,6 +428,12 @@ public void close() throws IOException
 
         // Let's mark row as closed, if we had any...
         finishRow();
+
+        // Write the header if necessary, occurs when no rows written
+        if (_handleFirstLine) {
+            _handleFirstLine();
+        }
+
         _writer.close(_ioContext.isResourceManaged() || isEnabled(JsonGenerator.Feature.AUTO_CLOSE_TARGET));
     }