@azure/cosmos 4.8.0 → 4.9.0

package/dist/browser/queryExecutionContext/pipelinedQueryExecutionContext.js

@@ -4,12 +4,16 @@ import { OrderByEndpointComponent } from "./EndpointComponent/OrderByEndpointCom
 import { OrderedDistinctEndpointComponent } from "./EndpointComponent/OrderedDistinctEndpointComponent.js";
 import { UnorderedDistinctEndpointComponent } from "./EndpointComponent/UnorderedDistinctEndpointComponent.js";
 import { GroupByEndpointComponent } from "./EndpointComponent/GroupByEndpointComponent.js";
-import { getInitialHeader, mergeHeaders } from "./headerUtils.js";
 import { OrderByQueryExecutionContext } from "./orderByQueryExecutionContext.js";
 import { ParallelQueryExecutionContext } from "./parallelQueryExecutionContext.js";
 import { GroupByValueEndpointComponent } from "./EndpointComponent/GroupByValueEndpointComponent.js";
 import { NonStreamingOrderByDistinctEndpointComponent } from "./EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js";
 import { NonStreamingOrderByEndpointComponent } from "./EndpointComponent/NonStreamingOrderByEndpointComponent.js";
+import { rejectContinuationTokenForUnsupportedQueries, QueryTypes, } from "./QueryValidationHelper.js";
+import { parseContinuationTokenFields } from "./ContinuationTokenParser.js";
+import { LegacyFetchImplementation } from "./LegacyFetchImplementation.js";
+import { QueryControlFetchImplementation } from "./QueryControlFetchImplementation.js";
+import { QueryExecution } from "../common/constants.js";
 /** @hidden */
 export class PipelinedQueryExecutionContext {
     clientContext;
@@ -19,13 +23,8 @@ export class PipelinedQueryExecutionContext {
     partitionedQueryExecutionInfo;
     emitRawOrderByPayload;
     fetchBuffer;
-    fetchMoreRespHeaders;
     endpoint;
-    pageSize;
-    vectorSearchBufferSize = 0;
-    static DEFAULT_PAGE_SIZE = 10;
-    static DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE = 50000;
-    nonStreamingOrderBy = false;
+    fetchImplementation;
     constructor(clientContext, collectionLink, query, options, partitionedQueryExecutionInfo, correlatedActivityId, emitRawOrderByPayload = false) {
         this.clientContext = clientContext;
         this.collectionLink = collectionLink;
@@ -33,131 +32,161 @@ export class PipelinedQueryExecutionContext {
         this.options = options;
         this.partitionedQueryExecutionInfo = partitionedQueryExecutionInfo;
         this.emitRawOrderByPayload = emitRawOrderByPayload;
-        this.endpoint = null;
-        this.pageSize = options["maxItemCount"];
-        if (this.pageSize === undefined) {
-            this.pageSize = PipelinedQueryExecutionContext.DEFAULT_PAGE_SIZE;
-        }
-        // Pick between Nonstreaming and streaming endpoints
-        this.nonStreamingOrderBy = partitionedQueryExecutionInfo.queryInfo.hasNonStreamingOrderBy;
-        // Pick between parallel vs order by execution context
-        const sortOrders = partitionedQueryExecutionInfo.queryInfo.orderBy;
-        // TODO: Currently we don't get any field from backend to determine streaming queries
-        if (this.nonStreamingOrderBy) {
-            if (!options.allowUnboundedNonStreamingQueries) {
-                this.checkQueryConstraints(partitionedQueryExecutionInfo.queryInfo);
-            }
-            this.vectorSearchBufferSize = this.calculateVectorSearchBufferSize(partitionedQueryExecutionInfo.queryInfo, options);
-            const maxBufferSize = options["vectorSearchBufferSize"]
-                ? options["vectorSearchBufferSize"]
-                : PipelinedQueryExecutionContext.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
-            if (this.vectorSearchBufferSize > maxBufferSize) {
-                throw new ErrorResponse(`Executing a vector search query with TOP or OFFSET + LIMIT value ${this.vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} ` +
-                    `is not allowed`);
-            }
-            const distinctType = partitionedQueryExecutionInfo.queryInfo.distinctType;
-            const context = new ParallelQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, this.partitionedQueryExecutionInfo, correlatedActivityId);
-            if (distinctType === "None") {
-                this.endpoint = new NonStreamingOrderByEndpointComponent(context, sortOrders, this.vectorSearchBufferSize, partitionedQueryExecutionInfo.queryInfo.offset, this.emitRawOrderByPayload);
-            }
-            else {
-                this.endpoint = new NonStreamingOrderByDistinctEndpointComponent(context, partitionedQueryExecutionInfo.queryInfo, this.vectorSearchBufferSize, this.emitRawOrderByPayload);
-            }
+        // Validate that queryInfo is present in partitioned query execution info
+        if (!partitionedQueryExecutionInfo.queryInfo) {
+            throw new ErrorResponse("Query execution requires valid query plan information. " +
+                "The partitioned query execution info is missing queryInfo. " +
+                "This may indicate an invalid query or a problem with query planning.");
         }
-        else {
-            if (Array.isArray(sortOrders) && sortOrders.length > 0) {
-                // Need to wrap orderby execution context in endpoint component, since the data is nested as a \
-                //      "payload" property.
-                this.endpoint = new OrderByEndpointComponent(new OrderByQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, this.partitionedQueryExecutionInfo, correlatedActivityId), this.emitRawOrderByPayload);
-            }
-            else {
-                this.endpoint = new ParallelQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, this.partitionedQueryExecutionInfo, correlatedActivityId);
-            }
-            if (Object.keys(partitionedQueryExecutionInfo.queryInfo.groupByAliasToAggregateType).length >
-                0 ||
-                partitionedQueryExecutionInfo.queryInfo.aggregates.length > 0 ||
-                partitionedQueryExecutionInfo.queryInfo.groupByExpressions.length > 0) {
-                if (partitionedQueryExecutionInfo.queryInfo.hasSelectValue) {
-                    this.endpoint = new GroupByValueEndpointComponent(this.endpoint, partitionedQueryExecutionInfo.queryInfo);
-                }
-                else {
-                    this.endpoint = new GroupByEndpointComponent(this.endpoint, partitionedQueryExecutionInfo.queryInfo);
-                }
-            }
-            // If distinct then add that to the pipeline
-            const distinctType = partitionedQueryExecutionInfo.queryInfo.distinctType;
-            if (distinctType === "Ordered") {
-                this.endpoint = new OrderedDistinctEndpointComponent(this.endpoint);
-            }
-            if (distinctType === "Unordered") {
-                this.endpoint = new UnorderedDistinctEndpointComponent(this.endpoint);
-            }
-            // If top then add that to the pipeline. TOP N is effectively OFFSET 0 LIMIT N
-            const top = partitionedQueryExecutionInfo.queryInfo.top;
-            if (typeof top === "number") {
-                this.endpoint = new OffsetLimitEndpointComponent(this.endpoint, 0, top);
-            }
-            // If offset+limit then add that to the pipeline
-            const limit = partitionedQueryExecutionInfo.queryInfo.limit;
-            const offset = partitionedQueryExecutionInfo.queryInfo.offset;
-            if (typeof limit === "number" && typeof offset === "number") {
-                this.endpoint = new OffsetLimitEndpointComponent(this.endpoint, offset, limit);
-            }
+        if (!this.options.maxItemCount) {
+            this.options.maxItemCount = QueryExecution.DEFAULT_PAGE_SIZE;
+        }
+        const pageSize = this.options.maxItemCount;
+        // Extract query information and characteristics
+        const analyzedQueryInfo = this.analyzeQueryInfo(partitionedQueryExecutionInfo.queryInfo);
+        const { sortOrders, nonStreamingOrderBy, isOrderByQuery, isGroupByQuery, isUnorderedDistinctQuery, querySupportsTokens, } = analyzedQueryInfo;
+        // Reject continuation token usage for unsupported query types
+        if (!querySupportsTokens) {
+            rejectContinuationTokenForUnsupportedQueries(this.options.continuationToken, [
+                QueryTypes.nonStreamingOrderBy(nonStreamingOrderBy),
+                QueryTypes.groupBy(isGroupByQuery),
+                QueryTypes.unorderedDistinct(isUnorderedDistinctQuery),
+            ]);
         }
+        // Parse continuation token fields once for reuse in pipeline construction
+        const queryContinuationFields = this.options.continuationToken
+            ? parseContinuationTokenFields(this.options.continuationToken)
+            : undefined;
+        // Pick between non-streaming vs streaming execution context
+        this.endpoint = nonStreamingOrderBy
+            ? this.createNonStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, options)
+            : this.createStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, isGroupByQuery, queryContinuationFields);
         this.fetchBuffer = [];
+        // Initialize the appropriate fetch implementation based on enableQueryControl
+        if (this.options.enableQueryControl) {
+            this.fetchImplementation = new QueryControlFetchImplementation(this.endpoint, pageSize, this.collectionLink, this.options.continuationToken, isOrderByQuery, querySupportsTokens);
+        }
+        else {
+            this.fetchImplementation = new LegacyFetchImplementation(this.endpoint, pageSize);
+        }
     }
     hasMoreResults() {
         return this.fetchBuffer.length !== 0 || this.endpoint.hasMoreResults();
     }
     async fetchMore(diagnosticNode) {
-        this.fetchMoreRespHeaders = getInitialHeader();
-        return this._fetchMoreImplementation(diagnosticNode);
+        return this.fetchImplementation.fetchMore(diagnosticNode, this.fetchBuffer);
     }
