Merge pull request #61 from zanata/rhbz986145

rhbz986145 Check for IO error after each write
zanata · Jul 22, 2013 · 6b66c90 · 6b66c90
2 parents d4c6611 + 8f83014
commit 6b66c90
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/zanata-war/src/main/java/org/zanata/rest/service/TMXStreamingOutput.java b/zanata-war/src/main/java/org/zanata/rest/service/TMXStreamingOutput.java
@@ -25,7 +25,6 @@
 import java.io.IOException;
 import java.io.OutputStream;
 import java.io.PrintWriter;
-import java.io.Writer;
 import java.util.Iterator;
 
 import javax.annotation.Nonnull;
@@ -79,7 +78,7 @@ public void write(OutputStream output) throws IOException, WebApplicationExcepti
       if (iter.hasNext()) iter.peek();
 
       @Cleanup
-      Writer writer = new PrintWriter(output);
+      PrintWriter writer = new PrintWriter(output);
       @Cleanup
       XMLWriter xmlWriter = new XMLWriter(writer);
       @Cleanup
@@ -102,6 +101,10 @@ public void write(OutputStream output) throws IOException, WebApplicationExcepti
          SourceContents tu = iter.next();
          net.sf.okapi.common.LocaleId sourceLocale = OkapiUtil.toOkapiLocale(tu.getLocale());
          exportTUStrategy.exportTranslationUnit(tmxWriter, tu, sourceLocale);
+         if (writer.checkError())
+         {
+            throw new IOException("error writing to output");
+         }
       }
       tmxWriter.writeEndDocument();
    }