internal async Task <OfferV2> GetOfferV2Async( string targetRID, CancellationToken cancellationToken = default(CancellationToken)) { if (string.IsNullOrWhiteSpace(targetRID)) { throw new ArgumentNullException(targetRID); } QueryDefinition queryDefinition = new QueryDefinition("select * from root r where r.offerResourceId= @targetRID"); queryDefinition.WithParameter("@targetRID", targetRID); FeedIterator <OfferV2> databaseStreamIterator = this.GetOfferQueryIterator <OfferV2>( queryDefinition); OfferV2 offerV2 = await this.SingleOrDefaultAsync <OfferV2>(databaseStreamIterator); if (offerV2 == null) { throw new CosmosException(HttpStatusCode.NotFound, "Throughput is not configured"); } return(offerV2); }
public override FeedIterator <T> GetItemQueryIterator <T>( QueryDefinition queryDefinition, string continuationToken = null, QueryRequestOptions requestOptions = null) { requestOptions = requestOptions ?? new QueryRequestOptions(); if (requestOptions.IsEffectivePartitionKeyRouting) { requestOptions.PartitionKey = null; } if (!(this.GetItemQueryStreamIterator( queryDefinition, continuationToken, requestOptions) is FeedIteratorInternal feedIterator)) { throw new InvalidOperationException($"Expected a FeedIteratorInternal."); } return(new FeedIteratorCore <T>( feedIterator: feedIterator, responseCreator: this.ClientContext.ResponseFactory.CreateQueryFeedUserTypeResponse <T>)); }
internal static FeedIteratorCore CreateForPartitionedResource( ContainerCore containerCore, Uri resourceLink, ResourceType resourceType, QueryDefinition queryDefinition, string continuationToken, FeedTokenInternal feedTokenInternal, QueryRequestOptions options) { if (containerCore == null) { throw new ArgumentNullException(nameof(containerCore)); } return new FeedIteratorCore( containerCore: containerCore, clientContext: containerCore.ClientContext, resourceLink: resourceLink, resourceType: resourceType, queryDefinition: queryDefinition, continuationToken: continuationToken, feedTokenInternal: feedTokenInternal, options: options); }
public abstract FeedIterator <T> GetItemQueryIterator <T>( FeedRange feedRange, QueryDefinition queryDefinition, string continuationToken = null, QueryRequestOptions requestOptions = null);
public abstract FeedIterator GetItemQueryStreamIterator( FeedRange feedRange, QueryDefinition queryDefinition, string continuationToken, QueryRequestOptions requestOptions = null);
/// <summary> /// This method creates a query for users under an database using a SQL statement. It returns a FeedIterator. /// For more information on preparing SQL statements with parameterized values, please see <see cref="QueryDefinition"/> overload. /// </summary> /// <param name="queryDefinition">The cosmos SQL query definition.</param> /// <param name="continuationToken">(Optional) The continuation token in the Azure Cosmos DB service.</param> /// <param name="requestOptions">(Optional) The options for the user query request <see cref="QueryRequestOptions"/></param> /// <returns>An iterator to go through the users</returns> /// <example> /// This create the type feed iterator for users with queryDefinition as input. /// <code language="c#"> /// <![CDATA[ /// string queryText = "SELECT * FROM c where c.id like @testId"; /// QueryDefinition queryDefinition = new QueryDefinition(queryText); /// queryDefinition.WithParameter("@testId", "testUserId"); /// FeedIterator<UserProperties> resultSet = this.cosmosDatabase.GetUserQueryIterator<UserProperties>(queryDefinition); /// while (feedIterator.HasMoreResults) /// { /// foreach (UserProperties properties in await feedIterator.ReadNextAsync()) /// { /// Console.WriteLine(properties.Id); /// } /// } /// ]]> /// </code> /// </example> public abstract FeedIterator <T> GetUserQueryIterator <T>( QueryDefinition queryDefinition, string continuationToken = null, QueryRequestOptions requestOptions = null);
/// <summary> /// This method creates a query for containers under an database using a SQL statement. It returns a FeedIterator. /// For more information on preparing SQL statements with parameterized values, please see <see cref="QueryDefinition"/> overload. /// </summary> /// <param name="queryDefinition">The cosmos SQL query definition.</param> /// <param name="continuationToken">The continuation token in the Azure Cosmos DB service.</param> /// <param name="requestOptions">(Optional) The options for the container request <see cref="QueryRequestOptions"/></param> /// <returns>An iterator to go through the containers</returns> /// <example> /// This create the stream feed iterator for containers with queryDefinition as input. /// <code language="c#"> /// <![CDATA[ /// string queryText = "SELECT * FROM c where c.id like '%testId%'"; /// QueryDefinition queryDefinition = new QueryDefinition(queryText); /// FeedIterator resultSet = this.cosmosDatabase.GetContainerQueryStreamIterator(queryDefinition); /// while (feedIterator.HasMoreResults) /// { /// using (ResponseMessage response = await feedIterator.ReadNextAsync()) /// { /// using (StreamReader sr = new StreamReader(response.Content)) /// using (JsonTextReader jtr = new JsonTextReader(sr)) /// { /// JObject result = JObject.Load(jtr); /// } /// } /// } /// ]]> /// </code> /// </example> public abstract FeedIterator GetContainerQueryStreamIterator( QueryDefinition queryDefinition, string continuationToken = null, QueryRequestOptions requestOptions = null);
/// <summary> /// Used in the compute gateway to support legacy gateway interface. /// </summary> public override async Task <TryExecuteQueryResult> TryExecuteQueryAsync( QueryFeatures supportedQueryFeatures, QueryDefinition queryDefinition, string continuationToken, FeedRangeInternal feedRangeInternal, QueryRequestOptions requestOptions, CancellationToken cancellationToken = default(CancellationToken)) { if (queryDefinition == null) { throw new ArgumentNullException(nameof(queryDefinition)); } if (requestOptions == null) { throw new ArgumentNullException(nameof(requestOptions)); } if (feedRangeInternal != null) { // The user has scoped down to a physical partition or logical partition. // In either case let the query execute as a passthrough. QueryIterator passthroughQueryIterator = QueryIterator.Create( client: this.queryClient, clientContext: this.ClientContext, sqlQuerySpec: queryDefinition.ToSqlQuerySpec(), continuationToken: continuationToken, feedRangeInternal: feedRangeInternal, queryRequestOptions: requestOptions, resourceLink: this.LinkUri, isContinuationExpected: false, allowNonValueAggregateQuery: true, forcePassthrough: true, // Forcing a passthrough, since we don't want to get the query plan nor try to rewrite it. partitionedQueryExecutionInfo: null); return(new QueryPlanIsSupportedResult(passthroughQueryIterator)); } cancellationToken.ThrowIfCancellationRequested(); Documents.PartitionKeyDefinition partitionKeyDefinition; if (requestOptions.Properties != null && requestOptions.Properties.TryGetValue("x-ms-query-partitionkey-definition", out object partitionKeyDefinitionObject)) { if (partitionKeyDefinitionObject is Documents.PartitionKeyDefinition definition) { partitionKeyDefinition = definition; } else { throw new ArgumentException( "partitionkeydefinition has invalid type", nameof(partitionKeyDefinitionObject)); } } else { ContainerQueryProperties containerQueryProperties = await this.queryClient.GetCachedContainerQueryPropertiesAsync( this.LinkUri, requestOptions.PartitionKey, cancellationToken); partitionKeyDefinition = containerQueryProperties.PartitionKeyDefinition; } QueryPlanHandler queryPlanHandler = new QueryPlanHandler(this.queryClient); TryCatch <(PartitionedQueryExecutionInfo queryPlan, bool supported)> tryGetQueryInfoAndIfSupported = await queryPlanHandler.TryGetQueryInfoAndIfSupportedAsync( supportedQueryFeatures, queryDefinition.ToSqlQuerySpec(), partitionKeyDefinition, requestOptions.PartitionKey.HasValue, cancellationToken); if (tryGetQueryInfoAndIfSupported.Failed) { return(new FailedToGetQueryPlanResult(tryGetQueryInfoAndIfSupported.Exception)); } (PartitionedQueryExecutionInfo queryPlan, bool supported) = tryGetQueryInfoAndIfSupported.Result; TryExecuteQueryResult tryExecuteQueryResult; if (supported) { QueryIterator queryIterator = QueryIterator.Create( client: this.queryClient, clientContext: this.ClientContext, sqlQuerySpec: queryDefinition.ToSqlQuerySpec(), continuationToken: continuationToken, feedRangeInternal: feedRangeInternal, queryRequestOptions: requestOptions, resourceLink: this.LinkUri, isContinuationExpected: false, allowNonValueAggregateQuery: true, forcePassthrough: false, partitionedQueryExecutionInfo: queryPlan); tryExecuteQueryResult = new QueryPlanIsSupportedResult(queryIterator); } else { tryExecuteQueryResult = new QueryPlanNotSupportedResult(queryPlan); } return(tryExecuteQueryResult); }
public ParametersListAdapter(QueryDefinition queryDefinition) { this.queryDefinition = queryDefinition; }
/// <summary> /// Used in the compute gateway to support legacy gateway interface. /// </summary> internal async Task <((Exception, PartitionedQueryExecutionInfo), (bool, QueryIterator))> TryExecuteQueryAsync( QueryFeatures supportedQueryFeatures, QueryDefinition queryDefinition, string continuationToken, QueryRequestOptions requestOptions, CancellationToken cancellationToken = default(CancellationToken)) { if (queryDefinition == null) { throw new ArgumentNullException(nameof(queryDefinition)); } if (requestOptions == null) { throw new ArgumentNullException(nameof(requestOptions)); } cancellationToken.ThrowIfCancellationRequested(); Documents.PartitionKeyDefinition partitionKeyDefinition; if (requestOptions.Properties != null && requestOptions.Properties.TryGetValue("x-ms-query-partitionkey-definition", out object partitionKeyDefinitionObject)) { if (partitionKeyDefinitionObject is Documents.PartitionKeyDefinition definition) { partitionKeyDefinition = definition; } else { throw new ArgumentException( "partitionkeydefinition has invalid type", nameof(partitionKeyDefinitionObject)); } } else { ContainerQueryProperties containerQueryProperties = await this.queryClient.GetCachedContainerQueryPropertiesAsync( this.LinkUri, requestOptions.PartitionKey, cancellationToken); partitionKeyDefinition = containerQueryProperties.PartitionKeyDefinition; } QueryPlanHandler queryPlanHandler = new QueryPlanHandler(this.queryClient); ((Exception exception, PartitionedQueryExecutionInfo partitionedQueryExecutionInfo), bool supported) = await queryPlanHandler.TryGetQueryInfoAndIfSupportedAsync( supportedQueryFeatures, queryDefinition.ToSqlQuerySpec(), partitionKeyDefinition, requestOptions.PartitionKey.HasValue, cancellationToken); if (exception != null) { return((exception, null), (false, null)); } QueryIterator queryIterator; if (supported) { queryIterator = QueryIterator.Create( client: this.queryClient, clientContext: this.ClientContext, sqlQuerySpec: queryDefinition.ToSqlQuerySpec(), continuationToken: continuationToken, queryRequestOptions: requestOptions, resourceLink: this.LinkUri, isContinuationExpected: false, allowNonValueAggregateQuery: true, partitionedQueryExecutionInfo: partitionedQueryExecutionInfo); } else { queryIterator = null; } return((null, partitionedQueryExecutionInfo), (supported, queryIterator)); }
public override FeedIteratorInternal GetReadFeedIterator(QueryDefinition queryDefinition, QueryRequestOptions queryRequestOptions, string resourceLink, Documents.ResourceType resourceType, string continuationToken, int pageSize) { return(base.GetReadFeedIterator(queryDefinition, queryRequestOptions, resourceLink, resourceType, continuationToken, pageSize)); }