-    async _fetchMoreImplementation(diagnosticNode) {
-        try {
-            if (this.fetchBuffer.length >= this.pageSize) {
-                const temp = this.fetchBuffer.slice(0, this.pageSize);
-                this.fetchBuffer = this.fetchBuffer.slice(this.pageSize);
-                return { result: temp, headers: this.fetchMoreRespHeaders };
-            }
-            else {
-                const response = await this.endpoint.fetchMore(diagnosticNode);
-                mergeHeaders(this.fetchMoreRespHeaders, response.headers);
-                if (response === undefined || response.result === undefined) {
-                    if (this.fetchBuffer.length > 0) {
-                        const temp = this.fetchBuffer;
-                        this.fetchBuffer = [];
-                        return { result: temp, headers: this.fetchMoreRespHeaders };
-                    }
-                    else {
-                        return { result: undefined, headers: this.fetchMoreRespHeaders };
-                    }
-                }
-                this.fetchBuffer.push(...response.result);
-                if (this.options.enableQueryControl) {
-                    if (this.fetchBuffer.length >= this.pageSize) {
-                        const temp = this.fetchBuffer.slice(0, this.pageSize);
-                        this.fetchBuffer = this.fetchBuffer.slice(this.pageSize);
-                        return { result: temp, headers: this.fetchMoreRespHeaders };
-                    }
-                    else {
-                        const temp = this.fetchBuffer;
-                        this.fetchBuffer = [];
-                        return { result: temp, headers: this.fetchMoreRespHeaders };
-                    }
-                }
-                // Recursively fetch more results to ensure the pageSize number of results are returned
-                // to maintain compatibility with the previous implementation
-                return this._fetchMoreImplementation(diagnosticNode);
+    /**
+     * Creates a non-streaming endpoint for vector search and similar queries that require buffering
+     */
+    createNonStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, options) {
+        const queryInfo = partitionedQueryExecutionInfo.queryInfo; // Safe to use ! after validation in constructor
+        if (!options.allowUnboundedNonStreamingQueries) {
+            this.checkQueryConstraints(queryInfo);
+        }
+        const vectorSearchBufferSize = this.calculateVectorSearchBufferSize(queryInfo, options);
+        this.validateVectorSearchBufferSize(vectorSearchBufferSize, options);
+        const baseContext = new ParallelQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, this.partitionedQueryExecutionInfo, correlatedActivityId);
+        return this.wrapWithNonStreamingComponent(baseContext, queryInfo, sortOrders, vectorSearchBufferSize);
+    }
+    /**
+     * Creates a streaming endpoint with proper pipeline components
+     */
+    createStreamingEndpoint(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId, isGroupByQuery, queryContinuationFields) {
+        const queryInfo = partitionedQueryExecutionInfo.queryInfo; // Safe to use ! after validation in constructor
+        // Create base execution context
+        let endpoint = this.createBaseExecutionContext(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId);
+        // Apply pipeline transformations
+        endpoint = this.applyGroupByComponents(endpoint, queryInfo, isGroupByQuery);
+        endpoint = this.applyDistinctComponents(endpoint, queryInfo, queryContinuationFields);
+        endpoint = this.applyLimitComponents(endpoint, queryInfo, queryContinuationFields);
+        return endpoint;
+    }
+    /**
+     * Creates the base execution context (OrderBy or Parallel)
+     */
+    createBaseExecutionContext(partitionedQueryExecutionInfo, sortOrders, correlatedActivityId) {
+        if (Array.isArray(sortOrders) && sortOrders.length > 0) {
+            // OrderBy queries need special wrapping for payload structure
+            return new OrderByEndpointComponent(new OrderByQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, partitionedQueryExecutionInfo, correlatedActivityId), this.emitRawOrderByPayload);
+        }
+        // Parallel queries
+        return new ParallelQueryExecutionContext(this.clientContext, this.collectionLink, this.query, this.options, partitionedQueryExecutionInfo, correlatedActivityId);
+    }
+    /**
+     * Wraps base context with appropriate non-streaming component
+     */
+    wrapWithNonStreamingComponent(baseContext, queryInfo, sortOrders, vectorSearchBufferSize) {
+        const distinctType = queryInfo.distinctType;
+        if (distinctType === "None") {
+            return new NonStreamingOrderByEndpointComponent(baseContext, sortOrders, vectorSearchBufferSize, queryInfo.offset, this.emitRawOrderByPayload);
+        }
+        return new NonStreamingOrderByDistinctEndpointComponent(baseContext, queryInfo, vectorSearchBufferSize, this.emitRawOrderByPayload);
+    }
+    /**
+     * Applies GROUP BY components to the pipeline if needed
+     */
+    applyGroupByComponents(endpoint, queryInfo, isGroupByQuery) {
+        if (!isGroupByQuery) {
+            return endpoint;
+        }
+        return queryInfo.hasSelectValue
+            ? new GroupByValueEndpointComponent(endpoint, queryInfo)
+            : new GroupByEndpointComponent(endpoint, queryInfo);
+    }
+    /**
+     * Applies DISTINCT components to the pipeline if needed
+     */
+    applyDistinctComponents(endpoint, queryInfo, queryContinuationFields) {
+        const distinctType = queryInfo.distinctType;
+        if (distinctType === "Ordered") {
+            const lastHash = queryContinuationFields?.hashedLastResult;
+            return new OrderedDistinctEndpointComponent(endpoint, lastHash);
+        }
+        if (distinctType === "Unordered") {
+            return new UnorderedDistinctEndpointComponent(endpoint);
+        }
+        return endpoint;
+    }
+    /**
+     * Applies TOP and OFFSET+LIMIT components to the pipeline if needed
+     */
+    applyLimitComponents(endpoint, queryInfo, queryContinuationFields) {
+        // Apply TOP component (TOP N is effectively OFFSET 0 LIMIT N)
+        let top = queryInfo.top;
+        if (typeof top === "number") {
+            if (queryContinuationFields?.limit !== undefined) {
+                top = queryContinuationFields.limit;
             }
+            endpoint = new OffsetLimitEndpointComponent(endpoint, 0, top);
         }
-        catch (err) {
-            mergeHeaders(this.fetchMoreRespHeaders, err.headers);
-            err.headers = this.fetchMoreRespHeaders;
-            if (err) {
-                throw err;
+        // Apply OFFSET+LIMIT component
+        let limit = queryInfo.limit;
+        let offset = queryInfo.offset;
+        if (queryContinuationFields) {
+            if (queryContinuationFields.limit !== undefined) {
+                limit = queryContinuationFields.limit;
+            }
+            if (queryContinuationFields.offset !== undefined) {
+                offset = queryContinuationFields.offset;
             }
         }
+        if (typeof limit === "number" && typeof offset === "number") {
+            endpoint = new OffsetLimitEndpointComponent(endpoint, offset, limit);
+        }
+        return endpoint;
+    }
+    /**
+     * Validates vector search buffer size constraints
+     */
+    validateVectorSearchBufferSize(vectorSearchBufferSize, options) {
+        const maxBufferSize = options["vectorSearchBufferSize"]
+            ? options["vectorSearchBufferSize"]
+            : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
+        if (vectorSearchBufferSize > maxBufferSize) {
+            throw new ErrorResponse(`Executing a vector search query with TOP or OFFSET + LIMIT value ${vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} ` +
+                `is not allowed`);
+        }
     }
     calculateVectorSearchBufferSize(queryInfo, options) {
         if (queryInfo.top === 0 || queryInfo.limit === 0)
@@ -168,7 +197,7 @@ export class PipelinedQueryExecutionContext {
                 ? queryInfo.offset + queryInfo.limit
                 : options["vectorSearchBufferSize"] && options["vectorSearchBufferSize"] > 0
                     ? options["vectorSearchBufferSize"]
-                    : PipelinedQueryExecutionContext.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
+                    : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;
     }
     checkQueryConstraints(queryInfo) {
         const hasTop = queryInfo.top || queryInfo.top === 0;
@@ -180,5 +209,29 @@ export class PipelinedQueryExecutionContext {
         }
         return;
     }
+    /**
+     * Analyzes query information and extracts key characteristics for query execution planning
+     */
+    analyzeQueryInfo(queryInfo) {
+        const sortOrders = queryInfo.orderBy;
+        const nonStreamingOrderBy = queryInfo.hasNonStreamingOrderBy;
+        const isOrderByQuery = Array.isArray(sortOrders) && sortOrders.length > 0;
+        // Check if this is a GROUP BY query
+        const isGroupByQuery = Object.keys(queryInfo.groupByAliasToAggregateType || {}).length > 0 ||
+            (queryInfo.aggregates?.length || 0) > 0 ||
+            (queryInfo.groupByExpressions?.length || 0) > 0;
+        // Check if this is an unordered DISTINCT query
+        const isUnorderedDistinctQuery = queryInfo.distinctType === "Unordered";
+        // Determine if this query type supports continuation tokens
+        const querySupportsTokens = !isUnorderedDistinctQuery && !isGroupByQuery && !nonStreamingOrderBy;
+        return {
+            sortOrders,
+            nonStreamingOrderBy,
+            isOrderByQuery,
+            isGroupByQuery,
+            isUnorderedDistinctQuery,
+            querySupportsTokens,
+        };
+    }
 }
 //# sourceMappingURL=pipelinedQueryExecutionContext.js.map

package/dist/browser/queryExecutionContext/pipelinedQueryExecutionContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"pipelinedQueryExecutionContext.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/pipelinedQueryExecutionContext.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAE5D,OAAO,EAAE,4BAA4B,EAAE,MAAM,qDAAqD,CAAC;AACnG,OAAO,EAAE,wBAAwB,EAAE,MAAM,iDAAiD,CAAC;AAC3F,OAAO,EAAE,gCAAgC,EAAE,MAAM,yDAAyD,CAAC;AAC3G,OAAO,EAAE,kCAAkC,EAAE,MAAM,2DAA2D,CAAC;AAC/G,OAAO,EAAE,wBAAwB,EAAE,MAAM,iDAAiD,CAAC;AAE3F,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAClE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AACnF,OAAO,EAAE,6BAA6B,EAAE,MAAM,sDAAsD,CAAC;AAGrG,OAAO,EAAE,4CAA4C,EAAE,MAAM,qEAAqE,CAAC;AACnI,OAAO,EAAE,oCAAoC,EAAE,MAAM,6DAA6D,CAAC;AAEnH,cAAc;AACd,MAAM,OAAO,8BAA8B;IAW/B;IACA;IACA;IACA;IACA;IAEA;IAhBF,WAAW,CAAQ;IACnB,oBAAoB,CAAgB;IACpC,QAAQ,CAAmB;IAC3B,QAAQ,CAAS;IACjB,sBAAsB,GAAW,CAAC,CAAC;IACnC,MAAM,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC9B,MAAM,CAAC,qCAAqC,GAAG,KAAK,CAAC;IACrD,mBAAmB,GAAG,KAAK,CAAC;IAEpC,YACU,aAA4B,EAC5B,cAAsB,EACtB,KAA4B,EAC5B,OAAoB,EACpB,6BAA4D,EACpE,oBAA4B,EACpB,wBAAiC,KAAK;QANtC,kBAAa,GAAb,aAAa,CAAe;QAC5B,mBAAc,GAAd,cAAc,CAAQ;QACtB,UAAK,GAAL,KAAK,CAAuB;QAC5B,YAAO,GAAP,OAAO,CAAa;QACpB,kCAA6B,GAA7B,6BAA6B,CAA+B;QAE5D,0BAAqB,GAArB,qBAAqB,CAAiB;QAE9C,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QACxC,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAChC,IAAI,CAAC,QAAQ,GAAG,8BAA8B,CAAC,iBAAiB,CAAC;QACnE,CAAC;QACD,oDAAoD;QACpD,IAAI,CAAC,mBAAmB,GAAG,6BAA6B,CAAC,SAAS,CAAC,sBAAsB,CAAC;QAE1F,sDAAsD;QACtD,MAAM,UAAU,GAAG,6BAA6B,CAAC,SAAS,CAAC,OAAO,CAAC;QACnE,qFAAqF;QACrF,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,CAAC,iCAAiC,EAAE,CAAC;gBAC/C,IAAI,CAAC,qBAAqB,CAAC,6BAA6B,CAAC,SAAS,CAAC,CAAC;YACtE,CAAC;YAED,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,+BAA+B,CAChE,6BAA6B,CAAC,SAAS,EACvC,OAAO,CACR,CAAC;YACF,MAAM,aAAa,GAAG,OAAO,CAAC,wBAAwB,CAAC;gBACrD,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC;gBACnC,CAAC,CAAC,8BAA8B,CAAC,qCAAqC,CAAC;YAEzE,IAAI,IAAI,CAAC,sBAAsB,GAAG,aAAa,EAAE,CAAC;gBAChD,MAAM,IAAI,aAAa,CACrB,oEAAoE,IAAI,CAAC,sBAAsB,2CAA2C,aAAa,GAAG;oBACxJ,gBAAgB,CACnB,CAAC;YACJ,CAAC;YAED,MAAM,YAAY,GAAG,6BAA6B,CAAC,SAAS,CAAC,YAAY,CAAC;YAC1E,MAAM,OAAO,GAAqB,IAAI,6BAA6B,CACjE,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,6BAA6B,EAClC,oBAAoB,CACrB,CAAC;YAEF,IAAI,YAAY,KAAK,MAAM,EAAE,CAAC;gBAC5B,IAAI,CAAC,QAAQ,GAAG,IAAI,oCAAoC,CACtD,OAAO,EACP,UAAU,EACV,IAAI,CAAC,sBAAsB,EAC3B,6BAA6B,CAAC,SAAS,CAAC,MAAM,EAC9C,IAAI,CAAC,qBAAqB,CAC3B,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,QAAQ,GAAG,IAAI,4CAA4C,CAC9D,OAAO,EACP,6BAA6B,CAAC,SAAS,EACvC,IAAI,CAAC,sBAAsB,EAC3B,IAAI,CAAC,qBAAqB,CAC3B,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvD,gGAAgG;gBAChG,2BAA2B;gBAC3B,IAAI,CAAC,QAAQ,GAAG,IAAI,wBAAwB,CAC1C,IAAI,4BAA4B,CAC9B,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,6BAA6B,EAClC,oBAAoB,CACrB,EACD,IAAI,CAAC,qBAAqB,CAC3B,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,QAAQ,GAAG,IAAI,6BAA6B,CAC/C,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,6BAA6B,EAClC,oBAAoB,CACrB,CAAC;YACJ,CAAC;YACD,IACE,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,SAAS,CAAC,2BAA2B,CAAC,CAAC,MAAM;gBACrF,CAAC;gBACH,6BAA6B,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;gBAC7D,6BAA6B,CAAC,SAAS,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,EACrE,CAAC;gBACD,IAAI,6BAA6B,CAAC,SAAS,CAAC,cAAc,EAAE,CAAC;oBAC3D,IAAI,CAAC,QAAQ,GAAG,IAAI,6BAA6B,CAC/C,IAAI,CAAC,QAAQ,EACb,6BAA6B,CAAC,SAAS,CACxC,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,QAAQ,GAAG,IAAI,wBAAwB,CAC1C,IAAI,CAAC,QAAQ,EACb,6BAA6B,CAAC,SAAS,CACxC,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,4CAA4C;YAC5C,MAAM,YAAY,GAAG,6BAA6B,CAAC,SAAS,CAAC,YAAY,CAAC;YAC1E,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC/B,IAAI,CAAC,QAAQ,GAAG,IAAI,gCAAgC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,YAAY,KAAK,WAAW,EAAE,CAAC;gBACjC,IAAI,CAAC,QAAQ,GAAG,IAAI,kCAAkC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACxE,CAAC;YAED,8EAA8E;YAC9E,MAAM,GAAG,GAAG,6BAA6B,CAAC,SAAS,CAAC,GAAG,CAAC;YACxD,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;gBAC5B,IAAI,CAAC,QAAQ,GAAG,IAAI,4BAA4B,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;YAC1E,CAAC;YAED,gDAAgD;YAChD,MAAM,KAAK,GAAG,6BAA6B,CAAC,SAAS,CAAC,KAAK,CAAC;YAC5D,MAAM,MAAM,GAAG,6BAA6B,CAAC,SAAS,CAAC,MAAM,CAAC;YAC9D,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC5D,IAAI,CAAC,QAAQ,GAAG,IAAI,4BAA4B,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;YACjF,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;IACxB,CAAC;IAEM,cAAc;QACnB,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;IACzE,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAsC;QAC3D,IAAI,CAAC,oBAAoB,GAAG,gBAAgB,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC,wBAAwB,CAAC,cAAc,CAAC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,wBAAwB,CACpC,cAAsC;QAEtC,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACtD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACzD,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9D,CAAC;iBAAM,CAAC;gBACN,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;gBAC/D,YAAY,CAAC,IAAI,CAAC,oBAAoB,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAC1D,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC5D,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAChC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC;wBAC9B,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;wBACtB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC9D,CAAC;yBAAM,CAAC;wBACN,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBACnE,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAE1C,IAAI,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;oBACpC,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;wBAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;wBACtD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBACzD,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC9D,CAAC;yBAAM,CAAC;wBACN,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC;wBAC9B,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;wBACtB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,oBAAoB,EAAE,CAAC;oBAC9D,CAAC;gBACH,CAAC;gBACD,uFAAuF;gBACvF,6DAA6D;gBAC7D,OAAO,IAAI,CAAC,wBAAwB,CAAC,cAAc,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,YAAY,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YACrD,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC;YACxC,IAAI,GAAG,EAAE,CAAC;gBACR,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,+BAA+B,CAAC,SAAoB,EAAE,OAAoB;QAChF,IAAI,SAAS,CAAC,GAAG,KAAK,CAAC,IAAI,SAAS,CAAC,KAAK,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAC3D,OAAO,SAAS,CAAC,GAAG;YAClB,CAAC,CAAC,SAAS,CAAC,GAAG;YACf,CAAC,CAAC,SAAS,CAAC,KAAK;gBACf,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,SAAS,CAAC,KAAK;gBACpC,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC,IAAI,OAAO,CAAC,wBAAwB,CAAC,GAAG,CAAC;oBAC1E,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC;oBACnC,CAAC,CAAC,8BAA8B,CAAC,qCAAqC,CAAC;IAC/E,CAAC;IAEO,qBAAqB,CAAC,SAAoB;QAChD,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,IAAI,SAAS,CAAC,GAAG,KAAK,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,IAAI,SAAS,CAAC,KAAK,KAAK,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,aAAa,CACrB,gGAAgG;gBAC9F,6FAA6F;gBAC7F,gCAAgC,CACnC,CAAC;QACJ,CAAC;QACD,OAAO;IACT,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { ClientContext } from \"../ClientContext.js\";\nimport type { Response, FeedOptions } from \"../request/index.js\";\nimport type { PartitionedQueryExecutionInfo, QueryInfo } from \"../request/ErrorResponse.js\";\nimport { ErrorResponse } from \"../request/ErrorResponse.js\";\nimport type { CosmosHeaders } from \"./CosmosHeaders.js\";\nimport { OffsetLimitEndpointComponent } from \"./EndpointComponent/OffsetLimitEndpointComponent.js\";\nimport { OrderByEndpointComponent } from \"./EndpointComponent/OrderByEndpointComponent.js\";\nimport { OrderedDistinctEndpointComponent } from \"./EndpointComponent/OrderedDistinctEndpointComponent.js\";\nimport { UnorderedDistinctEndpointComponent } from \"./EndpointComponent/UnorderedDistinctEndpointComponent.js\";\nimport { GroupByEndpointComponent } from \"./EndpointComponent/GroupByEndpointComponent.js\";\nimport type { ExecutionContext } from \"./ExecutionContext.js\";\nimport { getInitialHeader, mergeHeaders } from \"./headerUtils.js\";\nimport { OrderByQueryExecutionContext } from \"./orderByQueryExecutionContext.js\";\nimport { ParallelQueryExecutionContext } from \"./parallelQueryExecutionContext.js\";\nimport { GroupByValueEndpointComponent } from \"./EndpointComponent/GroupByValueEndpointComponent.js\";\nimport type { SqlQuerySpec } from \"./SqlQuerySpec.js\";\nimport type { DiagnosticNodeInternal } from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport { NonStreamingOrderByDistinctEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js\";\nimport { NonStreamingOrderByEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByEndpointComponent.js\";\n\n/** @hidden */\nexport class PipelinedQueryExecutionContext implements ExecutionContext {\n  private fetchBuffer: any[];\n  private fetchMoreRespHeaders: CosmosHeaders;\n  private endpoint: ExecutionContext;\n  private pageSize: number;\n  private vectorSearchBufferSize: number = 0;\n  private static DEFAULT_PAGE_SIZE = 10;\n  private static DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE = 50000;\n  private nonStreamingOrderBy = false;\n\n  constructor(\n    private clientContext: ClientContext,\n    private collectionLink: string,\n    private query: string | SqlQuerySpec,\n    private options: FeedOptions,\n    private partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    correlatedActivityId: string,\n    private emitRawOrderByPayload: boolean = false,\n  ) {\n    this.endpoint = null;\n    this.pageSize = options[\"maxItemCount\"];\n    if (this.pageSize === undefined) {\n      this.pageSize = PipelinedQueryExecutionContext.DEFAULT_PAGE_SIZE;\n    }\n    // Pick between Nonstreaming and streaming endpoints\n    this.nonStreamingOrderBy = partitionedQueryExecutionInfo.queryInfo.hasNonStreamingOrderBy;\n\n    // Pick between parallel vs order by execution context\n    const sortOrders = partitionedQueryExecutionInfo.queryInfo.orderBy;\n    // TODO: Currently we don't get any field from backend to determine streaming queries\n    if (this.nonStreamingOrderBy) {\n      if (!options.allowUnboundedNonStreamingQueries) {\n        this.checkQueryConstraints(partitionedQueryExecutionInfo.queryInfo);\n      }\n\n      this.vectorSearchBufferSize = this.calculateVectorSearchBufferSize(\n        partitionedQueryExecutionInfo.queryInfo,\n        options,\n      );\n      const maxBufferSize = options[\"vectorSearchBufferSize\"]\n        ? options[\"vectorSearchBufferSize\"]\n        : PipelinedQueryExecutionContext.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n\n      if (this.vectorSearchBufferSize > maxBufferSize) {\n        throw new ErrorResponse(\n          `Executing a vector search query with TOP or OFFSET + LIMIT value ${this.vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} ` +\n            `is not allowed`,\n        );\n      }\n\n      const distinctType = partitionedQueryExecutionInfo.queryInfo.distinctType;\n      const context: ExecutionContext = new ParallelQueryExecutionContext(\n        this.clientContext,\n        this.collectionLink,\n        this.query,\n        this.options,\n        this.partitionedQueryExecutionInfo,\n        correlatedActivityId,\n      );\n\n      if (distinctType === \"None\") {\n        this.endpoint = new NonStreamingOrderByEndpointComponent(\n          context,\n          sortOrders,\n          this.vectorSearchBufferSize,\n          partitionedQueryExecutionInfo.queryInfo.offset,\n          this.emitRawOrderByPayload,\n        );\n      } else {\n        this.endpoint = new NonStreamingOrderByDistinctEndpointComponent(\n          context,\n          partitionedQueryExecutionInfo.queryInfo,\n          this.vectorSearchBufferSize,\n          this.emitRawOrderByPayload,\n        );\n      }\n    } else {\n      if (Array.isArray(sortOrders) && sortOrders.length > 0) {\n        // Need to wrap orderby execution context in endpoint component, since the data is nested as a \\\n        //      \"payload\" property.\n        this.endpoint = new OrderByEndpointComponent(\n          new OrderByQueryExecutionContext(\n            this.clientContext,\n            this.collectionLink,\n            this.query,\n            this.options,\n            this.partitionedQueryExecutionInfo,\n            correlatedActivityId,\n          ),\n          this.emitRawOrderByPayload,\n        );\n      } else {\n        this.endpoint = new ParallelQueryExecutionContext(\n          this.clientContext,\n          this.collectionLink,\n          this.query,\n          this.options,\n          this.partitionedQueryExecutionInfo,\n          correlatedActivityId,\n        );\n      }\n      if (\n        Object.keys(partitionedQueryExecutionInfo.queryInfo.groupByAliasToAggregateType).length >\n          0 ||\n        partitionedQueryExecutionInfo.queryInfo.aggregates.length > 0 ||\n        partitionedQueryExecutionInfo.queryInfo.groupByExpressions.length > 0\n      ) {\n        if (partitionedQueryExecutionInfo.queryInfo.hasSelectValue) {\n          this.endpoint = new GroupByValueEndpointComponent(\n            this.endpoint,\n            partitionedQueryExecutionInfo.queryInfo,\n          );\n        } else {\n          this.endpoint = new GroupByEndpointComponent(\n            this.endpoint,\n            partitionedQueryExecutionInfo.queryInfo,\n          );\n        }\n      }\n\n      // If distinct then add that to the pipeline\n      const distinctType = partitionedQueryExecutionInfo.queryInfo.distinctType;\n      if (distinctType === \"Ordered\") {\n        this.endpoint = new OrderedDistinctEndpointComponent(this.endpoint);\n      }\n      if (distinctType === \"Unordered\") {\n        this.endpoint = new UnorderedDistinctEndpointComponent(this.endpoint);\n      }\n\n      // If top then add that to the pipeline. TOP N is effectively OFFSET 0 LIMIT N\n      const top = partitionedQueryExecutionInfo.queryInfo.top;\n      if (typeof top === \"number\") {\n        this.endpoint = new OffsetLimitEndpointComponent(this.endpoint, 0, top);\n      }\n\n      // If offset+limit then add that to the pipeline\n      const limit = partitionedQueryExecutionInfo.queryInfo.limit;\n      const offset = partitionedQueryExecutionInfo.queryInfo.offset;\n      if (typeof limit === \"number\" && typeof offset === \"number\") {\n        this.endpoint = new OffsetLimitEndpointComponent(this.endpoint, offset, limit);\n      }\n    }\n    this.fetchBuffer = [];\n  }\n\n  public hasMoreResults(): boolean {\n    return this.fetchBuffer.length !== 0 || this.endpoint.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    this.fetchMoreRespHeaders = getInitialHeader();\n    return this._fetchMoreImplementation(diagnosticNode);\n  }\n\n  private async _fetchMoreImplementation(\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<Response<any>> {\n    try {\n      if (this.fetchBuffer.length >= this.pageSize) {\n        const temp = this.fetchBuffer.slice(0, this.pageSize);\n        this.fetchBuffer = this.fetchBuffer.slice(this.pageSize);\n        return { result: temp, headers: this.fetchMoreRespHeaders };\n      } else {\n        const response = await this.endpoint.fetchMore(diagnosticNode);\n        mergeHeaders(this.fetchMoreRespHeaders, response.headers);\n        if (response === undefined || response.result === undefined) {\n          if (this.fetchBuffer.length > 0) {\n            const temp = this.fetchBuffer;\n            this.fetchBuffer = [];\n            return { result: temp, headers: this.fetchMoreRespHeaders };\n          } else {\n            return { result: undefined, headers: this.fetchMoreRespHeaders };\n          }\n        }\n        this.fetchBuffer.push(...response.result);\n\n        if (this.options.enableQueryControl) {\n          if (this.fetchBuffer.length >= this.pageSize) {\n            const temp = this.fetchBuffer.slice(0, this.pageSize);\n            this.fetchBuffer = this.fetchBuffer.slice(this.pageSize);\n            return { result: temp, headers: this.fetchMoreRespHeaders };\n          } else {\n            const temp = this.fetchBuffer;\n            this.fetchBuffer = [];\n            return { result: temp, headers: this.fetchMoreRespHeaders };\n          }\n        }\n        // Recursively fetch more results to ensure the pageSize number of results are returned\n        // to maintain compatibility with the previous implementation\n        return this._fetchMoreImplementation(diagnosticNode);\n      }\n    } catch (err: any) {\n      mergeHeaders(this.fetchMoreRespHeaders, err.headers);\n      err.headers = this.fetchMoreRespHeaders;\n      if (err) {\n        throw err;\n      }\n    }\n  }\n\n  private calculateVectorSearchBufferSize(queryInfo: QueryInfo, options: FeedOptions): number {\n    if (queryInfo.top === 0 || queryInfo.limit === 0) return 0;\n    return queryInfo.top\n      ? queryInfo.top\n      : queryInfo.limit\n        ? queryInfo.offset + queryInfo.limit\n        : options[\"vectorSearchBufferSize\"] && options[\"vectorSearchBufferSize\"] > 0\n          ? options[\"vectorSearchBufferSize\"]\n          : PipelinedQueryExecutionContext.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n  }\n\n  private checkQueryConstraints(queryInfo: QueryInfo): void {\n    const hasTop = queryInfo.top || queryInfo.top === 0;\n    const hasLimit = queryInfo.limit || queryInfo.limit === 0;\n    if (!hasTop && !hasLimit) {\n      throw new ErrorResponse(\n        \"Executing a non-streaming search query without TOP or LIMIT can consume a large number of RUs \" +\n          \"very fast and have long runtimes. Please ensure you are using one of the above two filters \" +\n          \"with your vector search query.\",\n      );\n    }\n    return;\n  }\n}\n"]}
+{"version":3,"file":"pipelinedQueryExecutionContext.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/pipelinedQueryExecutionContext.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EAAE,4BAA4B,EAAE,MAAM,qDAAqD,CAAC;AACnG,OAAO,EAAE,wBAAwB,EAAE,MAAM,iDAAiD,CAAC;AAC3F,OAAO,EAAE,gCAAgC,EAAE,MAAM,yDAAyD,CAAC;AAC3G,OAAO,EAAE,kCAAkC,EAAE,MAAM,2DAA2D,CAAC;AAC/G,OAAO,EAAE,wBAAwB,EAAE,MAAM,iDAAiD,CAAC;AAE3F,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AACnF,OAAO,EAAE,6BAA6B,EAAE,MAAM,sDAAsD,CAAC;AAGrG,OAAO,EAAE,4CAA4C,EAAE,MAAM,qEAAqE,CAAC;AACnI,OAAO,EAAE,oCAAoC,EAAE,MAAM,6DAA6D,CAAC;AACnH,OAAO,EACL,4CAA4C,EAC5C,UAAU,GACX,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,4BAA4B,EAAE,MAAM,8BAA8B,CAAC;AAC5E,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAAE,+BAA+B,EAAE,MAAM,sCAAsC,CAAC;AACvF,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAExD,cAAc;AACd,MAAM,OAAO,8BAA8B;IAM/B;IACA;IACA;IACA;IACA;IAEA;IAXF,WAAW,CAAQ;IACnB,QAAQ,CAAmB;IAClB,mBAAmB,CAA8D;IAElG,YACU,aAA4B,EAC5B,cAAsB,EACtB,KAA4B,EAC5B,OAAoB,EACpB,6BAA4D,EACpE,oBAA4B,EACpB,wBAAiC,KAAK;QANtC,kBAAa,GAAb,aAAa,CAAe;QAC5B,mBAAc,GAAd,cAAc,CAAQ;QACtB,UAAK,GAAL,KAAK,CAAuB;QAC5B,YAAO,GAAP,OAAO,CAAa;QACpB,kCAA6B,GAA7B,6BAA6B,CAA+B;QAE5D,0BAAqB,GAArB,qBAAqB,CAAiB;QAE9C,yEAAyE;QACzE,IAAI,CAAC,6BAA6B,CAAC,SAAS,EAAE,CAAC;YAC7C,MAAM,IAAI,aAAa,CACrB,yDAAyD;gBACvD,6DAA6D;gBAC7D,sEAAsE,CACzE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,YAAY,GAAG,cAAc,CAAC,iBAAiB,CAAC;QAC/D,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;QAE3C,gDAAgD;QAChD,MAAM,iBAAiB,GAAG,IAAI,CAAC,gBAAgB,CAAC,6BAA6B,CAAC,SAAS,CAAC,CAAC;QACzF,MAAM,EACJ,UAAU,EACV,mBAAmB,EACnB,cAAc,EACd,cAAc,EACd,wBAAwB,EACxB,mBAAmB,GACpB,GAAG,iBAAiB,CAAC;QAEtB,8DAA8D;QAC9D,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACzB,4CAA4C,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE;gBAC3E,UAAU,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;gBACnD,UAAU,CAAC,OAAO,CAAC,cAAc,CAAC;gBAClC,UAAU,CAAC,iBAAiB,CAAC,wBAAwB,CAAC;aACvD,CAAC,CAAC;QACL,CAAC;QAED,0EAA0E;QAC1E,MAAM,uBAAuB,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB;YAC5D,CAAC,CAAC,4BAA4B,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC;YAC9D,CAAC,CAAC,SAAS,CAAC;QAEd,4DAA4D;QAC5D,IAAI,CAAC,QAAQ,GAAG,mBAAmB;YACjC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAC7B,6BAA6B,EAC7B,UAAU,EACV,oBAAoB,EACpB,OAAO,CACR;YACH,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAC1B,6BAA6B,EAC7B,UAAU,EACV,oBAAoB,EACpB,cAAc,EACd,uBAAuB,CACxB,CAAC;QACN,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,8EAA8E;QAC9E,IAAI,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;YACpC,IAAI,CAAC,mBAAmB,GAAG,IAAI,+BAA+B,CAC5D,IAAI,CAAC,QAAQ,EACb,QAAQ,EACR,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAC9B,cAAc,EACd,mBAAmB,CACpB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,mBAAmB,GAAG,IAAI,yBAAyB,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACpF,CAAC;IACH,CAAC;IAEM,cAAc;QACnB,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;IACzE,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAsC;QAC3D,OAAO,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC9E,CAAC;IAED;;OAEG;IACK,0BAA0B,CAChC,6BAA4D,EAC5D,UAAiB,EACjB,oBAA4B,EAC5B,OAAoB;QAEpB,MAAM,SAAS,GAAG,6BAA6B,CAAC,SAAU,CAAC,CAAC,gDAAgD;QAE5G,IAAI,CAAC,OAAO,CAAC,iCAAiC,EAAE,CAAC;YAC/C,IAAI,CAAC,qBAAqB,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,sBAAsB,GAAG,IAAI,CAAC,+BAA+B,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAExF,IAAI,CAAC,8BAA8B,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;QAErE,MAAM,WAAW,GAAG,IAAI,6BAA6B,CACnD,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,6BAA6B,EAClC,oBAAoB,CACrB,CAAC;QAEF,OAAO,IAAI,CAAC,6BAA6B,CACvC,WAAW,EACX,SAAS,EACT,UAAU,EACV,sBAAsB,CACvB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,uBAAuB,CAC7B,6BAA4D,EAC5D,UAAiB,EACjB,oBAA4B,EAC5B,cAAuB,EACvB,uBAA4B;QAE5B,MAAM,SAAS,GAAG,6BAA6B,CAAC,SAAU,CAAC,CAAC,gDAAgD;QAE5G,gCAAgC;QAChC,IAAI,QAAQ,GAAG,IAAI,CAAC,0BAA0B,CAC5C,6BAA6B,EAC7B,UAAU,EACV,oBAAoB,CACrB,CAAC;QAEF,iCAAiC;QACjC,QAAQ,GAAG,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;QAC5E,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;QACtF,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,SAAS,EAAE,uBAAuB,CAAC,CAAC;QAEnF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,0BAA0B,CAChC,6BAA4D,EAC5D,UAAiB,EACjB,oBAA4B;QAE5B,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,8DAA8D;YAC9D,OAAO,IAAI,wBAAwB,CACjC,IAAI,4BAA4B,CAC9B,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,6BAA6B,EAC7B,oBAAoB,CACrB,EACD,IAAI,CAAC,qBAAqB,CAC3B,CAAC;QACJ,CAAC;QAED,mBAAmB;QACnB,OAAO,IAAI,6BAA6B,CACtC,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,OAAO,EACZ,6BAA6B,EAC7B,oBAAoB,CACrB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,6BAA6B,CACnC,WAA6B,EAC7B,SAAoB,EACpB,UAAiB,EACjB,sBAA8B;QAE9B,MAAM,YAAY,GAAG,SAAS,CAAC,YAAY,CAAC;QAE5C,IAAI,YAAY,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO,IAAI,oCAAoC,CAC7C,WAAW,EACX,UAAU,EACV,sBAAsB,EACtB,SAAS,CAAC,MAAM,EAChB,IAAI,CAAC,qBAAqB,CAC3B,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,4CAA4C,CACrD,WAAW,EACX,SAAS,EACT,sBAAsB,EACtB,IAAI,CAAC,qBAAqB,CAC3B,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,sBAAsB,CAC5B,QAA0B,EAC1B,SAAoB,EACpB,cAAuB;QAEvB,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,OAAO,SAAS,CAAC,cAAc;YAC7B,CAAC,CAAC,IAAI,6BAA6B,CAAC,QAAQ,EAAE,SAAS,CAAC;YACxD,CAAC,CAAC,IAAI,wBAAwB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IACxD,CAAC;IAED;;OAEG;IACK,uBAAuB,CAC7B,QAA0B,EAC1B,SAAoB,EACpB,uBAA4B;QAE5B,MAAM,YAAY,GAAG,SAAS,CAAC,YAAY,CAAC;QAE5C,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAG,uBAAuB,EAAE,gBAAgB,CAAC;YAC3D,OAAO,IAAI,gCAAgC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,YAAY,KAAK,WAAW,EAAE,CAAC;YACjC,OAAO,IAAI,kCAAkC,CAAC,QAAQ,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,oBAAoB,CAC1B,QAA0B,EAC1B,SAAoB,EACpB,uBAA4B;QAE5B,8DAA8D;QAC9D,IAAI,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC;QACxB,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;YAC5B,IAAI,uBAAuB,EAAE,KAAK,KAAK,SAAS,EAAE,CAAC;gBACjD,GAAG,GAAG,uBAAuB,CAAC,KAAK,CAAC;YACtC,CAAC;YACD,QAAQ,GAAG,IAAI,4BAA4B,CAAC,QAAQ,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;QAChE,CAAC;QAED,+BAA+B;QAC/B,IAAI,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC;QAC5B,IAAI,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;QAE9B,IAAI,uBAAuB,EAAE,CAAC;YAC5B,IAAI,uBAAuB,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBAChD,KAAK,GAAG,uBAAuB,CAAC,KAAK,CAAC;YACxC,CAAC;YACD,IAAI,uBAAuB,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACjD,MAAM,GAAG,uBAAuB,CAAC,MAAM,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC5D,QAAQ,GAAG,IAAI,4BAA4B,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QACvE,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,8BAA8B,CACpC,sBAA8B,EAC9B,OAAoB;QAEpB,MAAM,aAAa,GAAG,OAAO,CAAC,wBAAwB,CAAC;YACrD,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC;YACnC,CAAC,CAAC,cAAc,CAAC,qCAAqC,CAAC;QAEzD,IAAI,sBAAsB,GAAG,aAAa,EAAE,CAAC;YAC3C,MAAM,IAAI,aAAa,CACrB,oEAAoE,sBAAsB,2CAA2C,aAAa,GAAG;gBACnJ,gBAAgB,CACnB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,+BAA+B,CAAC,SAAoB,EAAE,OAAoB;QAChF,IAAI,SAAS,CAAC,GAAG,KAAK,CAAC,IAAI,SAAS,CAAC,KAAK,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAC3D,OAAO,SAAS,CAAC,GAAG;YAClB,CAAC,CAAC,SAAS,CAAC,GAAG;YACf,CAAC,CAAC,SAAS,CAAC,KAAK;gBACf,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,SAAS,CAAC,KAAK;gBACpC,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC,IAAI,OAAO,CAAC,wBAAwB,CAAC,GAAG,CAAC;oBAC1E,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC;oBACnC,CAAC,CAAC,cAAc,CAAC,qCAAqC,CAAC;IAC/D,CAAC;IAEO,qBAAqB,CAAC,SAAoB;QAChD,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,IAAI,SAAS,CAAC,GAAG,KAAK,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,IAAI,SAAS,CAAC,KAAK,KAAK,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACzB,MAAM,IAAI,aAAa,CACrB,gGAAgG;gBAC9F,6FAA6F;gBAC7F,gCAAgC,CACnC,CAAC;QACJ,CAAC;QACD,OAAO;IACT,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,SAAoB;QAC3C,MAAM,UAAU,GAAG,SAAS,CAAC,OAAO,CAAC;QACrC,MAAM,mBAAmB,GAAG,SAAS,CAAC,sBAAsB,CAAC;QAC7D,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;QAE1E,oCAAoC;QACpC,MAAM,cAAc,GAClB,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,2BAA2B,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;YACnE,CAAC,SAAS,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC;YACvC,CAAC,SAAS,CAAC,kBAAkB,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QAElD,+CAA+C;QAC/C,MAAM,wBAAwB,GAAG,SAAS,CAAC,YAAY,KAAK,WAAW,CAAC;QAExE,4DAA4D;QAC5D,MAAM,mBAAmB,GACvB,CAAC,wBAAwB,IAAI,CAAC,cAAc,IAAI,CAAC,mBAAmB,CAAC;QAEvE,OAAO;YACL,UAAU;YACV,mBAAmB;YACnB,cAAc;YACd,cAAc;YACd,wBAAwB;YACxB,mBAAmB;SACpB,CAAC;IACJ,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { ClientContext } from \"../ClientContext.js\";\nimport type { Response, FeedOptions } from \"../request/index.js\";\nimport type { PartitionedQueryExecutionInfo, QueryInfo } from \"../request/ErrorResponse.js\";\nimport { ErrorResponse } from \"../request/ErrorResponse.js\";\nimport { OffsetLimitEndpointComponent } from \"./EndpointComponent/OffsetLimitEndpointComponent.js\";\nimport { OrderByEndpointComponent } from \"./EndpointComponent/OrderByEndpointComponent.js\";\nimport { OrderedDistinctEndpointComponent } from \"./EndpointComponent/OrderedDistinctEndpointComponent.js\";\nimport { UnorderedDistinctEndpointComponent } from \"./EndpointComponent/UnorderedDistinctEndpointComponent.js\";\nimport { GroupByEndpointComponent } from \"./EndpointComponent/GroupByEndpointComponent.js\";\nimport type { ExecutionContext } from \"./ExecutionContext.js\";\nimport { OrderByQueryExecutionContext } from \"./orderByQueryExecutionContext.js\";\nimport { ParallelQueryExecutionContext } from \"./parallelQueryExecutionContext.js\";\nimport { GroupByValueEndpointComponent } from \"./EndpointComponent/GroupByValueEndpointComponent.js\";\nimport type { SqlQuerySpec } from \"./SqlQuerySpec.js\";\nimport type { DiagnosticNodeInternal } from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport { NonStreamingOrderByDistinctEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js\";\nimport { NonStreamingOrderByEndpointComponent } from \"./EndpointComponent/NonStreamingOrderByEndpointComponent.js\";\nimport {\n  rejectContinuationTokenForUnsupportedQueries,\n  QueryTypes,\n} from \"./QueryValidationHelper.js\";\nimport { parseContinuationTokenFields } from \"./ContinuationTokenParser.js\";\nimport { LegacyFetchImplementation } from \"./LegacyFetchImplementation.js\";\nimport { QueryControlFetchImplementation } from \"./QueryControlFetchImplementation.js\";\nimport { QueryExecution } from \"../common/constants.js\";\n\n/** @hidden */\nexport class PipelinedQueryExecutionContext implements ExecutionContext {\n  private fetchBuffer: any[];\n  private endpoint: ExecutionContext;\n  private readonly fetchImplementation: LegacyFetchImplementation | QueryControlFetchImplementation;\n\n  constructor(\n    private clientContext: ClientContext,\n    private collectionLink: string,\n    private query: string | SqlQuerySpec,\n    private options: FeedOptions,\n    private partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    correlatedActivityId: string,\n    private emitRawOrderByPayload: boolean = false,\n  ) {\n    // Validate that queryInfo is present in partitioned query execution info\n    if (!partitionedQueryExecutionInfo.queryInfo) {\n      throw new ErrorResponse(\n        \"Query execution requires valid query plan information. \" +\n          \"The partitioned query execution info is missing queryInfo. \" +\n          \"This may indicate an invalid query or a problem with query planning.\",\n      );\n    }\n\n    if (!this.options.maxItemCount) {\n      this.options.maxItemCount = QueryExecution.DEFAULT_PAGE_SIZE;\n    }\n    const pageSize = this.options.maxItemCount;\n\n    // Extract query information and characteristics\n    const analyzedQueryInfo = this.analyzeQueryInfo(partitionedQueryExecutionInfo.queryInfo);\n    const {\n      sortOrders,\n      nonStreamingOrderBy,\n      isOrderByQuery,\n      isGroupByQuery,\n      isUnorderedDistinctQuery,\n      querySupportsTokens,\n    } = analyzedQueryInfo;\n\n    // Reject continuation token usage for unsupported query types\n    if (!querySupportsTokens) {\n      rejectContinuationTokenForUnsupportedQueries(this.options.continuationToken, [\n        QueryTypes.nonStreamingOrderBy(nonStreamingOrderBy),\n        QueryTypes.groupBy(isGroupByQuery),\n        QueryTypes.unorderedDistinct(isUnorderedDistinctQuery),\n      ]);\n    }\n\n    // Parse continuation token fields once for reuse in pipeline construction\n    const queryContinuationFields = this.options.continuationToken\n      ? parseContinuationTokenFields(this.options.continuationToken)\n      : undefined;\n\n    // Pick between non-streaming vs streaming execution context\n    this.endpoint = nonStreamingOrderBy\n      ? this.createNonStreamingEndpoint(\n          partitionedQueryExecutionInfo,\n          sortOrders,\n          correlatedActivityId,\n          options,\n        )\n      : this.createStreamingEndpoint(\n          partitionedQueryExecutionInfo,\n          sortOrders,\n          correlatedActivityId,\n          isGroupByQuery,\n          queryContinuationFields,\n        );\n    this.fetchBuffer = [];\n    // Initialize the appropriate fetch implementation based on enableQueryControl\n    if (this.options.enableQueryControl) {\n      this.fetchImplementation = new QueryControlFetchImplementation(\n        this.endpoint,\n        pageSize,\n        this.collectionLink,\n        this.options.continuationToken,\n        isOrderByQuery,\n        querySupportsTokens,\n      );\n    } else {\n      this.fetchImplementation = new LegacyFetchImplementation(this.endpoint, pageSize);\n    }\n  }\n\n  public hasMoreResults(): boolean {\n    return this.fetchBuffer.length !== 0 || this.endpoint.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    return this.fetchImplementation.fetchMore(diagnosticNode, this.fetchBuffer);\n  }\n\n  /**\n   * Creates a non-streaming endpoint for vector search and similar queries that require buffering\n   */\n  private createNonStreamingEndpoint(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n    options: FeedOptions,\n  ): ExecutionContext {\n    const queryInfo = partitionedQueryExecutionInfo.queryInfo!; // Safe to use ! after validation in constructor\n\n    if (!options.allowUnboundedNonStreamingQueries) {\n      this.checkQueryConstraints(queryInfo);\n    }\n\n    const vectorSearchBufferSize = this.calculateVectorSearchBufferSize(queryInfo, options);\n\n    this.validateVectorSearchBufferSize(vectorSearchBufferSize, options);\n\n    const baseContext = new ParallelQueryExecutionContext(\n      this.clientContext,\n      this.collectionLink,\n      this.query,\n      this.options,\n      this.partitionedQueryExecutionInfo,\n      correlatedActivityId,\n    );\n\n    return this.wrapWithNonStreamingComponent(\n      baseContext,\n      queryInfo,\n      sortOrders,\n      vectorSearchBufferSize,\n    );\n  }\n\n  /**\n   * Creates a streaming endpoint with proper pipeline components\n   */\n  private createStreamingEndpoint(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n    isGroupByQuery: boolean,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    const queryInfo = partitionedQueryExecutionInfo.queryInfo!; // Safe to use ! after validation in constructor\n\n    // Create base execution context\n    let endpoint = this.createBaseExecutionContext(\n      partitionedQueryExecutionInfo,\n      sortOrders,\n      correlatedActivityId,\n    );\n\n    // Apply pipeline transformations\n    endpoint = this.applyGroupByComponents(endpoint, queryInfo, isGroupByQuery);\n    endpoint = this.applyDistinctComponents(endpoint, queryInfo, queryContinuationFields);\n    endpoint = this.applyLimitComponents(endpoint, queryInfo, queryContinuationFields);\n\n    return endpoint;\n  }\n\n  /**\n   * Creates the base execution context (OrderBy or Parallel)\n   */\n  private createBaseExecutionContext(\n    partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo,\n    sortOrders: any[],\n    correlatedActivityId: string,\n  ): ExecutionContext {\n    if (Array.isArray(sortOrders) && sortOrders.length > 0) {\n      // OrderBy queries need special wrapping for payload structure\n      return new OrderByEndpointComponent(\n        new OrderByQueryExecutionContext(\n          this.clientContext,\n          this.collectionLink,\n          this.query,\n          this.options,\n          partitionedQueryExecutionInfo,\n          correlatedActivityId,\n        ),\n        this.emitRawOrderByPayload,\n      );\n    }\n\n    // Parallel queries\n    return new ParallelQueryExecutionContext(\n      this.clientContext,\n      this.collectionLink,\n      this.query,\n      this.options,\n      partitionedQueryExecutionInfo,\n      correlatedActivityId,\n    );\n  }\n\n  /**\n   * Wraps base context with appropriate non-streaming component\n   */\n  private wrapWithNonStreamingComponent(\n    baseContext: ExecutionContext,\n    queryInfo: QueryInfo,\n    sortOrders: any[],\n    vectorSearchBufferSize: number,\n  ): ExecutionContext {\n    const distinctType = queryInfo.distinctType;\n\n    if (distinctType === \"None\") {\n      return new NonStreamingOrderByEndpointComponent(\n        baseContext,\n        sortOrders,\n        vectorSearchBufferSize,\n        queryInfo.offset,\n        this.emitRawOrderByPayload,\n      );\n    }\n\n    return new NonStreamingOrderByDistinctEndpointComponent(\n      baseContext,\n      queryInfo,\n      vectorSearchBufferSize,\n      this.emitRawOrderByPayload,\n    );\n  }\n\n  /**\n   * Applies GROUP BY components to the pipeline if needed\n   */\n  private applyGroupByComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    isGroupByQuery: boolean,\n  ): ExecutionContext {\n    if (!isGroupByQuery) {\n      return endpoint;\n    }\n\n    return queryInfo.hasSelectValue\n      ? new GroupByValueEndpointComponent(endpoint, queryInfo)\n      : new GroupByEndpointComponent(endpoint, queryInfo);\n  }\n\n  /**\n   * Applies DISTINCT components to the pipeline if needed\n   */\n  private applyDistinctComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    const distinctType = queryInfo.distinctType;\n\n    if (distinctType === \"Ordered\") {\n      const lastHash = queryContinuationFields?.hashedLastResult;\n      return new OrderedDistinctEndpointComponent(endpoint, lastHash);\n    }\n\n    if (distinctType === \"Unordered\") {\n      return new UnorderedDistinctEndpointComponent(endpoint);\n    }\n\n    return endpoint;\n  }\n\n  /**\n   * Applies TOP and OFFSET+LIMIT components to the pipeline if needed\n   */\n  private applyLimitComponents(\n    endpoint: ExecutionContext,\n    queryInfo: QueryInfo,\n    queryContinuationFields: any,\n  ): ExecutionContext {\n    // Apply TOP component (TOP N is effectively OFFSET 0 LIMIT N)\n    let top = queryInfo.top;\n    if (typeof top === \"number\") {\n      if (queryContinuationFields?.limit !== undefined) {\n        top = queryContinuationFields.limit;\n      }\n      endpoint = new OffsetLimitEndpointComponent(endpoint, 0, top);\n    }\n\n    // Apply OFFSET+LIMIT component\n    let limit = queryInfo.limit;\n    let offset = queryInfo.offset;\n\n    if (queryContinuationFields) {\n      if (queryContinuationFields.limit !== undefined) {\n        limit = queryContinuationFields.limit;\n      }\n      if (queryContinuationFields.offset !== undefined) {\n        offset = queryContinuationFields.offset;\n      }\n    }\n\n    if (typeof limit === \"number\" && typeof offset === \"number\") {\n      endpoint = new OffsetLimitEndpointComponent(endpoint, offset, limit);\n    }\n\n    return endpoint;\n  }\n\n  /**\n   * Validates vector search buffer size constraints\n   */\n  private validateVectorSearchBufferSize(\n    vectorSearchBufferSize: number,\n    options: FeedOptions,\n  ): void {\n    const maxBufferSize = options[\"vectorSearchBufferSize\"]\n      ? options[\"vectorSearchBufferSize\"]\n      : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n\n    if (vectorSearchBufferSize > maxBufferSize) {\n      throw new ErrorResponse(\n        `Executing a vector search query with TOP or OFFSET + LIMIT value ${vectorSearchBufferSize} larger than the vectorSearchBufferSize ${maxBufferSize} ` +\n          `is not allowed`,\n      );\n    }\n  }\n\n  private calculateVectorSearchBufferSize(queryInfo: QueryInfo, options: FeedOptions): number {\n    if (queryInfo.top === 0 || queryInfo.limit === 0) return 0;\n    return queryInfo.top\n      ? queryInfo.top\n      : queryInfo.limit\n        ? queryInfo.offset + queryInfo.limit\n        : options[\"vectorSearchBufferSize\"] && options[\"vectorSearchBufferSize\"] > 0\n          ? options[\"vectorSearchBufferSize\"]\n          : QueryExecution.DEFAULT_MAX_VECTOR_SEARCH_BUFFER_SIZE;\n  }\n\n  private checkQueryConstraints(queryInfo: QueryInfo): void {\n    const hasTop = queryInfo.top || queryInfo.top === 0;\n    const hasLimit = queryInfo.limit || queryInfo.limit === 0;\n    if (!hasTop && !hasLimit) {\n      throw new ErrorResponse(\n        \"Executing a non-streaming search query without TOP or LIMIT can consume a large number of RUs \" +\n          \"very fast and have long runtimes. Please ensure you are using one of the above two filters \" +\n          \"with your vector search query.\",\n      );\n    }\n    return;\n  }\n\n  /**\n   * Analyzes query information and extracts key characteristics for query execution planning\n   */\n  private analyzeQueryInfo(queryInfo: QueryInfo) {\n    const sortOrders = queryInfo.orderBy;\n    const nonStreamingOrderBy = queryInfo.hasNonStreamingOrderBy;\n    const isOrderByQuery = Array.isArray(sortOrders) && sortOrders.length > 0;\n\n    // Check if this is a GROUP BY query\n    const isGroupByQuery =\n      Object.keys(queryInfo.groupByAliasToAggregateType || {}).length > 0 ||\n      (queryInfo.aggregates?.length || 0) > 0 ||\n      (queryInfo.groupByExpressions?.length || 0) > 0;\n\n    // Check if this is an unordered DISTINCT query\n    const isUnorderedDistinctQuery = queryInfo.distinctType === \"Unordered\";\n\n    // Determine if this query type supports continuation tokens\n    const querySupportsTokens =\n      !isUnorderedDistinctQuery && !isGroupByQuery && !nonStreamingOrderBy;\n\n    return {\n      sortOrders,\n      nonStreamingOrderBy,\n      isOrderByQuery,\n      isGroupByQuery,\n      isUnorderedDistinctQuery,\n      querySupportsTokens,\n    };\n  }\n}\n"]}

