在NPOI workbook.write之后,MemoryStream似乎已关闭?

时间:2014-04-08 08:28:54

标签: c# excel asp.net-web-api npoi

我正在使用NPOI将DataTable转换为ASP.NET Web API项目中的Excel。

但我从回应中得不到任何结果。这是我的代码:

public HttpResponseMessage GetExcelFromDataTable(DataTable dt)
{
    IWorkbook workbook = new XSSFWorkbook(); // create *.xlsx file, use HSSFWorkbook() for creating *.xls file.
    ISheet sheet1 = workbook.CreateSheet();
    IRow row1 = sheet1.CreateRow(0);
    for (int i = 0; dt.Columns.Count > i; i++)
    {
        row1.CreateCell(i).SetCellValue(dt.Columns[i].ColumnName);
    }

    for (int i = 0; dt.Rows.Count > i; i++)
    {
        IRow row = sheet1.CreateRow(i + 1);
        for (int j = 0; dt.Columns.Count > j; j++)
        {
            row.CreateCell(j).SetCellValue(dt.Rows[i][j].ToString());
        }
    }

    MemoryStream ms = new MemoryStream();
    workbook.Write(ms);
    HttpResponseMessage result = new HttpResponseMessage(HttpStatusCode.OK);
    result.Content = new StreamContent(ms);
    result.Content.Headers.ContentType = new MediaTypeHeaderValue("application/octet-stream");
    result.Content.Headers.ContentDisposition = new ContentDispositionHeaderValue("attachment");
    result.Content.Headers.ContentDisposition.FileName = string.Format("{0}.xlsx", dt.TableName);
    return result;
}

我设置了一个断点来检查ms.Length之后的workbook.Write(ms),但它会返回一个例外:System.ObjectDisposedException

我哪里出错了?

4 个答案:

答案 0 :(得分:27)

此问题的另一种解决方法......不使用多个MemoryStream个对象。

创建一个继承NpoiMemoryStream的{​​{1}}类,并覆盖MemoryStream方法:

Close

然后,使用这样的流:

public class NpoiMemoryStream : MemoryStream
{
    public NpoiMemoryStream()
    {
        // We always want to close streams by default to
        // force the developer to make the conscious decision
        // to disable it.  Then, they're more apt to remember
        // to re-enable it.  The last thing you want is to
        // enable memory leaks by default.  ;-)
        AllowClose = true;
    }

    public bool AllowClose { get; set; }

    public override void Close()
    {
        if (AllowClose)
            base.Close();
    }
}

在flush和seek之间的某个时刻,NPOI将尝试关闭流,但由于我们覆盖var ms = new NpoiMemoryStream(); ms.AllowClose = false; workbook.Write(ms); ms.Flush(); ms.Seek(0, SeekOrigin.Begin); ms.AllowClose = true; 并且Close()标志为false,我们可以保持流打开。然后,将AllowClose设置回true,以便正常的处理机制可以关闭它。

不要误会我的意思......这仍然是一个不应该被实现的黑客......但从内存使用的角度来看它有点清晰。

答案 1 :(得分:6)

如上所述[{3}}以及alun问题,您可以将流提供给另一个MemoryStream:

...
MemoryStream ms = new MemoryStream();
using(MemoryStream tempStream = new MemoryStream)
{
    workbook.Write(tempStream);
    var byteArray = tempStream.ToArray();
    ms.Write(byteArray, 0, byteArray.Length);
    HttpResponseMessage result = new HttpResponseMessage(HttpStatusCode.OK);
    result.Content = new StreamContent(ms);
    result.Content.Headers.ContentType = new MediaTypeHeaderValue("application/octet-stream");
    result.Content.Headers.ContentDisposition = new ContentDispositionHeaderValue("attachment");
    result.Content.Headers.ContentDisposition.FileName = string.Format("{0}.xlsx", dt.TableName);
    return result;
}

必须这样做有一点代码味道。但是,仅在输出.xlsx文件时才需要这样做,因为第三方库涉及处理流的方式。

答案 2 :(得分:2)

我遇到了关闭/处理它们的流的API的类似问题 不拥有。我不熟悉NPOI,但我认为Write方法接受Stream,而不是MemoryStream。如果是这种情况,您可以创建一个包装器Stream类,它将所有调用(读/写/搜索等)转发到内部流(在本例中为您的MemoryStream),但不转发对close / dispose的调用。将包装器传递给Write方法,当它返回时,MemoryStream应包含所有内容并仍然是“打开”。

此外,您可能需要ms.Seek(0, SeekOrigin.Begin)。在调用Write后,您的内存流将定位在流的末尾,因此如果您尝试从该位置读取它将显示为emtpy。

答案 3 :(得分:0)

我不知道是否仍然需要这样做,但是有一个overload

Write(Stream stream, bool leaveOpen)

如果您设置了leaveOpen = true,则将MemoryStream保持打开状态