Question

我使用以下代码编写日志文件：

Encoding enc = Encoding.GetEncoding(932);
MemoryStream msLog = new MemoryStream();
StreamWriter swLog = new StreamWriter(msLog, enc);
swLog.WriteLine("Line Number,Error,Additional Information"); //log header

经过一些复杂的处理后，我想知道除了标题之外是否添加了任何日志行。显然，一种方法是每当我使用swLog.WriteLine（）时将一些布尔变量设置为true，但由于代码冗长而复杂，我想避免这种方法。如何轻松检查内存流的行数？

Answer 1

正如您所指出的，还有其他更好的方法可以做到这一点。但是，这里是您问题的直接答案：

首先，确保StreamWriter已将数据刷新到流中，如下所示：

swLog.Flush();

然后，您可以使用以下方法检测MemoryStream是否有多行：

private bool HasMoreThanNumberOfLines(Stream stream, Encoding enc, int number_of_lines)
{
    long current_position = stream.Position;

    stream.Position = 0;

    try
    {
        using(StreamReader sr = new StreamReader(stream, enc, true, 1024, true))
        { 
            for (int i = 0; i < number_of_lines + 1 ; i++)
            {
                string line = sr.ReadLine();

                if (line == null)
                    return false;
            }
        }
        return true;
    }
    finally
    {
        stream.Position = current_position;
    }
}

请注意，我使用的是StreamReader的特殊构造函数，以确保它在处理时不会关闭基础流（stream）。

另请注意此方法如何保存流的当前位置，然后在执行其逻辑后恢复它，以便StreamWriter继续正常工作。

您可以像这样使用此方法：

var has_another_line = HasMoreThanNumberOfLines(msLog, enc, 1);

请注意，这不是线程安全的。我假设在任何时间点都可以通过单个线程访问流。您需要放置一些锁以使其成为线程安全的。

c＃：从MemoryStream / StreamWriter获取行数

1 个答案: