public static async Task <TryCatch <IDocumentQueryExecutionComponent> > TryCreateTopDocumentQueryExecutionComponentAsync(
int topCount,
CosmosElement requestContinuationToken,
Func <CosmosElement, Task <TryCatch <IDocumentQueryExecutionComponent> > > tryCreateSourceAsync)
{
if (topCount < 0)
{
throw new ArgumentException($"{nameof(topCount)}: {topCount} must be a non negative number.");
}
if (tryCreateSourceAsync == null)
{
throw new ArgumentNullException(nameof(tryCreateSourceAsync));
}
TopContinuationToken topContinuationToken;
if (requestContinuationToken != null)
{
if (!TopContinuationToken.TryParse(requestContinuationToken.ToString(), out topContinuationToken))
{
return(TryCatch <IDocumentQueryExecutionComponent> .FromException(
new MalformedContinuationTokenException($"Malformed {nameof(LimitContinuationToken)}: {requestContinuationToken}.")));
}
}
else
{
topContinuationToken = new TopContinuationToken(topCount, null);
}
if (topContinuationToken.Top > topCount)
{
return(TryCatch <IDocumentQueryExecutionComponent> .FromException(
new MalformedContinuationTokenException($"{nameof(TopContinuationToken.Top)} in {nameof(TopContinuationToken)}: {requestContinuationToken}: {topContinuationToken.Top} can not be greater than the top count in the query: {topCount}.")));
}
CosmosElement sourceToken;
if (topContinuationToken.SourceToken != null)
{
if (!CosmosElement.TryParse(topContinuationToken.SourceToken, out sourceToken))
{
return(TryCatch <IDocumentQueryExecutionComponent> .FromException(
new MalformedContinuationTokenException($"{nameof(TopContinuationToken.SourceToken)} in {nameof(TopContinuationToken)}: {requestContinuationToken}: {topContinuationToken.SourceToken} was malformed.")));
}
}
else
{
sourceToken = null;
}
return((await tryCreateSourceAsync(sourceToken))
.Try <IDocumentQueryExecutionComponent>((source) => new ClientTakeDocumentQueryExecutionComponent(
source,
topContinuationToken.Top,
TakeEnum.Top)));
}
public static TryCatch <IQueryPipelineStage> MonadicCreateTopStage(
int topCount,
CosmosElement requestContinuationToken,
CancellationToken cancellationToken,
MonadicCreatePipelineStage monadicCreatePipelineStage)
{
if (topCount < 0)
{
throw new ArgumentException($"{nameof(topCount)}: {topCount} must be a non negative number.");
}
if (monadicCreatePipelineStage == null)
{
throw new ArgumentNullException(nameof(monadicCreatePipelineStage));
}
TopContinuationToken topContinuationToken;
if (requestContinuationToken != null)
{
if (!TopContinuationToken.TryParse(requestContinuationToken.ToString(), out topContinuationToken))
{
return(TryCatch <IQueryPipelineStage> .FromException(
new MalformedContinuationTokenException(
$"Malformed {nameof(LimitContinuationToken)}: {requestContinuationToken}.")));
}
}
else
{
topContinuationToken = new TopContinuationToken(topCount, null);
}
if (topContinuationToken.Top > topCount)
{
return(TryCatch <IQueryPipelineStage> .FromException(
new MalformedContinuationTokenException(
$"{nameof(TopContinuationToken.Top)} in {nameof(TopContinuationToken)}: {requestContinuationToken}: {topContinuationToken.Top} can not be greater than the top count in the query: {topCount}.")));
}
CosmosElement sourceToken;
if (topContinuationToken.SourceToken != null)
{
TryCatch <CosmosElement> tryParse = CosmosElement.Monadic.Parse(topContinuationToken.SourceToken);
if (tryParse.Failed)
{
return(TryCatch <IQueryPipelineStage> .FromException(
new MalformedContinuationTokenException(
message : $"{nameof(TopContinuationToken.SourceToken)} in {nameof(TopContinuationToken)}: {requestContinuationToken}: {topContinuationToken.SourceToken ?? "<null>"} was malformed.",
innerException : tryParse.Exception)));
}
sourceToken = tryParse.Result;
}
else
{
sourceToken = null;
}
TryCatch <IQueryPipelineStage> tryCreateSource = monadicCreatePipelineStage(sourceToken, cancellationToken);
if (tryCreateSource.Failed)
{
return(tryCreateSource);
}
IQueryPipelineStage stage = new ClientTakeQueryPipelineStage(
tryCreateSource.Result,
cancellationToken,
topContinuationToken.Top,
TakeEnum.Top);
return(TryCatch <IQueryPipelineStage> .FromResult(stage));
}
