Clean up some pipes things (#10728)

- Use the Kestrel memory pool
- Fix bad casts
This commit is contained in:
David Fowler 2019-06-01 11:10:09 -07:00 committed by GitHub
parent 6e35229b01
commit bb9b0cbfa5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions

View File

@ -109,11 +109,11 @@ namespace Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Http
if (!ReferenceEquals(_requestStreamInternal, RequestBody)) if (!ReferenceEquals(_requestStreamInternal, RequestBody))
{ {
_requestStreamInternal = RequestBody; _requestStreamInternal = RequestBody;
RequestBodyPipeReader = PipeReader.Create(RequestBody); RequestBodyPipeReader = PipeReader.Create(RequestBody, new StreamPipeReaderOptions(_context.MemoryPool, _context.MemoryPool.GetMinimumSegmentSize(), _context.MemoryPool.GetMinimumAllocSize()));
OnCompleted((self) => OnCompleted((self) =>
{ {
((PipeWriter)self).Complete(); ((PipeReader)self).Complete();
return Task.CompletedTask; return Task.CompletedTask;
}, RequestBodyPipeReader); }, RequestBodyPipeReader);
} }
@ -243,11 +243,11 @@ namespace Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Http
if (!ReferenceEquals(_responseStreamInternal, ResponseBody)) if (!ReferenceEquals(_responseStreamInternal, ResponseBody))
{ {
_responseStreamInternal = ResponseBody; _responseStreamInternal = ResponseBody;
ResponseBodyPipeWriter = PipeWriter.Create(ResponseBody); ResponseBodyPipeWriter = PipeWriter.Create(ResponseBody, new StreamPipeWriterOptions(_context.MemoryPool));
OnCompleted((self) => OnCompleted((self) =>
{ {
((PipeReader)self).Complete(); ((PipeWriter)self).Complete();
return Task.CompletedTask; return Task.CompletedTask;
}, ResponseBodyPipeWriter); }, ResponseBodyPipeWriter);
} }