Merge remote-tracking branch 'eengbrec/serializer_fix'

djspiewak · Oct 1, 2011 · b56ce1e · b56ce1e
2 parents f17cfc0 + 753ccff
commit b56ce1e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/main/scala/com/codecommit/antixml/XMLSerializer.scala b/src/main/scala/com/codecommit/antixml/XMLSerializer.scala
@@ -63,11 +63,18 @@ class XMLSerializer(encoding: String, outputDeclaration: Boolean) {
    * Uses the character encoding of this XMLSerializer (default is UTF-8).
    */
   def serializeDocument(elem: Elem, o: OutputStream) {
-    serializeDocument(elem, new OutputStreamWriter(o, encoding))
+    val writer = new OutputStreamWriter(o, encoding)
+    serializeDocument(elem, writer)
+    writer.flush()
   }
 
   def serializeDocument(elem: Elem, outputFile: File) {
-    serializeDocument(elem, new FileOutputStream(outputFile))
+    val fos = new FileOutputStream(outputFile)
+    try {
+      serializeDocument(elem, fos)
+    } finally {
+      fos.close()
+    }
   }
 
   def serialize(elem: Elem, w: Writer) {