我继承了Stream类,但我不知道如何正确实现Read()函数,以免出现大量嵌套的ifs和难以调试的代码。问题在于从流源读取返回大小不变的缓冲区(例如,不可更改),但Read()函数接受不同大小的缓冲区。我考虑添加BufferedStream,但我认为这是一个坏主意。谢谢你的帮助!
内部源会返回固定大小的缓冲区吗?如果是这种情况,那就不是BufferedStream
所做的 - 它只是减少对物理流的调用次数。您需要一个单独的机制来缓存 - 填充和清空的MemoryStream将是一个合理的选择。例如(完全未经测试):
MemoryStream localBuffer = new MemoryStream();
bool ReadNextChunk()
{
// clear
localBuffer.Position = 0;
localBuffer.SetLength(0);
// get data
byte[] chunk = null; // TODO - read from source
if(chunk == null || chunk.Length == 0) return false; // EOF
localBuffer.Write(chunk, 0, chunk.Length);
localBuffer.Position = 0;
return true;
}
public override int Read(byte[] buffer, int offset, int count)
{
int bytes;
if ((bytes = localBuffer.Read(buffer, offset, count)) > 0) return bytes;
if (!ReadNextChunk()) return 0;
return localBuffer.Read(buffer, offset, count);
}
以下是您的“10分好题”(不确定是否可以全球通用)。
byte[] myBuffer = new byte[fixedSize];
int myBufferPos = fixedSize;
public int Read(byte[] buffer, int offset, int count)
{
int copiedCount = 0
while (copiedCount < count)
{
if (myBufferPos >= fixedSize)
{
//Read new fixed buffer into myBuffer
// use break on no more buffer.
myBufferPos = 0;
}
int bytesToCopy = fixedSize - myBufferPos;
if (bytesToCopy > count - copiedCount)
byteToCopy = count - copiedCount;
Array.Copy(myBuffer, myBufferPos, buffer, offset, byteToCopy);
offset += bytesToCopy;
myBufferPos += bytesToCopy;
copiedCount += bytesToCopy;
}
return copiedCount;
}
未经测试,可能存在一些错误。不清楚您的源流是否具有其固定大小的精确倍数长度。如果不是,则最终的部分缓冲区需要一些额外的逻辑。
基本原则是维护自己的固定大小缓冲区,并跟踪到目前为止已被读取消耗的位置。