package/dist/browser/queryExecutionContext/queryFilteringStrategy/FilterStrategy.d.ts

@@ -0,0 +1,40 @@
+/**
+ * Represents the context for filtering, including the values from the continuation token.
+ */
+export interface FilterContext {
+    /**
+     * The _rid of the document from which the continuation token was created.
+     */
+    rid: string;
+    /**
+     * The number of documents with the same _rid to skip, used in JOIN queries.
+     */
+    skipCount: number;
+    /**
+     * The order by item values from the last document of the previous page.
+     */
+    orderByItems: any[];
+    /**
+     * The sort orders for the ORDER BY fields ("Ascending" or "Descending").
+     */
+    sortOrders: any[];
+    /**
+     * Query execution information from x-ms-cosmos-query-execution-info header.
+     */
+    queryExecutionInfo?: {
+        reverseRidEnabled: boolean;
+        reverseIndexScan: boolean;
+    };
+}
+/**
+ * Defines the interface for a post-fetch query filter.
+ */
+export interface FilterStrategy {
+    /**
+     * Applies the filtering logic to a set of documents.
+     * @param documents - The documents to filter.
+     * @returns The filtered set of documents.
+     */
+    applyFilter(documents: any[]): any[];
+}
+//# sourceMappingURL=FilterStrategy.d.ts.map

