4

我想允许客户端下载包含来自 ResultSet 的数据的 csv 文件。我正在使用 java API (PLAY! 2.1)。

我的解决方案有效，但在 ResultSet 包含大量数据时不可用。

在我的控制器中：

ResultSet rs = st.executeQuery(sql);
String filename = "";
filename = createCSV(rs);.
response().setContentType("text/csv");
response().setHeader("Content-disposition","attachment; filename="+"export.csv");
return ok(new java.io.File("D:\\UTILIS~1\\user\\AppData\\Local\\Temp\\"+filename));

createCSV 方法：

public static String createCSV(ResultSet _resultSet) {
    String filename = "";
    try{
        ResultSetMetaData meta = _resultSet.getMetaData();

        // CSV file creation 
        File tempDir = new File(System.getProperty("java.io.tmpdir"));
        System.out.println(System.getProperty("java.io.tmpdir"));
        File tempFile = File.createTempFile("test", ".csv", tempDir);
        filename = tempFile.getName();
        FileWriter fileWriter = new FileWriter(tempFile, true);
        System.out.println(tempFile.getAbsolutePath());
        BufferedWriter bw = new BufferedWriter(fileWriter);

        for(int i = 1; i <= meta.getColumnCount(); i++)
        {
            String columnLabel = meta.getColumnName(i);
            bw.write(columnLabel + "|");
        }
        bw.write("\r\n");
        while(_resultSet.next())
        {
            for(int i = 1, count = meta.getColumnCount(); i <= count; i++){
                if (_resultSet.getObject(i) == null)
                    bw.write("null|");
                else
                    bw.write(_resultSet.getObject(i).toString() + "|");
            }
            bw.write("\r\n");
        }
        bw.close();
        tempFile.deleteOnExit();
    } catch(Exception e){
        e.printStackTrace();
    } 
    return filename;
}

为了节省内存，我如何写入文件并使用流将其发送给用户。

我想我需要使用 Chunks，但我不知道该怎么做。

任何样品，提示？

4

3 回答 3

1

Ok 方法接受一个字节数组作为参数

因此，获取文件内容的字节数组并尝试

return ok(s.toByteArray()).as("application/octet-stream");

看到这个得到一个文件的字节数组

于 2013-06-05T07:14:04.657 回答
1

您可以在文档中找到分块响应的示例：

http://www.playframework.com/documentation/2.1.1/JavaStream

您可以在生成 csv 内容时发送它。

于 2013-06-05T09:09:39.787 回答
0

1) 从响应中获取输出流

2）将此输出流传递给createCSV方法并直接写入

您还可以通过使用 ZipOutputStream 装饰输出流来压缩结果。

注意：无需创建临时文件。

于 2013-07-09T16:24:30.277 回答