public static TryCatch <IQueryPipelineStage> MonadicCreate(
ExecutionEnvironment executionEnvironment,
int offsetCount,
CosmosElement continuationToken,
CancellationToken cancellationToken,
MonadicCreatePipelineStage monadicCreatePipelineStage)
{
TryCatch <IQueryPipelineStage> tryCreate = executionEnvironment switch
{
ExecutionEnvironment.Client => ClientSkipQueryPipelineStage.MonadicCreate(
offsetCount,
continuationToken,
cancellationToken,
monadicCreatePipelineStage),
ExecutionEnvironment.Compute => ComputeSkipQueryPipelineStage.MonadicCreate(
offsetCount,
continuationToken,
cancellationToken,
monadicCreatePipelineStage),
_ => throw new ArgumentException($"Unknown {nameof(ExecutionEnvironment)}: {executionEnvironment}"),
};
return(tryCreate);
}
public static TryCatch <IQueryPipelineStage> MonadicCreate(
int offsetCount,
CosmosElement continuationToken,
CancellationToken cancellationToken,
MonadicCreatePipelineStage monadicCreatePipelineStage)
{
if (monadicCreatePipelineStage == null)
{
throw new ArgumentNullException(nameof(monadicCreatePipelineStage));
}
OffsetContinuationToken offsetContinuationToken;
if (continuationToken != null)
{
if (!OffsetContinuationToken.TryParse(continuationToken.ToString(), out offsetContinuationToken))
{
return(TryCatch <IQueryPipelineStage> .FromException(
new MalformedContinuationTokenException(
$"Invalid {nameof(SkipQueryPipelineStage)}: {continuationToken}.")));
}
}
else
{
offsetContinuationToken = new OffsetContinuationToken(offsetCount, null);
}
if (offsetContinuationToken.Offset > offsetCount)
{
return(TryCatch <IQueryPipelineStage> .FromException(
new MalformedContinuationTokenException(
"offset count in continuation token can not be greater than the offsetcount in the query.")));
}
CosmosElement sourceToken;
if (offsetContinuationToken.SourceToken != null)
{
TryCatch <CosmosElement> tryParse = CosmosElement.Monadic.Parse(offsetContinuationToken.SourceToken);
if (tryParse.Failed)
{
return(TryCatch <IQueryPipelineStage> .FromException(
new MalformedContinuationTokenException(
message : $"source token: '{offsetContinuationToken.SourceToken ?? "<null>"}' is not valid.",
innerException : tryParse.Exception)));
}
sourceToken = tryParse.Result;
}
else
{
sourceToken = null;
}
TryCatch <IQueryPipelineStage> tryCreateSource = monadicCreatePipelineStage(sourceToken, cancellationToken);
if (tryCreateSource.Failed)
{
return(tryCreateSource);
}
IQueryPipelineStage stage = new ClientSkipQueryPipelineStage(
tryCreateSource.Result,
cancellationToken,
offsetContinuationToken.Offset);
return(TryCatch <IQueryPipelineStage> .FromResult(stage));
}