package/dist/browser/queryExecutionContext/queryFilteringStrategy/FilterStrategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"FilterStrategy.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/queryFilteringStrategy/FilterStrategy.ts"],"names":[],"mappings":"AAGA;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,GAAG,EAAE,MAAM,CAAC;IACZ;;OAEG;IACH,SAAS,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,CAAC;IACpB;;OAEG;IACH,UAAU,EAAE,GAAG,EAAE,CAAC;IAClB;;OAEG;IACH,kBAAkB,CAAC,EAAE;QACnB,iBAAiB,EAAE,OAAO,CAAC;QAC3B,gBAAgB,EAAE,OAAO,CAAC;KAC3B,CAAC;CACH;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;;;OAIG;IACH,WAAW,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;CACtC"}

package/dist/browser/queryExecutionContext/queryFilteringStrategy/FilterStrategy.js

@@ -0,0 +1,4 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+export {};
+//# sourceMappingURL=FilterStrategy.js.map

package/dist/browser/queryExecutionContext/queryFilteringStrategy/FilterStrategy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FilterStrategy.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/queryFilteringStrategy/FilterStrategy.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/**\n * Represents the context for filtering, including the values from the continuation token.\n */\nexport interface FilterContext {\n  /**\n   * The _rid of the document from which the continuation token was created.\n   */\n  rid: string;\n  /**\n   * The number of documents with the same _rid to skip, used in JOIN queries.\n   */\n  skipCount: number;\n  /**\n   * The order by item values from the last document of the previous page.\n   */\n  orderByItems: any[];\n  /**\n   * The sort orders for the ORDER BY fields (\"Ascending\" or \"Descending\").\n   */\n  sortOrders: any[];\n  /**\n   * Query execution information from x-ms-cosmos-query-execution-info header.\n   */\n  queryExecutionInfo?: {\n    reverseRidEnabled: boolean;\n    reverseIndexScan: boolean;\n  };\n}\n\n/**\n * Defines the interface for a post-fetch query filter.\n */\nexport interface FilterStrategy {\n  /**\n   * Applies the filtering logic to a set of documents.\n   * @param documents - The documents to filter.\n   * @returns The filtered set of documents.\n   */\n  applyFilter(documents: any[]): any[];\n}\n"]}

