public async ValueTask <HandshakeRequestMessage> ReadHandshakeAsync(CancellationToken cancellationToken = default)
{
var result = await _protocolReader.ReadAsync(new HubHandshakeMessageReader(), _maximumMessageSize, cancellationToken).ConfigureAwait(false);
var message = result.Message;
_protocolReader.Advance();
return(message);
}
public async ValueTask <HttpRequestMessage> ReadRequestAsync()
{
var content = new HttpBodyContent();
var headerReader = new Http1RequestMessageReader(content);
var result = await _reader.ReadAsync(headerReader).ConfigureAwait(false);
if (result.IsCompleted)
{
throw new ConnectionAbortedException();
}
var request = result.Message;
// TODO: Handle upgrade
if (content.Headers.ContentLength != null)
{
content.SetStream(new HttpBodyStream(_reader, new ContentLengthHttpBodyReader(request.Content.Headers.ContentLength.Value)));
}
else if (request.Headers.TransferEncodingChunked.HasValue)
{
content.SetStream(new HttpBodyStream(_reader, new ChunkedHttpBodyReader()));
}
else
{
content.SetStream(new HttpBodyStream(_reader, new ContentLengthHttpBodyReader(0)));
}
_reader.Advance();
return(request);
}
public async ValueTask <HttpResponseMessage> SendAsync(HttpRequestMessage requestMessage, HttpCompletionOption completionOption = HttpCompletionOption.ResponseHeadersRead)
{
// Write request message headers
_messageWriter.WriteMessage(requestMessage, _connection.Transport.Output);
// Write the body directly
if (requestMessage.Content != null)
{
await requestMessage.Content.CopyToAsync(_connection.Transport.Output.AsStream()).ConfigureAwait(false);
}
await _connection.Transport.Output.FlushAsync().ConfigureAwait(false);
var content = new HttpBodyContent();
var headerReader = new Http1ResponseMessageReader(content);
var result = await _reader.ReadAsync(headerReader).ConfigureAwait(false);
if (result.IsCompleted)
{
throw new ConnectionAbortedException();
}
var response = result.Message;
// TODO: Handle upgrade
if (content.Headers.ContentLength != null)
{
content.SetStream(new HttpBodyStream(_reader, new ContentLengthHttpBodyReader(response.Content.Headers.ContentLength.Value)));
}
else if (response.Headers.TransferEncodingChunked.HasValue)
{
content.SetStream(new HttpBodyStream(_reader, new ChunkedHttpBodyReader()));
}
else
{
content.SetStream(new HttpBodyStream(_reader, new ContentLengthHttpBodyReader(0)));
}
_reader.Advance();
return(response);
}
