import org.apache.poi.EmptyFileException;
import org.apache.poi.POIDocument;
+import org.apache.poi.ss.usermodel.Workbook;
public final class IOUtils {
private static final POILogger logger = POILogFactory.getLogger( IOUtils.class );
}
}
+ public static void write(Workbook doc, OutputStream out) throws IOException {
+ try {
+ doc.write(out);
+ } finally {
+ closeQuietly(out);
+ }
+ }
+
/**
* Write a POI Document ({@link org.apache.poi.ss.usermodel.Workbook}, {@link org.apache.poi.sl.usermodel.SlideShow}, etc) to an output stream and close the output stream.
* This will attempt to close the output stream at the end even if there was a problem writing the document to the stream.
closeQuietly(doc);
}
}
+
+ // Since the Workbook interface doesn't derive from POIDocument
+ // We'll likely need one of these for each document container interface
+ public static void writeAndClose(Workbook doc, OutputStream out) throws IOException {
+ try {
+ doc.write(out);
+ } finally {
+ closeQuietly(doc);
+ }
+ }
/**
* Copies all the data from the given InputStream to the OutputStream. It
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
-import java.io.OutputStream;
import org.apache.poi.hssf.HSSFTestDataSamples;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.util.IOUtils;
import org.apache.poi.util.TempFile;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
*/
public class XSSFTestDataSamples {
/**
- * Used by {@link writeOutAndReadBack(R wb, String testName)}. If a
+ * Used by {@link #writeOutAndReadBack(Workbook, String)}. If a
* value is set for this in the System Properties, the xlsx file
* will be written out to that directory.
*/
* @throws IOException
*/
public static <R extends Workbook> File writeOut(R wb, String testName) throws IOException {
+ final File file = getOutputFile(testName);
+ writeOut(wb, file);
+ return file;
+ }
+
+ private static <R extends Workbook> void writeOut(R wb, File file) throws IOException {
+ IOUtils.write(wb, new FileOutputStream(file));
+ }
+
+ // Anticipates the location of where a workbook will be written to
+ // Note that if TEST_OUTPUT_DIR is not set, this will create temporary files
+ // with unique names. Subsequent calls with the same argument may return a different file.
+ // Gets a test data sample file, deleting the file if it exists.
+ // This is used in preparation for writing a workbook out to the returned output file.
+ private static File getOutputFile(String testName) throws IOException {
final String testOutputDir = System.getProperty(TEST_OUTPUT_DIR);
final File file;
if (testOutputDir != null) {
if (file.exists()) {
file.delete();
}
- final OutputStream out = new FileOutputStream(file);
- try {
- wb.write(out);
- } finally {
- out.close();
- }
return file;
}