package/dist/browser/queryExecutionContext/queryFilteringStrategy/OrderByQueryRangeStrategy.d.ts

@@ -0,0 +1,49 @@
+import type { PartitionKeyRange } from "../../index.js";
+import type { TargetPartitionRangeStrategy, PartitionRangeFilterResult } from "./TargetPartitionRangeStrategy.js";
+import type { PartitionRangeWithContinuationToken } from "./TargetPartitionRangeManager.js";
+/**
+ * Strategy for filtering partition ranges in ORDER BY query execution context
+ * Supports resuming from continuation tokens with proper range-token pair management
+ * @hidden
+ */
+export declare class OrderByQueryRangeStrategy implements TargetPartitionRangeStrategy {
+    getStrategyType(): string;
+    filterPartitionRanges(targetRanges: PartitionKeyRange[], continuationRanges?: PartitionRangeWithContinuationToken[], queryInfo?: Record<string, unknown>): PartitionRangeFilterResult;
+    /**
+     * Creates a filter condition for ranges based on ORDER BY items and sort orders
+     * This filter ensures that ranges only return documents based on their position relative to the continuation point
+     * @param orderByItems - Array of order by items from the continuation token
+     * @param queryInfo - Query information containing sort orders and other metadata
+     * @param rangePosition - Whether this is for "left" or "right" ranges relative to continuation point
+     * @returns SQL filter condition string for the specified range position
+     */
+    private createRangeFilterCondition;
+    /**
+     * Extracts sort orders from query info
+     * @throws Error if sort order information is missing or invalid
+     */
+    private extractSortOrders;
+    /**
+     * Extracts field path from ORDER BY expressions in query plan
+     * @throws Error if orderByExpressions are not found or index is out of bounds or expression format is invalid
+     */
+    private extractFieldPath;
+    /**
+     * Creates a comparison condition based on the field, value, sort order, and range position
+     */
+    private createComparisonCondition;
+    /**
+     * Formats a value for use in SQL condition
+     */
+    private formatValueForSQL;
+    /**
+     * Compares partition key range boundaries with proper handling for inclusive/exclusive semantics
+     * @param boundary1 - First boundary to compare
+     * @param boundary2 - Second boundary to compare
+     * @returns negative if boundary1 is less than boundary2, positive if boundary1 is greater than boundary2, 0 if equal
+     */
+    private comparePartitionKeyBoundaries;
+    private isRangeBeforeAnother;
+    private isRangeAfterAnother;
+}
+//# sourceMappingURL=OrderByQueryRangeStrategy.d.ts.map

