internal static Task<BufferedReadStream> BufferStreamAsync(Stream inputStream)
{
BufferedReadStream bufferedReadStream = new BufferedReadStream(inputStream);
return Task.Factory.Iterate(bufferedReadStream.BufferInputStream()).FollowAlwaysWith(delegate (Task task) {
inputStream.Dispose();
}).FollowOnSuccessWith<BufferedReadStream>(delegate (Task task) {
bufferedReadStream.ResetForReading();
return bufferedReadStream;
});
}
/// <summary>
/// Given the <paramref name="inputStream"/> this method returns a task which will asynchronously
/// read the entire content of that stream and return a new synchronous stream from which the data can be read.
/// </summary>
/// <param name="inputStream">The input stream to asynchronously buffer.</param>
/// <returns>A task which returns the buffered stream.</returns>
internal static Task <BufferedReadStream> BufferStreamAsync(Stream inputStream)
{
DebugUtils.CheckNoExternalCallers();
Debug.Assert(inputStream != null, "inputStream != null");
BufferedReadStream bufferedReadStream = new BufferedReadStream(inputStream);
// Note that this relies on lazy eval of the enumerator
return(Task.Factory.Iterate(bufferedReadStream.BufferInputStream())
.FollowAlwaysWith((task) => inputStream.Dispose())
.FollowOnSuccessWith(
(task) =>
{
bufferedReadStream.ResetForReading();
return bufferedReadStream;
}));
}
protected internal Task <Stream> GetStreamAsync(Func <Task <Stream> > streamFuncAsync, bool isRequest)
{
Func <Task <Stream>, Stream> operation = null;
if (!this.writing)
{
Stream stream = this.TryGetBufferingReadStream();
if (stream != null)
{
return(TaskUtils.GetCompletedTask <Stream>(stream));
}
}
Task <Stream> task = streamFuncAsync();
ValidateMessageStreamTask(task, isRequest);
task = task.FollowOnSuccessWith <Stream, Stream>(delegate(Task <Stream> streamTask) {
Stream result = streamTask.Result;
ValidateMessageStream(result, isRequest);
bool flag = !this.writing && (this.maxMessageSize > 0L);
if (this.disableMessageStreamDisposal && flag)
{
return(MessageStreamWrapper.CreateNonDisposingStreamWithMaxSize(result, this.maxMessageSize));
}
if (this.disableMessageStreamDisposal)
{
return(MessageStreamWrapper.CreateNonDisposingStream(result));
}
if (flag)
{
result = MessageStreamWrapper.CreateStreamWithMaxSize(result, this.maxMessageSize);
}
return(result);
});
if (this.writing)
{
return(task);
}
task = task.FollowOnSuccessWithTask <Stream, BufferedReadStream>(streamTask => BufferedReadStream.BufferStreamAsync(streamTask.Result)).FollowOnSuccessWith <BufferedReadStream, Stream>(streamTask => streamTask.Result);
if (this.useBufferingReadStream != true)
{
return(task);
}
if (operation == null)
{
operation = delegate(Task <Stream> streamTask) {
Stream result = streamTask.Result;
this.bufferingReadStream = new Microsoft.Data.OData.BufferingReadStream(result);
return(this.bufferingReadStream);
};
}
return(task.FollowOnSuccessWith <Stream, Stream>(operation));
}
/// <summary>
/// Asynchronously get the stream backing this message.
/// </summary>
/// <param name="streamFuncAsync">A function that returns a task for the stream backing the message.</param>
/// <param name="isRequest">true if the message is a request message; false for a response message.</param>
/// <returns>A task that when completed returns the stream backing the message.</returns>
protected internal Task <Stream> GetStreamAsync(Func <Task <Stream> > streamFuncAsync, bool isRequest)
{
// Check whether we have an existing buffering read stream when reading
if (!this.writing)
{
Stream existingBufferingReadStream = this.TryGetBufferingReadStream();
Debug.Assert(!this.writing || existingBufferingReadStream == null, "The buffering read stream should only be used when reading.");
if (existingBufferingReadStream != null)
{
Debug.Assert(this.useBufferingReadStream.HasValue, "UseBufferingReadStream must have been set.");
return(TaskUtils.GetCompletedTask(existingBufferingReadStream));
}
}
Task <Stream> task = streamFuncAsync();
ValidateMessageStreamTask(task, isRequest);
// Wrap it in a non-disposing stream if requested
task = task.FollowOnSuccessWith(
streamTask =>
{
Stream messageStream = streamTask.Result;
ValidateMessageStream(messageStream, isRequest);
// When reading, wrap the stream in a byte counting stream if a max message size was specified.
// When requested, wrap the stream in a non-disposing stream.
bool needByteCountingStream = !this.writing && this.maxMessageSize > 0;
if (this.disableMessageStreamDisposal && needByteCountingStream)
{
messageStream = MessageStreamWrapper.CreateNonDisposingStreamWithMaxSize(messageStream, this.maxMessageSize);
}
else if (this.disableMessageStreamDisposal)
{
messageStream = MessageStreamWrapper.CreateNonDisposingStream(messageStream);
}
else if (needByteCountingStream)
{
messageStream = MessageStreamWrapper.CreateStreamWithMaxSize(messageStream, this.maxMessageSize);
}
return(messageStream);
});
// When we are reading, also buffer the input stream
if (!this.writing)
{
task = task
.FollowOnSuccessWithTask(
streamTask =>
{
return(BufferedReadStream.BufferStreamAsync(streamTask.Result));
})
.FollowOnSuccessWith(
streamTask =>
{
BufferedReadStream bufferedReadStream = streamTask.Result;
return((Stream)bufferedReadStream);
});
// If requested also create a buffering stream for payload kind detection
if (this.useBufferingReadStream == true)
{
task = task.FollowOnSuccessWith(
streamTask =>
{
Stream messageStream = streamTask.Result;
this.bufferingReadStream = new BufferingReadStream(messageStream);
messageStream = this.bufferingReadStream;
return(messageStream);
});
}
}
return(task);
}