private async Task <CosmosQueryResponse> QueryRequestExecutor(
string continuationToken,
object state,
CancellationToken cancellationToken)
{
DocumentQuery <CosmosQueryResponse> documentQuery = (DocumentQuery <CosmosQueryResponse>)state;
// DEVNOTE: Remove try catch once query pipeline is converted to exceptionless
try
{
return(await documentQuery.ExecuteNextQueryStreamAsync(cancellationToken));
}
catch (DocumentClientException exception)
{
return(new CosmosQueryResponse(
errorMessage: exception.Message,
httpStatusCode: exception.StatusCode.HasValue ? exception.StatusCode.Value : HttpStatusCode.InternalServerError,
retryAfter: exception.RetryAfter));
}catch (AggregateException ae)
{
DocumentClientException exception = ae.InnerException as DocumentClientException;
if (exception == null)
{
throw;
}
return(new CosmosQueryResponse(
errorMessage: exception.Message,
httpStatusCode: exception.StatusCode.HasValue ? exception.StatusCode.Value : HttpStatusCode.InternalServerError,
retryAfter: exception.RetryAfter));
}
}