c#：从 MemoryStream/StreamWriter 获取行数

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

经过一些复杂的处理，我想知道是否添加了除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 获取行数

c#: Getting line count from MemoryStream/StreamWriter

c#

streamwriter

memorystream