96 lines
2.7 KiB
C#
96 lines
2.7 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Threading;
|
|
|
|
using Best.HTTP.Shared.PlatformSupport.Memory;
|
|
|
|
namespace Best.HTTP.Shared.Streams
|
|
{
|
|
public class BufferSegmentStream : Stream
|
|
{
|
|
public override bool CanRead { get { return true; } }
|
|
|
|
public override bool CanSeek { get { return false; } }
|
|
|
|
public override bool CanWrite { get { return false; } }
|
|
|
|
public override long Length { get { return this._length; } }
|
|
protected long _length;
|
|
|
|
public override long Position { get { return 0; } set { } }
|
|
|
|
protected List<BufferSegment> bufferList = new List<BufferSegment>();
|
|
|
|
private byte[] _tempByteArray = new byte[1];
|
|
|
|
public override int ReadByte()
|
|
{
|
|
if (Read(this._tempByteArray, 0, 1) == 0)
|
|
return -1;
|
|
|
|
return this._tempByteArray[0];
|
|
}
|
|
|
|
public override int Read(byte[] buffer, int offset, int count)
|
|
{
|
|
int sumReadCount = 0;
|
|
|
|
while (count > 0 && bufferList.Count > 0)
|
|
{
|
|
BufferSegment buff = this.bufferList[0];
|
|
|
|
int readCount = Math.Min(count, buff.Count);
|
|
|
|
Array.Copy(buff.Data, buff.Offset, buffer, offset, readCount);
|
|
|
|
sumReadCount += readCount;
|
|
offset += readCount;
|
|
count -= readCount;
|
|
|
|
this.bufferList[0] = buff = buff.Slice(buff.Offset + readCount);
|
|
|
|
if (buff.Count == 0)
|
|
{
|
|
this.bufferList.RemoveAt(0);
|
|
BufferPool.Release(buff.Data);
|
|
}
|
|
}
|
|
|
|
Interlocked.Add(ref this._length, -sumReadCount);
|
|
|
|
return sumReadCount;
|
|
}
|
|
|
|
public override void Write(byte[] buffer, int offset, int count) => Write(new BufferSegment(buffer, offset, count));
|
|
|
|
public virtual void Write(BufferSegment bufferSegment)
|
|
{
|
|
if (bufferSegment.Count == 0)
|
|
return;
|
|
|
|
this.bufferList.Add(bufferSegment);
|
|
Interlocked.Add(ref this._length, bufferSegment.Count);
|
|
}
|
|
|
|
public virtual void Reset()
|
|
{
|
|
BufferPool.ReleaseBulk(this.bufferList);
|
|
|
|
this.bufferList.Clear();
|
|
Interlocked.Exchange(ref this._length, 0);
|
|
}
|
|
|
|
protected override void Dispose(bool disposing)
|
|
{
|
|
base.Dispose(disposing);
|
|
|
|
Reset();
|
|
}
|
|
|
|
public override void Flush() { }
|
|
public override long Seek(long offset, SeekOrigin origin) => throw new NotImplementedException();
|
|
public override void SetLength(long value) => throw new NotImplementedException();
|
|
}
|
|
}
|