private static async Task <List <CosmosElement> > CreateAndDrainAsync( IReadOnlyList <IReadOnlyList <CosmosElement> > pages, ExecutionEnvironment executionEnvironment, int offsetCount, CosmosElement continuationToken) { IQueryPipelineStage source = new MockQueryPipelineStage(pages); TryCatch <IQueryPipelineStage> tryCreateSkipQueryPipelineStage = SkipQueryPipelineStage.MonadicCreate( executionEnvironment: executionEnvironment, offsetCount: offsetCount, continuationToken: continuationToken, cancellationToken: default,
private static async Task <(List <CosmosElement>, long)> CreateAndDrainAsync( IReadOnlyList <IReadOnlyList <CosmosElement> > pages, ExecutionEnvironment executionEnvironment, int takeCount, CosmosElement continuationToken) { MockQueryPipelineStage source = new MockQueryPipelineStage(pages); TryCatch <IQueryPipelineStage> tryCreateSkipQueryPipelineStage = TakeQueryPipelineStage.MonadicCreateLimitStage( executionEnvironment: executionEnvironment, limitCount: takeCount, requestContinuationToken: continuationToken, cancellationToken: default,
private static async Task <List <CosmosElement> > CreateAndDrainAsync( IReadOnlyList <IReadOnlyList <CosmosElement> > pages, ExecutionEnvironment executionEnvironment, CosmosElement continuationToken, DistinctQueryType distinctQueryType) { IQueryPipelineStage source = new MockQueryPipelineStage(pages); TryCatch <IQueryPipelineStage> tryCreateDistinctQueryPipelineStage = DistinctQueryPipelineStage.MonadicCreate( executionEnvironment: executionEnvironment, requestContinuation: continuationToken, distinctQueryType: distinctQueryType, cancellationToken: default,
public static MockQueryPipelineStage Create( IReadOnlyList <IReadOnlyList <CosmosElement> > pages, CosmosElement continuationToken) { MockQueryPipelineStage stage = new MockQueryPipelineStage(pages); if (continuationToken != null) { CosmosNumber index = continuationToken as CosmosNumber; stage.PageIndex = Number64.ToLong(index.Value); } return(stage); }
private static async Task <List <CosmosElement> > CreateAndDrainAsync( IReadOnlyList <IReadOnlyList <CosmosElement> > pages, ExecutionEnvironment executionEnvironment, CosmosElement continuationToken, IReadOnlyDictionary <string, AggregateOperator?> groupByAliasToAggregateType, IReadOnlyList <string> orderedAliases, bool hasSelectValue) { IQueryPipelineStage source = new MockQueryPipelineStage(pages); TryCatch <IQueryPipelineStage> tryCreateGroupByStage = GroupByQueryPipelineStage.MonadicCreate( executionEnvironment: executionEnvironment, continuationToken: continuationToken, cancellationToken: default,