OpenCSV 避免使用 FileWriter 并返回 InputStream

Question

我正在使用 OpenCsv，该new CSVWriter()方法将 Writer 作为参数。

我想要实现的是避免写入文件系统，而是返回一个InputStream. 我不知道该怎么做。不是很熟悉Streams也是我用的JAVA 7。

我是否可以让 writeValues 方法返回一个 InputStream 并避免在文件系统上写入输出。

这是我的工作实现：

private static void writeValues(String[][] cellValues) throws IOException {

    CSVWriter writer = new CSVWriter(new FileWriter("/tmp/myfile.csv"));
    for(String[] row : cellValues) { 
        writer.writeNext(row);
    }
    writer.close();
}

这就是我想要达到的目标。如何转换上述方法以避免使用 FileWriter。

private static InputStream writeValues(String[][] cellValues) throws IOException {

    InputStream inputStream = ?;

    CSVWriter writer = new CSVWriter(?);
    for(String[] row : cellValues) { 
        writer.writeNext(row);
    }
    writer.close();

    return inputStream;
}

Answer 1

为了补充JB Nizet给出的答案：

public static byte[] getBeansAsByteArray(final List<YourBean> beans) {
    ByteArrayOutputStream stream = new ByteArrayOutputStream();
    OutputStreamWriter streamWriter = new OutputStreamWriter(stream);
    CSVWriter writer = new CSVWriter(streamWriter);

    StatefulBeanToCsv<YourBean> beanToCsv = new StatefulBeanToCsvBuilder(writer).build();
    beanToCsv.write(beans);
    streamWriter.flush();
    
    return stream.toByteArray();
}

Answer 2

写一个OutpuStreamWriter自己编写的ByteArrayOutputStream，并在最后，你必须在内存中的字节数组（通过调用getBytes()的ByteArrayOutputStream）。

然后，您可以通过ByteArrayInputStream在字节数组上打开 a 来读取该字节数组。

Answer 3

获取一个没有 StatefulBeanToCsv 的工作片段，我不喜欢它。

 String[] headers = { "id", "result", "searchTerm"};
 ByteArrayOutputStream stream = new ByteArrayOutputStream();
 OutputStreamWriter streamWriter = new OutputStreamWriter(stream);
 CSVWriter writer = new CSVWriter(streamWriter);
 writer.writeNext(headers);
 // listOfRows -> business logic collection containing data for csv rows, as you can see I've separated the declaration of headers 
 listOfRows.forEach(it -> {
        writer.writeNext(new String[] {it.getId().toString(), it.getResult(), it.getSearchTerm()});
 });

 streamWriter.flush();
 byte[] byteArrayOutputStream = stream.toByteArray();