huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Faster MemoryPoolIterator2.CopyFrom

This commit is contained in:
Ben Adams 2015-12-17 11:32:01 +00:00
parent 5d6fdc5eb5
commit 7625f65e6c
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/Microsoft.AspNet.Server.Kestrel/Infrastructure/MemoryPoolIterator2.cs
View File

@ -585,15 +585,15 @@ namespace Microsoft.AspNet.Server.Kestrel.Infrastructure
public void CopyFrom(byte[] data) public void CopyFrom(byte[] data)
{ {
CopyFrom(new ArraySegment<byte>(data)); CopyFrom(data, 0, data.Length);
}
public void CopyFrom(byte[] data, int offset, int count)
{
CopyFrom(new ArraySegment<byte>(data, offset, count));
} }
public void CopyFrom(ArraySegment<byte> buffer) public void CopyFrom(ArraySegment<byte> buffer)
{
CopyFrom(buffer.Array, buffer.Offset, buffer.Count);
}
public void CopyFrom(byte[] data, int offset, int count)
{ {
Debug.Assert(_block != null); Debug.Assert(_block != null);
Debug.Assert(_block.Pool != null); Debug.Assert(_block.Pool != null);
@ -604,8 +604,8 @@ namespace Microsoft.AspNet.Server.Kestrel.Infrastructure
var block = _block; var block = _block;
var blockIndex = _index; var blockIndex = _index;
var bufferIndex = buffer.Offset; var bufferIndex = offset;
var remaining = buffer.Count; var remaining = count;
var bytesLeftInBlock = block.Data.Offset + block.Data.Count - blockIndex; var bytesLeftInBlock = block.Data.Offset + block.Data.Count - blockIndex;
while (remaining > 0) while (remaining > 0)
@ -613,6 +613,7 @@ namespace Microsoft.AspNet.Server.Kestrel.Infrastructure
if (bytesLeftInBlock == 0) if (bytesLeftInBlock == 0)
{ {
var nextBlock = pool.Lease(); var nextBlock = pool.Lease();
block.End = blockIndex;
block.Next = nextBlock; block.Next = nextBlock;
block = nextBlock; block = nextBlock;
@ -622,15 +623,15 @@ namespace Microsoft.AspNet.Server.Kestrel.Infrastructure
var bytesToCopy = remaining < bytesLeftInBlock ? remaining : bytesLeftInBlock; var bytesToCopy = remaining < bytesLeftInBlock ? remaining : bytesLeftInBlock;
Buffer.BlockCopy(buffer.Array, bufferIndex, block.Array, blockIndex, bytesToCopy); Buffer.BlockCopy(data, bufferIndex, block.Array, blockIndex, bytesToCopy);
blockIndex += bytesToCopy; blockIndex += bytesToCopy;
bufferIndex += bytesToCopy; bufferIndex += bytesToCopy;
remaining -= bytesToCopy; remaining -= bytesToCopy;
bytesLeftInBlock -= bytesToCopy; bytesLeftInBlock -= bytesToCopy;
block.End = blockIndex;
} }
block.End = blockIndex;
_block = block; _block = block;
_index = blockIndex; _index = blockIndex;
} }
@ -661,6 +662,7 @@ namespace Microsoft.AspNet.Server.Kestrel.Infrastructure
if (bytesLeftInBlock == 0) if (bytesLeftInBlock == 0)
{ {
var nextBlock = pool.Lease(); var nextBlock = pool.Lease();
block.End = blockIndex;
block.Next = nextBlock; block.Next = nextBlock;
block = nextBlock; block = nextBlock;
@ -692,10 +694,10 @@ namespace Microsoft.AspNet.Server.Kestrel.Infrastructure
bytesLeftInBlockMinusSpan -= copied; bytesLeftInBlockMinusSpan -= copied;
bytesLeftInBlock -= copied; bytesLeftInBlock -= copied;
} }
block.End = blockIndex;
} }
} }
block.End = blockIndex;
_block = block; _block = block;
_index = blockIndex; _index = blockIndex;
} }