private static async Task <ZResponse> _RegisterRequestAndWaitResponseAsync(ZRequest request)
{
// create request metadata
var requestMetadata = new ZRequestMetadata(request);
// register request to help route the response back
_requestsPool.Add(requestMetadata);
if (_internalConnectionState)
{
// we are know that we connected to host, so just send it
// and even if this is not the case, then the request will be rejected further down the pipeline
_SendRequest(request);
await _WaitResponseAsync(request, requestMetadata);
}
else
{
requestMetadata.Response.StatusCode = ZResponseStatusCode.Declined;
}
_requestsPool.Remove(requestMetadata);
return(requestMetadata.Response);
}
private static void _SendRequest(ZRequest request)
{
// convert request to byte array
var requestBytes = request.ToByteArray();
// send request to ZClient
_client.SendRequest(requestBytes);
}
public ZRequestMetadata(ZRequest request)
{
Request = request;
Response = new ZResponse(request)
{
StatusCode = RQ_DEFAULT_STATUS
};
TaskCompletionSource = request.Method == ZRequestMethod.Get
? new TaskCompletionSource <object>()
: null;
}
private static async Task <ZResponse> _TryCloseStreamAndWaitResponseAsync(ZRequest request)
{
// find stream metadata
var streamMetadata = _streamsPool.First(i => i.StreamCommand == request.RequestCommand);
// try close stream
var closeStreamResponse = streamMetadata.IsRejected
? new ZResponse(request)
{
StatusCode = ZResponseStatusCode.Rejected
}
: await _RegisterRequestAndWaitResponseAsync(request); // send close stream request
// in any case, we need to stop the stream of packets, no matter what the response was
_streamsPool.Remove(streamMetadata);
return(closeStreamResponse);
}
private static async Task <ZResponse> _TryRegisterStreamAndWaitResponseAsync(ZRequest request, ZPacketsStreamCallback onPacketsReceivedCallback, ZStreamRejectedCallback streamRejectedCallback)
{
// create stream metadata to register it
var streamMetadata = new ZStreamMetadata(request.RequestCommand, onPacketsReceivedCallback, streamRejectedCallback);
// register stream to help find it and pass packets
_streamsPool.Add(streamMetadata);
// send open stream request
var openStreamResponse = await _RegisterRequestAndWaitResponseAsync(request);
// check response status code
if (openStreamResponse.StatusCode != ZResponseStatusCode.Ok)
{
_streamsPool.Remove(streamMetadata);
}
return(openStreamResponse);
}
/// <summary>
/// Sends request to ZClient to close packets streaming back as an asynchronous operation.
/// The stream will close in any case, but despite this, it remains possible to react to the response
/// </summary>
/// <param name="request">A request to close stream</param>
/// <returns>The task object representing the asynchronous operation</returns>
public static async Task <ZResponse> CloseStreamAsync(ZRequest request)
{
// check incoming arguments
if (request == null)
{
throw new ArgumentNullException(nameof(request));
}
// check is it exists in pool
if (_streamsPool.All(i => i.StreamCommand != request.RequestCommand))
{
throw new ArgumentException($"This stream request is doesn't exists in streams pool. {request}");
}
// send request and get response
var response = await _TryCloseStreamAndWaitResponseAsync(request);
return(response);
}
/// <summary>
/// Sends request to ZClient and routes response back as an asynchronous operation
/// </summary>
/// <param name="request">A request to send</param>
/// <returns>The task object representing the asynchronous operation</returns>
public static async Task <ZResponse> GetResponseAsync(ZRequest request)
{
// check incoming argument
if (request == null)
{
throw new ArgumentNullException(nameof(request));
}
// check is it already exists in pool
if (_requestsPool.Any(i => i.RequestGuid == request.RequestGuid))
{
throw new ArgumentException($"This request is already exists in requests pool. {request}");
}
// send request and get response
var response = await _RegisterRequestAndWaitResponseAsync(request);
return(response);
}
private static async Task _WaitResponseAsync(ZRequest request, ZRequestMetadata metadata)
{
if (request.Method == ZRequestMethod.Get)
{
// create request timeout task
var timeoutTask = Task.Delay(RQ_TIMEOUT);
// wait response or timeout
var task = await Task.WhenAny(metadata.TaskCompletionSource.Task, timeoutTask);
// check if it's rejected, or not
// while waiting for a response, a disconnection may occur, which will close the request with the Rejected status
if (task == timeoutTask || metadata.Response.StatusCode == RQ_DEFAULT_STATUS)
{
// set result status code
metadata.Response.StatusCode = task == timeoutTask ? ZResponseStatusCode.Timeout : ZResponseStatusCode.Ok;
}
}
else
{
metadata.Response.StatusCode = ZResponseStatusCode.Ok;
}
}
/// <summary>
/// Sends request to ZClient to open packets streaming back as an asynchronous operation
/// </summary>
/// <param name="request">A request to open stream</param>
/// <param name="onPacketsReceivedCallback">An <see cref="ZPacketsStreamCallback"/> delegate that references the method to invoke when the stream packets received</param>
/// <param name="streamRejectedCallback">(Optional) An <see cref="ZStreamRejectedCallback"/> delegate that references the method to invoke when the stream was rejected</param>
/// <returns>The task object representing the asynchronous operation</returns>
public static async Task <ZResponse> OpenStreamAsync(ZRequest request, ZPacketsStreamCallback onPacketsReceivedCallback, ZStreamRejectedCallback streamRejectedCallback = null)
{
// check incoming arguments
if (request == null)
{
throw new ArgumentNullException(nameof(request));
}
if (onPacketsReceivedCallback == null)
{
throw new ArgumentNullException(nameof(onPacketsReceivedCallback));
}
// check is it already exists in pool
if (_streamsPool.Any(i => i.StreamCommand == request.RequestCommand))
{
throw new ArgumentException($"This stream request is already exists in streams pool. {request}");
}
// send request and get response
var response = await _TryRegisterStreamAndWaitResponseAsync(request, onPacketsReceivedCallback, streamRejectedCallback);
return(response);
}
