使用 StreamWriter 写入 MemoryStream 返回空

Question

我不确定我做错了什么，看过很多例子，但似乎无法让它发挥作用。

public static Stream Foo()
{
    var memStream = new MemoryStream();
    var streamWriter = new StreamWriter(memStream);

    for (int i = 0; i < 6; i++)
        streamWriter.WriteLine("TEST");

    memStream.Seek(0, SeekOrigin.Begin);
    return memStream;
}

我正在对此方法进行简单的测试，试图让它通过，但无论如何，我的集合计数都是 0。

[Test]
public void TestStreamRowCount()
{
    var stream = Foo();

    using (var reader = new StreamReader(stream))
    {
        var collection = new List<string>();
        string input;

        while ((input = reader.ReadLine()) != null)
            collection.Add(input);

        Assert.AreEqual(6, collection.Count);
    }
}

注意：我更改了上面的一些语法，而没有在测试方法中进行编译。更重要的是第一个方法似乎返回一个空流（我的 reader.ReadLine() 总是读取一次）。不确定我做错了什么。谢谢。

Answer 1

您忘记刷新您的 StreamWriter 实例。

public static Stream Foo()
{
    var memStream = new MemoryStream();
    var streamWriter = new StreamWriter(memStream);

    for (int i = 0; i < 6; i++)
        streamWriter.WriteLine("TEST");

    streamWriter.Flush();                                   <-- need this
    memStream.Seek(0, SeekOrigin.Begin);
    return memStream;
}

另请注意，StreamWriter 应该被释放，因为它实现了 IDisposable，但这又会产生另一个问题，它将关闭底层的 MemoryStream > 也是如此。

您确定要在此处返回 MemoryStream 吗？

我会将代码更改为：

public static byte[] Foo()
{
    using (var memStream = new MemoryStream())
    using (var streamWriter = new StreamWriter(memStream))
    {
        for (int i = 0; i < 6; i++)
            streamWriter.WriteLine("TEST");

        streamWriter.Flush();
        return memStream.ToArray();
    }
}

[Test]
public void TestStreamRowCount()
{
    var bytes = Foo();

    using (var stream = new MemoryStream(bytes))
    using (var reader = new StreamReader(stream))
    {
        var collection = new List<string>();
        string input;

        while ((input = reader.ReadLine()) != null)
            collection.Add(input);

        Assert.AreEqual(6, collection.Count);
    }
}

Answer 2

由于您没有使用“using”或streamWriter.Flush()，编写器没有向流提交更改。结果 Stream 本身还没有数据。一般来说，您希望使用 using 包装对 Stream 和 StremaWriter 实例的操作。

您还应该考虑返回 MemoryStream 的新实例：

using(var memStream = new MemoryStream())
{
   ....
   return new MemoryStream(memStream.ToArray(), false /*writable*/);
}

Answer 3

写完你的行后尝试刷新streamWriter。