package/dist/browser/queryExecutionContext/queryFilteringStrategy/OrderByQueryRangeStrategy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OrderByQueryRangeStrategy.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/queryFilteringStrategy/OrderByQueryRangeStrategy.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,KAAK,EACV,4BAA4B,EAC5B,0BAA0B,EAC3B,MAAM,mCAAmC,CAAC;AAC3C,OAAO,KAAK,EAAE,mCAAmC,EAAE,MAAM,kCAAkC,CAAC;AAE5F;;;;GAIG;AACH,qBAAa,yBAA0B,YAAW,4BAA4B;IAC5E,eAAe,IAAI,MAAM;IAIzB,qBAAqB,CACnB,YAAY,EAAE,iBAAiB,EAAE,EACjC,kBAAkB,CAAC,EAAE,mCAAmC,EAAE,EAC1D,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAClC,0BAA0B;IAkG7B;;;;;;;OAOG;IACH,OAAO,CAAC,0BAA0B;IAkFlC;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAyCzB;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAyDxB;;OAEG;IACH,OAAO,CAAC,yBAAyB;IAkCjC;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAyBzB;;;;;OAKG;IACH,OAAO,CAAC,6BAA6B;IAUrC,OAAO,CAAC,oBAAoB;IAM5B,OAAO,CAAC,mBAAmB;CAK5B"}