21

有没有办法让 StreamReader 不做任何缓冲?

我正在尝试处理来自可能是二进制或文本的进程的输出。输出看起来像一个 HTTP 响应,例如

Content-type: application/whatever
Another-header: value

text or binary data here

我想要做的是使用 a 解析标头StreamReader,然后从其BaseStream或中读取StreamReader以处理其余内容。这基本上是我开始的:

private static readonly Regex HttpHeader = new Regex("([^:]+): *(.*)");
private void HandleOutput(StreamReader reader)
{
  var headers = new NameValueCollection();
  string line;
  while((line = reader.ReadLine()) != null)
  {
    Match header = HttpHeader.Match(line);
    if(header.Success)
    {
      headers.Add(header.Groups[1].Value, header.Groups[2].Value);
    }
    else
    {
      break;
    }
  }
  DoStuff(reader.ReadToEnd());
}

这似乎会破坏二进制数据。所以我把最后一行改成这样:

if(headers["Content-type"] != "text/html")
{
  // reader.BaseStream.Position is not at the same place that reader
  // makes it looks like it is.
  // i.e. reader.Read() != reader.BaseStream.Read()
  DoBinaryStuff(reader.BaseStream);
}
else
{
  DoTextStuff(reader.ReadToEnd());
}

...但 StreamReader 缓冲其输入,因此 reader.BaseStream 位置错误。有没有办法解除 StreamReader 的缓冲?或者我可以告诉 StreamReader 将流重置回 StreamReader 所在的位置吗?

4

2 回答 2

9

这个答案很晚,可能不再与您相关,但对于偶然发现此问题的其他人可能会派上用场。

我的问题涉及PPM 文件,其格式类似:

  • ASCII 文本开头
  • 文件其余部分的二进制字节

我遇到的问题是StreamReader该类无法在不缓冲内容的情况下一次读取一个字节。这在某些情况下会导致意外结果,因为该Read()方法读取的是单个字符,而不是单个字节。

我的解决方案是围绕一个流编写一个包装器,一次读取一个字节。包装器有 2 个重要的方法,ReadLine()Read().

这两种方法允许我读取流的 ASCII 行,无缓冲,然后一次读取一个字节来读取流的其余部分。您可能需要进行一些调整以满足您的需求。

class UnbufferedStreamReader: TextReader
{
    Stream s;

    public UnbufferedStreamReader(string path)
    {
        s = new FileStream(path, FileMode.Open);
    }

    public UnbufferedStreamReader(Stream stream)
    {
        s = stream;
    }

    // This method assumes lines end with a line feed.
    // You may need to modify this method if your stream
    // follows the Windows convention of \r\n or some other 
    // convention that isn't just \n
    public override string ReadLine()
    {
        List<byte> bytes = new List<byte>();
        int current;
        while ((current = Read()) != -1 && current != (int)'\n')
        {
            byte b = (byte)current;
            bytes.Add(b);
        }
        return Encoding.ASCII.GetString(bytes.ToArray());
    }

    // Read works differently than the `Read()` method of a 
    // TextReader. It reads the next BYTE rather than the next character
    public override int Read()
    {
        return s.ReadByte();
    }

    public override void Close()
    {
        s.Close();
    }
    protected override void Dispose(bool disposing)
    {
        s.Dispose();
    }

    public override int Peek()
    {
        throw new NotImplementedException();
    }

    public override int Read(char[] buffer, int index, int count)
    {
        throw new NotImplementedException();
    }

    public override int ReadBlock(char[] buffer, int index, int count)
    {
        throw new NotImplementedException();
    }       

    public override string ReadToEnd()
    {
        throw new NotImplementedException();
    }
}
于 2010-03-23T00:30:17.477 回答
0

好吧,您可以使用Stream.Seek设置流的位置。在我看来,您在这里遇到的问题是 StreamReader 正在读取字符而不是字节(取决于编码,每个字符可能不同于 1 个字节)。从MSDN 库

StreamReader 设计用于特定编码的字符输入,而 Stream 类设计用于字节输入和输出。

当您调用 reader.ReadToEnd() 时,它会根据使用的任何编码将数据作为字符串读取。使用Stream.Read方法可能会更好。使用 StreamReader 读入您的字符串数据,然后在您读入通知您传入二进制数据的标头时将二进制数据提取到 byte[] 中。

于 2009-02-06T17:21:05.247 回答