@azure/cosmos 4.8.0 → 4.9.0

package/dist/browser/queryExecutionContext/ContinuationTokenManager/BaseContinuationTokenManager.js

@@ -0,0 +1,174 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { parseBaseContinuationToken, } from "../../documents/ContinuationToken/CompositeQueryContinuationToken.js";
+import { PartitionRangeManager } from "../PartitionRangeManager.js";
+/**
+ * Base abstract class for continuation token management.
+ * Provides common functionality shared between parallel and ORDER BY query token managers.
+ * @internal
+ */
+export class BaseContinuationTokenManager {
+    ranges = [];
+    partitionRangeManager = new PartitionRangeManager();
+    constructor(initialContinuationToken) {
+        if (initialContinuationToken) {
+            const token = parseBaseContinuationToken(initialContinuationToken);
+            if (token?.rangeMappings) {
+                this.ranges = token.rangeMappings;
+            }
+        }
+    }
+    /**
+     * Provides controlled access to partition range manager for subclasses.
+     * This is the only protected access point needed.
+     */
+    get partitionManager() {
+        return this.partitionRangeManager;
+    }
+    /**
+     * Provides controlled access to ranges for subclasses.
+     * Made protected to allow subclass range management.
+     */
+    get rangeList() {
+        return this.ranges;
+    }
+    /**
+     * Processes query results and generates continuation tokens for pagination.
+     * Handles response data processing, range management, and token generation.
+     *
+     * @param pageSize - Maximum number of items to return in this page
+     * @param isResponseEmpty - Whether the current response contains no data
+     * @param responseResult - Optional response data containing partition mappings and query-specific data
+     * @returns Object containing the end index for slicing results and optional continuation token for next page
+     */
+    paginateResults(pageSize, isResponseEmpty, responseResult) {
+        // Process response data
+        if (responseResult) {
+            this.processResponseResult(responseResult);
+        }
+        this.removeExhaustedRangesFromRanges();
+        const result = this.processRangesForPagination(pageSize, isResponseEmpty);
+        const tokenString = this.generateContinuationTokenString();
+        // Clean up processed ranges
+        this.trimProcessedData(result.processedRanges, result.endIndex);
+        return {
+            endIndex: result.endIndex,
+            continuationToken: tokenString,
+        };
+    }
+    /**
+     * Generates continuation token string using the appropriate serialization function.
+     * This provides a common implementation that delegates to query-specific logic.
+     */
+    generateContinuationTokenString() {
+        const token = this.getCurrentContinuationToken();
+        if (!token) {
+            return undefined;
+        }
+        const serializeFunction = this.getSerializationFunction();
+        return serializeFunction(token);
+    }
+    /**
+     * Cleans up processed data after a page has been returned.
+     * Handles both common and query-specific cleanup.
+     */
+    trimProcessedData(processedRanges, endIndex) {
+        processedRanges.forEach((rangeId) => {
+            this.partitionRangeManager.removePartitionRangeMapping(rangeId);
+        });
+        // Delegate query-specific cleanup to subclass
+        this.performQuerySpecificDataTrim(processedRanges, endIndex);
+    }
+    addPartitionKeyRangeMap(partitionKeyRangeMap) {
+        this.partitionRangeManager.addPartitionKeyRangeMap(partitionKeyRangeMap);
+    }
+    /**
+     * Processes the entire response result and updates the continuation token manager state.
+     * This encapsulates all response handling logic in one place.
+     */
+    processResponseResult(responseResult) {
+        // Handle partition key range map
+        if (responseResult.partitionKeyRangeMap) {
+            this.addPartitionKeyRangeMap(responseResult.partitionKeyRangeMap);
+        }
+        // Handle partition range updates
+        if (responseResult.updatedContinuationRanges) {
+            this.handlePartitionRangeChanges(responseResult.updatedContinuationRanges);
+        }
+        this.processQuerySpecificResponse(responseResult);
+    }
+    isPartitionExhausted(continuationToken) {
+        return (!continuationToken ||
+            continuationToken === "" ||
+            continuationToken === "null" ||
+            continuationToken.toLowerCase() === "null");
+    }
+    removeExhaustedRangesFromRanges() {
+        if (!this.ranges || !Array.isArray(this.ranges)) {
+            return;
+        }
+        this.ranges = this.ranges.filter((mapping) => {
+            if (!mapping) {
+                return false;
+            }
+            const isExhausted = this.isPartitionExhausted(mapping.continuationToken);
+            return !isExhausted;
+        });
+    }
+    handlePartitionRangeChanges(updatedContinuationRanges) {
+        if (updatedContinuationRanges && Object.keys(updatedContinuationRanges).length === 0) {
+            return;
+        }
+        Object.entries(updatedContinuationRanges).forEach(([rangeKey, rangeChange]) => {
+            this.processRangeChange(rangeKey, rangeChange);
+        });
+    }
+    processRangeChange(_rangeKey, rangeChange) {
+        const { oldRange, newRanges, continuationToken } = rangeChange;
+        if (newRanges.length === 1) {
+            this.handleRangeMerge(oldRange, newRanges[0], continuationToken);
+        }
+        else {
+            this.handleRangeSplit(oldRange, newRanges, continuationToken);
+        }
+    }
+    handleRangeMerge(oldRange, newRange, continuationToken) {
+        // Find existing range mapping to update in the common ranges array
+        const existingMappingIndex = this.ranges.findIndex((mapping) => mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max);
+        if (existingMappingIndex < 0) {
+            return;
+        }
+        // Update existing mapping with new range properties
+        const existingMapping = this.ranges[existingMappingIndex];
+        // Create new simplified QueryRange with updated boundaries
+        const updatedQueryRange = {
+            min: newRange.min,
+            max: newRange.max,
+        };
+        // Update the mapping
+        existingMapping.queryRange = updatedQueryRange;
+        existingMapping.continuationToken = continuationToken;
+    }
+    handleRangeSplit(oldRange, newRanges, continuationToken) {
+        // Remove the old range mapping from the common ranges array
+        this.ranges = this.ranges.filter((mapping) => !(mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max));
+        // Add new range mappings for each split range
+        newRanges.forEach((newRange) => {
+            this.createNewRangeMapping(newRange, continuationToken);
+        });
+    }
+    createNewRangeMapping(partitionKeyRange, continuationToken) {
+        // Create new simplified QueryRange
+        const queryRange = {
+            min: partitionKeyRange.min,
+            max: partitionKeyRange.max,
+        };
+        // Create new QueryRangeWithContinuationToken
+        const newRangeWithToken = {
+            queryRange: queryRange,
+            continuationToken: continuationToken,
+        };
+        this.ranges.push(newRangeWithToken);
+    }
+}
+//# sourceMappingURL=BaseContinuationTokenManager.js.map

package/dist/browser/queryExecutionContext/ContinuationTokenManager/BaseContinuationTokenManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/BaseContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EACL,0BAA0B,GAG3B,MAAM,sEAAsE,CAAC;AAK9E,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE;;;;GAIG;AACH,MAAM,OAAgB,4BAA4B;IACxC,MAAM,GAAsC,EAAE,CAAC;IACtC,qBAAqB,GAA0B,IAAI,qBAAqB,EAAE,CAAC;IAE5F,YAAY,wBAAiC;QAC3C,IAAI,wBAAwB,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,0BAA0B,CAAC,wBAAwB,CAAC,CAAC;YACnE,IAAI,KAAK,EAAE,aAAa,EAAE,CAAC;gBACzB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,IAAc,gBAAgB;QAC5B,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,IAAc,SAAS;QACrB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAQD;;;;;;;;OAQG;IACI,eAAe,CACpB,QAAgB,EAChB,eAAwB,EACxB,cAAoC;QAKpC,wBAAwB;QACxB,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,+BAA+B,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;QAE3D,4BAA4B;QAC5B,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,eAAe,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAEhE,OAAO;YACL,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,iBAAiB,EAAE,WAAW;SAC/B,CAAC;IACJ,CAAC;IAqBD;;;OAGG;IACK,+BAA+B;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACjD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,iBAAiB,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAC1D,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;;OAGG;IACK,iBAAiB,CAAC,eAAyB,EAAE,QAAgB;QACnE,eAAe,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAClC,IAAI,CAAC,qBAAqB,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,8CAA8C;QAC9C,IAAI,CAAC,4BAA4B,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;IAC/D,CAAC;IAEO,uBAAuB,CAAC,oBAAoD;QAClF,IAAI,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,oBAAoB,CAAC,CAAC;IAC3E,CAAC;IAED;;;OAGG;IACK,qBAAqB,CAAC,cAAmC;QAC/D,iCAAiC;QACjC,IAAI,cAAc,CAAC,oBAAoB,EAAE,CAAC;YACxC,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;QACpE,CAAC;QAED,iCAAiC;QACjC,IAAI,cAAc,CAAC,yBAAyB,EAAE,CAAC;YAC7C,IAAI,CAAC,2BAA2B,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;QAC7E,CAAC;QAED,IAAI,CAAC,4BAA4B,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC;IAEO,oBAAoB,CAAC,iBAAqC;QAChE,OAAO,CACL,CAAC,iBAAiB;YAClB,iBAAiB,KAAK,EAAE;YACxB,iBAAiB,KAAK,MAAM;YAC5B,iBAAiB,CAAC,WAAW,EAAE,KAAK,MAAM,CAC3C,CAAC;IACJ,CAAC;IAEO,+BAA+B;QACrC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACzE,OAAO,CAAC,WAAW,CAAC;QACtB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,2BAA2B,CAAC,yBAAgD;QAClF,IAAI,yBAAyB,IAAI,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrF,OAAO;QACT,CAAC;QACD,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,EAAE;YAC5E,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,SAAiB,EAAE,WAAiC;QAC7E,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,WAAW,CAAC;QAC/D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,EAAE,iBAAiB,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,QAAa,EAAE,QAAa,EAAE,iBAAyB;QAC9E,mEAAmE;QACnE,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAChD,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,CACrF,CAAC;QAEF,IAAI,oBAAoB,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,oDAAoD;QACpD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAE1D,2DAA2D;QAC3D,MAAM,iBAAiB,GAAkB;YACvC,GAAG,EAAE,QAAQ,CAAC,GAAG;YACjB,GAAG,EAAE,QAAQ,CAAC,GAAG;SAClB,CAAC;QAEF,qBAAqB;QACrB,eAAe,CAAC,UAAU,GAAG,iBAAiB,CAAC;QAC/C,eAAe,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IACxD,CAAC;IAEO,gBAAgB,CAAC,QAAa,EAAE,SAAgB,EAAE,iBAAyB;QACjF,4DAA4D;QAC5D,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAC9B,CAAC,OAAO,EAAE,EAAE,CACV,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,CAAC,CACxF,CAAC;QAEF,8CAA8C;QAC9C,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7B,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,iBAAsB,EAAE,iBAAyB;QAC7E,mCAAmC;QACnC,MAAM,UAAU,GAAkB;YAChC,GAAG,EAAE,iBAAiB,CAAC,GAAG;YAC1B,GAAG,EAAE,iBAAiB,CAAC,GAAG;SAC3B,CAAC;QAEF,6CAA6C;QAC7C,MAAM,iBAAiB,GAAoC;YACzD,UAAU,EAAE,UAAU;YACtB,iBAAiB,EAAE,iBAAiB;SACrC,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACtC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { QueryRangeMapping } from \"../queryRangeMapping.js\";\nimport {\n  parseBaseContinuationToken,\n  type QueryRangeWithContinuationToken,\n  type RangeBoundary,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport type {\n  PartitionRangeUpdate,\n  PartitionRangeUpdates,\n} from \"../../documents/ContinuationToken/PartitionRangeUpdate.js\";\nimport { PartitionRangeManager } from \"../PartitionRangeManager.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\n\n/**\n * Base abstract class for continuation token management.\n * Provides common functionality shared between parallel and ORDER BY query token managers.\n * @internal\n */\nexport abstract class BaseContinuationTokenManager {\n  private ranges: QueryRangeWithContinuationToken[] = [];\n  private readonly partitionRangeManager: PartitionRangeManager = new PartitionRangeManager();\n\n  constructor(initialContinuationToken?: string) {\n    if (initialContinuationToken) {\n      const token = parseBaseContinuationToken(initialContinuationToken);\n      if (token?.rangeMappings) {\n        this.ranges = token.rangeMappings;\n      }\n    }\n  }\n\n  /**\n   * Provides controlled access to partition range manager for subclasses.\n   * This is the only protected access point needed.\n   */\n  protected get partitionManager(): PartitionRangeManager {\n    return this.partitionRangeManager;\n  }\n\n  /**\n   * Provides controlled access to ranges for subclasses.\n   * Made protected to allow subclass range management.\n   */\n  protected get rangeList(): QueryRangeWithContinuationToken[] {\n    return this.ranges;\n  }\n\n  /**\n   * Gets the current continuation token for serialization.\n   * Returns undefined if no token exists or query is exhausted.\n   */\n  protected abstract getCurrentContinuationToken(): any;\n\n  /**\n   * Processes query results and generates continuation tokens for pagination.\n   * Handles response data processing, range management, and token generation.\n   *\n   * @param pageSize - Maximum number of items to return in this page\n   * @param isResponseEmpty - Whether the current response contains no data\n   * @param responseResult - Optional response data containing partition mappings and query-specific data\n   * @returns Object containing the end index for slicing results and optional continuation token for next page\n   */\n  public paginateResults(\n    pageSize: number,\n    isResponseEmpty: boolean,\n    responseResult?: ParallelQueryResult,\n  ): {\n    endIndex: number;\n    continuationToken?: string;\n  } {\n    // Process response data\n    if (responseResult) {\n      this.processResponseResult(responseResult);\n    }\n\n    this.removeExhaustedRangesFromRanges();\n    const result = this.processRangesForPagination(pageSize, isResponseEmpty);\n    const tokenString = this.generateContinuationTokenString();\n\n    // Clean up processed ranges\n    this.trimProcessedData(result.processedRanges, result.endIndex);\n\n    return {\n      endIndex: result.endIndex,\n      continuationToken: tokenString,\n    };\n  }\n\n  protected abstract processRangesForPagination(\n    pageSize: number,\n    isResponseEmpty: boolean,\n  ): {\n    endIndex: number;\n    processedRanges: string[];\n  };\n\n  protected abstract processQuerySpecificResponse(responseResult: ParallelQueryResult): void;\n  protected abstract performQuerySpecificDataTrim(\n    processedRanges: string[],\n    endIndex: number,\n  ): void;\n\n  /**\n   * Gets the serialization function for the specific continuation token type.\n   */\n  protected abstract getSerializationFunction(): (token: any) => string;\n\n  /**\n   * Generates continuation token string using the appropriate serialization function.\n   * This provides a common implementation that delegates to query-specific logic.\n   */\n  private generateContinuationTokenString(): string | undefined {\n    const token = this.getCurrentContinuationToken();\n    if (!token) {\n      return undefined;\n    }\n\n    const serializeFunction = this.getSerializationFunction();\n    return serializeFunction(token);\n  }\n\n  /**\n   * Cleans up processed data after a page has been returned.\n   * Handles both common and query-specific cleanup.\n   */\n  private trimProcessedData(processedRanges: string[], endIndex: number): void {\n    processedRanges.forEach((rangeId) => {\n      this.partitionRangeManager.removePartitionRangeMapping(rangeId);\n    });\n\n    // Delegate query-specific cleanup to subclass\n    this.performQuerySpecificDataTrim(processedRanges, endIndex);\n  }\n\n  private addPartitionKeyRangeMap(partitionKeyRangeMap: Map<string, QueryRangeMapping>): void {\n    this.partitionRangeManager.addPartitionKeyRangeMap(partitionKeyRangeMap);\n  }\n\n  /**\n   * Processes the entire response result and updates the continuation token manager state.\n   * This encapsulates all response handling logic in one place.\n   */\n  private processResponseResult(responseResult: ParallelQueryResult): void {\n    // Handle partition key range map\n    if (responseResult.partitionKeyRangeMap) {\n      this.addPartitionKeyRangeMap(responseResult.partitionKeyRangeMap);\n    }\n\n    // Handle partition range updates\n    if (responseResult.updatedContinuationRanges) {\n      this.handlePartitionRangeChanges(responseResult.updatedContinuationRanges);\n    }\n\n    this.processQuerySpecificResponse(responseResult);\n  }\n\n  private isPartitionExhausted(continuationToken: string | undefined): boolean {\n    return (\n      !continuationToken ||\n      continuationToken === \"\" ||\n      continuationToken === \"null\" ||\n      continuationToken.toLowerCase() === \"null\"\n    );\n  }\n\n  private removeExhaustedRangesFromRanges(): void {\n    if (!this.ranges || !Array.isArray(this.ranges)) {\n      return;\n    }\n    this.ranges = this.ranges.filter((mapping) => {\n      if (!mapping) {\n        return false;\n      }\n      const isExhausted = this.isPartitionExhausted(mapping.continuationToken);\n      return !isExhausted;\n    });\n  }\n\n  private handlePartitionRangeChanges(updatedContinuationRanges: PartitionRangeUpdates): void {\n    if (updatedContinuationRanges && Object.keys(updatedContinuationRanges).length === 0) {\n      return;\n    }\n    Object.entries(updatedContinuationRanges).forEach(([rangeKey, rangeChange]) => {\n      this.processRangeChange(rangeKey, rangeChange);\n    });\n  }\n\n  private processRangeChange(_rangeKey: string, rangeChange: PartitionRangeUpdate): void {\n    const { oldRange, newRanges, continuationToken } = rangeChange;\n    if (newRanges.length === 1) {\n      this.handleRangeMerge(oldRange, newRanges[0], continuationToken);\n    } else {\n      this.handleRangeSplit(oldRange, newRanges, continuationToken);\n    }\n  }\n\n  private handleRangeMerge(oldRange: any, newRange: any, continuationToken: string): void {\n    // Find existing range mapping to update in the common ranges array\n    const existingMappingIndex = this.ranges.findIndex(\n      (mapping) =>\n        mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max,\n    );\n\n    if (existingMappingIndex < 0) {\n      return;\n    }\n\n    // Update existing mapping with new range properties\n    const existingMapping = this.ranges[existingMappingIndex];\n\n    // Create new simplified QueryRange with updated boundaries\n    const updatedQueryRange: RangeBoundary = {\n      min: newRange.min,\n      max: newRange.max,\n    };\n\n    // Update the mapping\n    existingMapping.queryRange = updatedQueryRange;\n    existingMapping.continuationToken = continuationToken;\n  }\n\n  private handleRangeSplit(oldRange: any, newRanges: any[], continuationToken: string): void {\n    // Remove the old range mapping from the common ranges array\n    this.ranges = this.ranges.filter(\n      (mapping) =>\n        !(mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max),\n    );\n\n    // Add new range mappings for each split range\n    newRanges.forEach((newRange) => {\n      this.createNewRangeMapping(newRange, continuationToken);\n    });\n  }\n\n  private createNewRangeMapping(partitionKeyRange: any, continuationToken: string): void {\n    // Create new simplified QueryRange\n    const queryRange: RangeBoundary = {\n      min: partitionKeyRange.min,\n      max: partitionKeyRange.max,\n    };\n\n    // Create new QueryRangeWithContinuationToken\n    const newRangeWithToken: QueryRangeWithContinuationToken = {\n      queryRange: queryRange,\n      continuationToken: continuationToken,\n    };\n\n    this.ranges.push(newRangeWithToken);\n  }\n}\n"]}

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ContinuationTokenManagerFactory.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ContinuationTokenManagerFactory.d.ts.map

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ContinuationTokenManagerFactory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContinuationTokenManagerFactory.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/ContinuationTokenManagerFactory.ts"],"names":[],"mappings":""}

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ContinuationTokenManagerFactory.js

@@ -0,0 +1,28 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { ParallelQueryContinuationTokenManager } from "./ParallelQueryContinuationTokenManager.js";
+import { OrderByQueryContinuationTokenManager } from "./OrderByQueryContinuationTokenManager.js";
+/**
+ * Factory for creating appropriate continuation token managers based on query type.
+ * @internal
+ */
+export const ContinuationTokenManagerFactory = {
+    /**
+     * Creates the appropriate continuation token manager based on query type.
+     * Only call this for queries that support continuation tokens.
+     * @param collectionLink - Collection link/RID
+     * @param initialContinuationToken - Optional initial continuation token
+     * @param isOrderByQuery - Whether this is an ORDER BY query
+     * @returns Appropriate continuation token manager instance
+     * @internal
+     */
+    create(collectionLink, initialContinuationToken, isOrderByQuery = false) {
+        if (isOrderByQuery) {
+            return new OrderByQueryContinuationTokenManager(collectionLink, initialContinuationToken);
+        }
+        else {
+            return new ParallelQueryContinuationTokenManager(collectionLink, initialContinuationToken);
+        }
+    },
+};
+//# sourceMappingURL=ContinuationTokenManagerFactory.js.map

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ContinuationTokenManagerFactory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContinuationTokenManagerFactory.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/ContinuationTokenManagerFactory.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,qCAAqC,EAAE,MAAM,4CAA4C,CAAC;AACnG,OAAO,EAAE,oCAAoC,EAAE,MAAM,2CAA2C,CAAC;AAEjG;;;GAGG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG;IAC7C;;;;;;;;OAQG;IACH,MAAM,CACJ,cAAsB,EACtB,wBAAiC,EACjC,iBAA0B,KAAK;QAE/B,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,IAAI,oCAAoC,CAAC,cAAc,EAAE,wBAAwB,CAAC,CAAC;QAC5F,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,qCAAqC,CAAC,cAAc,EAAE,wBAAwB,CAAC,CAAC;QAC7F,CAAC;IACH,CAAC;CACF,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { BaseContinuationTokenManager } from \"./BaseContinuationTokenManager.js\";\nimport { ParallelQueryContinuationTokenManager } from \"./ParallelQueryContinuationTokenManager.js\";\nimport { OrderByQueryContinuationTokenManager } from \"./OrderByQueryContinuationTokenManager.js\";\n\n/**\n * Factory for creating appropriate continuation token managers based on query type.\n * @internal\n */\nexport const ContinuationTokenManagerFactory = {\n  /**\n   * Creates the appropriate continuation token manager based on query type.\n   * Only call this for queries that support continuation tokens.\n   * @param collectionLink - Collection link/RID\n   * @param initialContinuationToken - Optional initial continuation token\n   * @param isOrderByQuery - Whether this is an ORDER BY query\n   * @returns Appropriate continuation token manager instance\n   * @internal\n   */\n  create(\n    collectionLink: string,\n    initialContinuationToken?: string,\n    isOrderByQuery: boolean = false,\n  ): BaseContinuationTokenManager {\n    if (isOrderByQuery) {\n      return new OrderByQueryContinuationTokenManager(collectionLink, initialContinuationToken);\n    } else {\n      return new ParallelQueryContinuationTokenManager(collectionLink, initialContinuationToken);\n    }\n  },\n};\n"]}

package/dist/browser/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=OrderByQueryContinuationTokenManager.d.ts.map

package/dist/browser/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OrderByQueryContinuationTokenManager.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.ts"],"names":[],"mappings":""}

package/dist/browser/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.js

@@ -0,0 +1,129 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { BaseContinuationTokenManager } from "./BaseContinuationTokenManager.js";
+import { createOrderByQueryContinuationToken, parseOrderByQueryContinuationToken, serializeOrderByQueryContinuationToken, } from "../../documents/ContinuationToken/OrderByQueryContinuationToken.js";
+import { convertRangeMappingToQueryRange } from "../../documents/ContinuationToken/CompositeQueryContinuationToken.js";
+/**
+ * Manages continuation tokens for ORDER BY queries using single-range sequential processing.
+ * Uses OrderByQueryContinuationToken for tracking ORDER BY items and skip counts.
+ * @internal
+ */
+export class OrderByQueryContinuationTokenManager extends BaseContinuationTokenManager {
+    continuationToken;
+    orderByItemsArray;
+    collectionLink;
+    constructor(collectionLink, initialContinuationToken) {
+        super(initialContinuationToken);
+        this.collectionLink = collectionLink;
+        this.orderByItemsArray = [];
+        if (initialContinuationToken) {
+            this.continuationToken = parseOrderByQueryContinuationToken(initialContinuationToken);
+        }
+    }
+    processQuerySpecificResponse(responseResult) {
+        // Clear existing items and add new ones without reassigning the array reference
+        this.orderByItemsArray.length = 0;
+        if (responseResult.orderByItems) {
+            this.orderByItemsArray.push(...responseResult.orderByItems);
+        }
+    }
+    performQuerySpecificDataTrim(_processedRanges, endIndex) {
+        this.sliceOrderByItemsArray(endIndex);
+    }
+    sliceOrderByItemsArray(endIndex) {
+        if (endIndex === 0 || endIndex >= this.orderByItemsArray.length) {
+            // Clear the array without reassigning
+            this.orderByItemsArray.length = 0;
+        }
+        else {
+            // Remove items from 0 to endIndex-1, keeping items from endIndex onwards
+            this.orderByItemsArray.splice(0, endIndex);
+        }
+    }
+    processRangesForPagination(pageSize, isResponseEmpty = false) {
+        // Handle empty response case - update the previous valid continuation token
+        if (isResponseEmpty && this.continuationToken) {
+            let rangeProcessingResult;
+            if (this.rangeList.length === 0) {
+                rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);
+            }
+            else {
+                rangeProcessingResult = this.partitionManager.processEmptyOrderByRanges(this.rangeList);
+            }
+            const { lastRangeBeforePageLimit } = rangeProcessingResult;
+            if (lastRangeBeforePageLimit) {
+                // Use the range matching the continuation token for empty response
+                this.continuationToken.rangeMappings = [
+                    convertRangeMappingToQueryRange(lastRangeBeforePageLimit),
+                ];
+            }
+            else {
+                // Range is exhausted - clear the continuation token
+                this.continuationToken = undefined;
+            }
+            return {
+                endIndex: rangeProcessingResult.endIndex,
+                processedRanges: rangeProcessingResult.processedRanges,
+            };
+        }
+        // Normal processing path - handle non-empty responses
+        const rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);
+        const { lastRangeBeforePageLimit } = rangeProcessingResult;
+        // Check if we have a valid range to continue with
+        if (!lastRangeBeforePageLimit) {
+            this.continuationToken = undefined;
+            return {
+                endIndex: rangeProcessingResult.endIndex,
+                processedRanges: rangeProcessingResult.processedRanges,
+            };
+        }
+        const queryRange = convertRangeMappingToQueryRange(lastRangeBeforePageLimit);
+        // Extract ORDER BY items from the last item on the page
+        let lastOrderByItems;
+        let documentRid;
+        let skipCount = 0;
+        if (rangeProcessingResult.endIndex > 0 && this.orderByItemsArray.length > 0) {
+            const lastItemIndexOnPage = rangeProcessingResult.endIndex - 1;
+            if (lastItemIndexOnPage < this.orderByItemsArray.length) {
+                lastOrderByItems = this.orderByItemsArray[lastItemIndexOnPage].orderByItems;
+                documentRid = this.orderByItemsArray[lastItemIndexOnPage]._rid;
+                // Calculate skip count: count how many documents in the page have the same RID
+                // This handles JOIN queries where multiple documents can have the same RID
+                skipCount = 0;
+                for (let i = 0; i <= lastItemIndexOnPage; i++) {
+                    if (this.orderByItemsArray[i]._rid === documentRid) {
+                        skipCount++;
+                    }
+                }
+            }
+        }
+        // If we don't have valid ORDER BY items, we cannot create a proper continuation token
+        // This can happen when the response doesn't contain ORDER BY metadata or when there are no results
+        if (!lastOrderByItems || lastOrderByItems.length === 0) {
+            this.continuationToken = undefined;
+            return {
+                endIndex: rangeProcessingResult.endIndex,
+                processedRanges: rangeProcessingResult.processedRanges,
+            };
+        }
+        const rangeMappings = [queryRange];
+        // Create new ORDER BY continuation token
+        this.continuationToken = createOrderByQueryContinuationToken(rangeMappings, lastOrderByItems, this.collectionLink, // Container RID/link
+        skipCount, // Number of documents with the same RID already processed
+        documentRid, // Document RID from the last item in the page
+        lastRangeBeforePageLimit.offset, // Current offset for OFFSET/LIMIT queries
+        lastRangeBeforePageLimit.limit, // Current limit for OFFSET/LIMIT queries
+        lastRangeBeforePageLimit.hashedLastResult);
+        return {
+            endIndex: rangeProcessingResult.endIndex,
+            processedRanges: rangeProcessingResult.processedRanges,
+        };
+    }
+    getCurrentContinuationToken() {
+        return this.continuationToken;
+    }
+    getSerializationFunction() {
+        return serializeOrderByQueryContinuationToken;
+    }
+}
+//# sourceMappingURL=OrderByQueryContinuationTokenManager.js.map

package/dist/browser/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OrderByQueryContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAGjF,OAAO,EACL,mCAAmC,EACnC,kCAAkC,EAClC,sCAAsC,GACvC,MAAM,oEAAoE,CAAC;AAC5E,OAAO,EAAE,+BAA+B,EAAE,MAAM,sEAAsE,CAAC;AAEvH;;;;GAIG;AACH,MAAM,OAAO,oCAAqC,SAAQ,4BAA4B;IAC5E,iBAAiB,CAA4C;IACpD,iBAAiB,CAAuB;IACxC,cAAc,CAAS;IAExC,YAAY,cAAsB,EAAE,wBAAiC;QACnE,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;QAC5B,IAAI,wBAAwB,EAAE,CAAC;YAC7B,IAAI,CAAC,iBAAiB,GAAG,kCAAkC,CAAC,wBAAwB,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;IAES,4BAA4B,CAAC,cAAmC;QACxE,gFAAgF;QAChF,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;QAClC,IAAI,cAAc,CAAC,YAAY,EAAE,CAAC;YAChC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAES,4BAA4B,CAAC,gBAA0B,EAAE,QAAgB;QACjF,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAEO,sBAAsB,CAAC,QAAgB;QAC7C,IAAI,QAAQ,KAAK,CAAC,IAAI,QAAQ,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC;YAChE,sCAAsC;YACtC,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,yEAAyE;YACzE,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAES,0BAA0B,CAClC,QAAgB,EAChB,kBAA2B,KAAK;QAEhC,4EAA4E;QAC5E,IAAI,eAAe,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC9C,IAAI,qBAAqB,CAAC;YAE1B,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAChC,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAC/E,CAAC;iBAAM,CAAC;gBACN,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1F,CAAC;YAED,MAAM,EAAE,wBAAwB,EAAE,GAAG,qBAAqB,CAAC;YAC3D,IAAI,wBAAwB,EAAE,CAAC;gBAC7B,mEAAmE;gBACnE,IAAI,CAAC,iBAAiB,CAAC,aAAa,GAAG;oBACrC,+BAA+B,CAAC,wBAAwB,CAAC;iBAC1D,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,oDAAoD;gBACpD,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACrC,CAAC;YACD,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,MAAM,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACnF,MAAM,EAAE,wBAAwB,EAAE,GAAG,qBAAqB,CAAC;QAE3D,kDAAkD;QAClD,IAAI,CAAC,wBAAwB,EAAE,CAAC;YAC9B,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACnC,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,+BAA+B,CAAC,wBAAwB,CAAC,CAAC;QAE7E,wDAAwD;QACxD,IAAI,gBAAmC,CAAC;QACxC,IAAI,WAAmB,CAAC;QACxB,IAAI,SAAS,GAAW,CAAC,CAAC;QAC1B,IAAI,qBAAqB,CAAC,QAAQ,GAAG,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5E,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,QAAQ,GAAG,CAAC,CAAC;YAE/D,IAAI,mBAAmB,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC;gBACxD,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC,YAAY,CAAC;gBAC5E,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC;gBAC/D,+EAA+E;gBAC/E,2EAA2E;gBAC3E,SAAS,GAAG,CAAC,CAAC;gBACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,mBAAmB,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC9C,IAAI,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACnD,SAAS,EAAE,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,sFAAsF;QACtF,mGAAmG;QACnG,IAAI,CAAC,gBAAgB,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACnC,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,UAAU,CAAC,CAAC;QAEnC,yCAAyC;QACzC,IAAI,CAAC,iBAAiB,GAAG,mCAAmC,CAC1D,aAAa,EACb,gBAAgB,EAChB,IAAI,CAAC,cAAc,EAAE,qBAAqB;QAC1C,SAAS,EAAE,0DAA0D;QACrE,WAAW,EAAE,8CAA8C;QAC3D,wBAAwB,CAAC,MAAM,EAAE,0CAA0C;QAC3E,wBAAwB,CAAC,KAAK,EAAE,yCAAyC;QACzE,wBAAwB,CAAC,gBAAgB,CAC1C,CAAC;QAEF,OAAO;YACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;YACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;SACvD,CAAC;IACJ,CAAC;IAES,2BAA2B;QACnC,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAES,wBAAwB;QAChC,OAAO,sCAAsC,CAAC;IAChD,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { BaseContinuationTokenManager } from \"./BaseContinuationTokenManager.js\";\nimport type { ParallelQueryResult, OrderByItemWithRid } from \"../parallelQueryResult.js\";\nimport type { OrderByQueryContinuationToken } from \"../../documents/ContinuationToken/OrderByQueryContinuationToken.js\";\nimport {\n  createOrderByQueryContinuationToken,\n  parseOrderByQueryContinuationToken,\n  serializeOrderByQueryContinuationToken,\n} from \"../../documents/ContinuationToken/OrderByQueryContinuationToken.js\";\nimport { convertRangeMappingToQueryRange } from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\n\n/**\n * Manages continuation tokens for ORDER BY queries using single-range sequential processing.\n * Uses OrderByQueryContinuationToken for tracking ORDER BY items and skip counts.\n * @internal\n */\nexport class OrderByQueryContinuationTokenManager extends BaseContinuationTokenManager {\n  private continuationToken: OrderByQueryContinuationToken | undefined;\n  private readonly orderByItemsArray: OrderByItemWithRid[];\n  private readonly collectionLink: string;\n\n  constructor(collectionLink: string, initialContinuationToken?: string) {\n    super(initialContinuationToken);\n    this.collectionLink = collectionLink;\n    this.orderByItemsArray = [];\n    if (initialContinuationToken) {\n      this.continuationToken = parseOrderByQueryContinuationToken(initialContinuationToken);\n    }\n  }\n\n  protected processQuerySpecificResponse(responseResult: ParallelQueryResult): void {\n    // Clear existing items and add new ones without reassigning the array reference\n    this.orderByItemsArray.length = 0;\n    if (responseResult.orderByItems) {\n      this.orderByItemsArray.push(...responseResult.orderByItems);\n    }\n  }\n\n  protected performQuerySpecificDataTrim(_processedRanges: string[], endIndex: number): void {\n    this.sliceOrderByItemsArray(endIndex);\n  }\n\n  private sliceOrderByItemsArray(endIndex: number): void {\n    if (endIndex === 0 || endIndex >= this.orderByItemsArray.length) {\n      // Clear the array without reassigning\n      this.orderByItemsArray.length = 0;\n    } else {\n      // Remove items from 0 to endIndex-1, keeping items from endIndex onwards\n      this.orderByItemsArray.splice(0, endIndex);\n    }\n  }\n\n  protected processRangesForPagination(\n    pageSize: number,\n    isResponseEmpty: boolean = false,\n  ): { endIndex: number; processedRanges: string[] } {\n    // Handle empty response case - update the previous valid continuation token\n    if (isResponseEmpty && this.continuationToken) {\n      let rangeProcessingResult;\n\n      if (this.rangeList.length === 0) {\n        rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);\n      } else {\n        rangeProcessingResult = this.partitionManager.processEmptyOrderByRanges(this.rangeList);\n      }\n\n      const { lastRangeBeforePageLimit } = rangeProcessingResult;\n      if (lastRangeBeforePageLimit) {\n        // Use the range matching the continuation token for empty response\n        this.continuationToken.rangeMappings = [\n          convertRangeMappingToQueryRange(lastRangeBeforePageLimit),\n        ];\n      } else {\n        // Range is exhausted - clear the continuation token\n        this.continuationToken = undefined;\n      }\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    // Normal processing path - handle non-empty responses\n    const rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);\n    const { lastRangeBeforePageLimit } = rangeProcessingResult;\n\n    // Check if we have a valid range to continue with\n    if (!lastRangeBeforePageLimit) {\n      this.continuationToken = undefined;\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    const queryRange = convertRangeMappingToQueryRange(lastRangeBeforePageLimit);\n\n    // Extract ORDER BY items from the last item on the page\n    let lastOrderByItems: any[] | undefined;\n    let documentRid: string;\n    let skipCount: number = 0;\n    if (rangeProcessingResult.endIndex > 0 && this.orderByItemsArray.length > 0) {\n      const lastItemIndexOnPage = rangeProcessingResult.endIndex - 1;\n\n      if (lastItemIndexOnPage < this.orderByItemsArray.length) {\n        lastOrderByItems = this.orderByItemsArray[lastItemIndexOnPage].orderByItems;\n        documentRid = this.orderByItemsArray[lastItemIndexOnPage]._rid;\n        // Calculate skip count: count how many documents in the page have the same RID\n        // This handles JOIN queries where multiple documents can have the same RID\n        skipCount = 0;\n        for (let i = 0; i <= lastItemIndexOnPage; i++) {\n          if (this.orderByItemsArray[i]._rid === documentRid) {\n            skipCount++;\n          }\n        }\n      }\n    }\n\n    // If we don't have valid ORDER BY items, we cannot create a proper continuation token\n    // This can happen when the response doesn't contain ORDER BY metadata or when there are no results\n    if (!lastOrderByItems || lastOrderByItems.length === 0) {\n      this.continuationToken = undefined;\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    const rangeMappings = [queryRange];\n\n    // Create new ORDER BY continuation token\n    this.continuationToken = createOrderByQueryContinuationToken(\n      rangeMappings,\n      lastOrderByItems,\n      this.collectionLink, // Container RID/link\n      skipCount, // Number of documents with the same RID already processed\n      documentRid, // Document RID from the last item in the page\n      lastRangeBeforePageLimit.offset, // Current offset for OFFSET/LIMIT queries\n      lastRangeBeforePageLimit.limit, // Current limit for OFFSET/LIMIT queries\n      lastRangeBeforePageLimit.hashedLastResult, // Hash for distinct queries\n    );\n\n    return {\n      endIndex: rangeProcessingResult.endIndex,\n      processedRanges: rangeProcessingResult.processedRanges,\n    };\n  }\n\n  protected getCurrentContinuationToken(): OrderByQueryContinuationToken | undefined {\n    return this.continuationToken;\n  }\n\n  protected getSerializationFunction(): (token: OrderByQueryContinuationToken) => string {\n    return serializeOrderByQueryContinuationToken;\n  }\n}\n"]}

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ParallelQueryContinuationTokenManager.d.ts.map

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ParallelQueryContinuationTokenManager.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.ts"],"names":[],"mappings":""}

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.js

@@ -0,0 +1,66 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { BaseContinuationTokenManager } from "./BaseContinuationTokenManager.js";
+import { createCompositeQueryContinuationToken, serializeCompositeToken, parseCompositeQueryContinuationToken, convertRangeMappingToQueryRange, } from "../../documents/ContinuationToken/CompositeQueryContinuationToken.js";
+/**
+ * Manages continuation tokens for parallel queries using multi-range aggregation.
+ * Uses CompositeQueryContinuationToken for range tracking across multiple partitions.
+ * @internal
+ */
+export class ParallelQueryContinuationTokenManager extends BaseContinuationTokenManager {
+    continuationToken;
+    collectionLink;
+    constructor(collectionLink, initialContinuationToken) {
+        super(initialContinuationToken);
+        this.collectionLink = collectionLink;
+        if (initialContinuationToken) {
+            this.continuationToken = parseCompositeQueryContinuationToken(initialContinuationToken);
+        }
+    }
+    processRangesForPagination(pageSize, _isResponseEmpty) {
+        const result = this.partitionManager.processParallelRanges(pageSize);
+        if (!result || !result.processedRangeMappings || result.processedRangeMappings.length === 0) {
+            return { endIndex: 0, processedRanges: [] };
+        }
+        const rangeMappings = result.processedRangeMappings.map((mapping) => convertRangeMappingToQueryRange(mapping));
+        if (!this.continuationToken) {
+            this.continuationToken = createCompositeQueryContinuationToken(this.collectionLink, rangeMappings);
+        }
+        else {
+            this.updateExistingCompositeContinuationToken(rangeMappings);
+        }
+        if (result.lastPartitionBeforeCutoff && result.lastPartitionBeforeCutoff.mapping) {
+            this.continuationToken.offset = result.lastPartitionBeforeCutoff.mapping.offset;
+            this.continuationToken.limit = result.lastPartitionBeforeCutoff.mapping.limit;
+        }
+        return { endIndex: result.endIndex, processedRanges: result.processedRanges };
+    }
+    getCurrentContinuationToken() {
+        return this.continuationToken;
+    }
+    getSerializationFunction() {
+        return serializeCompositeToken;
+    }
+    processQuerySpecificResponse(_responseResult) {
+        // Parallel queries don't need additional response processing
+    }
+    performQuerySpecificDataTrim(_processedRanges, _endIndex) {
+        // Parallel queries don't need additional cleanup
+    }
+    updateExistingCompositeContinuationToken(rangeMappings) {
+        for (const newRange of rangeMappings) {
+            // Check if this range already exists in the token
+            const existingRangeIndex = this.continuationToken.rangeMappings.findIndex((existingRange) => existingRange.queryRange.min === newRange.queryRange.min &&
+                existingRange.queryRange.max === newRange.queryRange.max);
+            if (existingRangeIndex >= 0) {
+                // Range exists - update the continuation token
+                this.continuationToken.rangeMappings[existingRangeIndex] = newRange;
+            }
+            else {
+                // New range - add to the rangeMappings array
+                this.continuationToken.rangeMappings.push(newRange);
+            }
+        }
+    }
+}
+//# sourceMappingURL=ParallelQueryContinuationTokenManager.js.map

package/dist/browser/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ParallelQueryContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAMjF,OAAO,EACL,qCAAqC,EACrC,uBAAuB,EACvB,oCAAoC,EACpC,+BAA+B,GAChC,MAAM,sEAAsE,CAAC;AAG9E;;;;GAIG;AACH,MAAM,OAAO,qCAAsC,SAAQ,4BAA4B;IAC7E,iBAAiB,CAA8C;IACtD,cAAc,CAAS;IAExC,YAAY,cAAsB,EAAE,wBAAiC;QACnE,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,wBAAwB,EAAE,CAAC;YAC7B,IAAI,CAAC,iBAAiB,GAAG,oCAAoC,CAAC,wBAAwB,CAAC,CAAC;QAC1F,CAAC;IACH,CAAC;IAES,0BAA0B,CAClC,QAAgB,EAChB,gBAAyB;QAEzB,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACrE,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,sBAAsB,IAAI,MAAM,CAAC,sBAAsB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5F,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;QAC9C,CAAC;QAED,MAAM,aAAa,GAAsC,MAAM,CAAC,sBAAsB,CAAC,GAAG,CACxF,CAAC,OAA0B,EAAE,EAAE,CAAC,+BAA+B,CAAC,OAAO,CAAC,CACzE,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,IAAI,CAAC,iBAAiB,GAAG,qCAAqC,CAC5D,IAAI,CAAC,cAAc,EACnB,aAAa,CACd,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,wCAAwC,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,MAAM,CAAC,yBAAyB,IAAI,MAAM,CAAC,yBAAyB,CAAC,OAAO,EAAE,CAAC;YACjF,IAAI,CAAC,iBAAkB,CAAC,MAAM,GAAG,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,MAAM,CAAC;YACjF,IAAI,CAAC,iBAAkB,CAAC,KAAK,GAAG,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,KAAK,CAAC;QACjF,CAAC;QACD,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;IAChF,CAAC;IAES,2BAA2B;QACnC,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAES,wBAAwB;QAChC,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAES,4BAA4B,CAAC,eAAoC;QACzE,6DAA6D;IAC/D,CAAC;IAES,4BAA4B,CAAC,gBAA0B,EAAE,SAAiB;QAClF,iDAAiD;IACnD,CAAC;IAEO,wCAAwC,CAC9C,aAAgD;QAEhD,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;YACrC,kDAAkD;YAClD,MAAM,kBAAkB,GAAG,IAAI,CAAC,iBAAkB,CAAC,aAAa,CAAC,SAAS,CACxE,CAAC,aAAa,EAAE,EAAE,CAChB,aAAa,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,UAAU,CAAC,GAAG;gBACxD,aAAa,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,UAAU,CAAC,GAAG,CAC3D,CAAC;YAEF,IAAI,kBAAkB,IAAI,CAAC,EAAE,CAAC;gBAC5B,+CAA+C;gBAC/C,IAAI,CAAC,iBAAkB,CAAC,aAAa,CAAC,kBAAkB,CAAC,GAAG,QAAQ,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACN,6CAA6C;gBAC7C,IAAI,CAAC,iBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { BaseContinuationTokenManager } from \"./BaseContinuationTokenManager.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\nimport type {\n  CompositeQueryContinuationToken,\n  QueryRangeWithContinuationToken,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport {\n  createCompositeQueryContinuationToken,\n  serializeCompositeToken,\n  parseCompositeQueryContinuationToken,\n  convertRangeMappingToQueryRange,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport type { QueryRangeMapping } from \"../queryRangeMapping.js\";\n\n/**\n * Manages continuation tokens for parallel queries using multi-range aggregation.\n * Uses CompositeQueryContinuationToken for range tracking across multiple partitions.\n * @internal\n */\nexport class ParallelQueryContinuationTokenManager extends BaseContinuationTokenManager {\n  private continuationToken: CompositeQueryContinuationToken | undefined;\n  private readonly collectionLink: string;\n\n  constructor(collectionLink: string, initialContinuationToken?: string) {\n    super(initialContinuationToken);\n    this.collectionLink = collectionLink;\n    if (initialContinuationToken) {\n      this.continuationToken = parseCompositeQueryContinuationToken(initialContinuationToken);\n    }\n  }\n\n  protected processRangesForPagination(\n    pageSize: number,\n    _isResponseEmpty: boolean,\n  ): { endIndex: number; processedRanges: string[] } {\n    const result = this.partitionManager.processParallelRanges(pageSize);\n    if (!result || !result.processedRangeMappings || result.processedRangeMappings.length === 0) {\n      return { endIndex: 0, processedRanges: [] };\n    }\n\n    const rangeMappings: QueryRangeWithContinuationToken[] = result.processedRangeMappings.map(\n      (mapping: QueryRangeMapping) => convertRangeMappingToQueryRange(mapping),\n    );\n\n    if (!this.continuationToken) {\n      this.continuationToken = createCompositeQueryContinuationToken(\n        this.collectionLink,\n        rangeMappings,\n      );\n    } else {\n      this.updateExistingCompositeContinuationToken(rangeMappings);\n    }\n\n    if (result.lastPartitionBeforeCutoff && result.lastPartitionBeforeCutoff.mapping) {\n      this.continuationToken!.offset = result.lastPartitionBeforeCutoff.mapping.offset;\n      this.continuationToken!.limit = result.lastPartitionBeforeCutoff.mapping.limit;\n    }\n    return { endIndex: result.endIndex, processedRanges: result.processedRanges };\n  }\n\n  protected getCurrentContinuationToken(): CompositeQueryContinuationToken | undefined {\n    return this.continuationToken;\n  }\n\n  protected getSerializationFunction(): (token: CompositeQueryContinuationToken) => string {\n    return serializeCompositeToken;\n  }\n\n  protected processQuerySpecificResponse(_responseResult: ParallelQueryResult): void {\n    // Parallel queries don't need additional response processing\n  }\n\n  protected performQuerySpecificDataTrim(_processedRanges: string[], _endIndex: number): void {\n    // Parallel queries don't need additional cleanup\n  }\n\n  private updateExistingCompositeContinuationToken(\n    rangeMappings: QueryRangeWithContinuationToken[],\n  ): void {\n    for (const newRange of rangeMappings) {\n      // Check if this range already exists in the token\n      const existingRangeIndex = this.continuationToken!.rangeMappings.findIndex(\n        (existingRange) =>\n          existingRange.queryRange.min === newRange.queryRange.min &&\n          existingRange.queryRange.max === newRange.queryRange.max,\n      );\n\n      if (existingRangeIndex >= 0) {\n        // Range exists - update the continuation token\n        this.continuationToken!.rangeMappings[existingRangeIndex] = newRange;\n      } else {\n        // New range - add to the rangeMappings array\n        this.continuationToken!.rangeMappings.push(newRange);\n      }\n    }\n  }\n}\n"]}

package/dist/browser/queryExecutionContext/ContinuationTokenParser.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ContinuationTokenParser.d.ts.map

package/dist/browser/queryExecutionContext/ContinuationTokenParser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContinuationTokenParser.d.ts","sourceRoot":"","sources":["../../../src/queryExecutionContext/ContinuationTokenParser.ts"],"names":[],"mappings":""}

package/dist/browser/queryExecutionContext/ContinuationTokenParser.js

@@ -0,0 +1,23 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+/**
+ * Parses a continuation token string to extract specific fields without full deserialization
+ * Supports both CompositeQueryContinuationToken and OrderByQueryContinuationToken formats
+ * @param continuationToken - The continuation token string to parse
+ * @returns Object containing offset, limit, and hashedLastResult if present
+ * @internal
+ */
+export function parseContinuationTokenFields(continuationToken) {
+    try {
+        const parsed = JSON.parse(continuationToken);
+        return {
+            offset: parsed.offset,
+            limit: parsed.limit,
+            hashedLastResult: parsed.hashedLastResult,
+        };
+    }
+    catch (error) {
+        throw new Error(`Failed to parse continuation token: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}
+//# sourceMappingURL=ContinuationTokenParser.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ContinuationTokenParser.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/ContinuationTokenParser.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAYlC;;;;;;GAMG;AACH,MAAM,UAAU,4BAA4B,CAAC,iBAAyB;IACpE,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAE7C,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;SAC1C,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,uCAAuC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAChG,CAAC;IACJ,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/**\n * Lightweight interface for continuation token fields needed by pipeline\n * @internal\n */\nexport interface ContinuationTokenFields {\n  offset?: number;\n  limit?: number;\n  hashedLastResult?: string;\n}\n\n/**\n * Parses a continuation token string to extract specific fields without full deserialization\n * Supports both CompositeQueryContinuationToken and OrderByQueryContinuationToken formats\n * @param continuationToken - The continuation token string to parse\n * @returns Object containing offset, limit, and hashedLastResult if present\n * @internal\n */\nexport function parseContinuationTokenFields(continuationToken: string): ContinuationTokenFields {\n  try {\n    const parsed = JSON.parse(continuationToken);\n\n    return {\n      offset: parsed.offset,\n      limit: parsed.limit,\n      hashedLastResult: parsed.hashedLastResult,\n    };\n  } catch (error) {\n    throw new Error(\n      `Failed to parse continuation token: ${error instanceof Error ? error.message : String(error)}`,\n    );\n  }\n}\n"]}

package/dist/browser/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"GroupByEndpointComponent.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAMhE,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6CAA6C,CAAC;AAO1F,cAAc;AACd,qBAAa,wBAAyB,YAAW,gBAAgB;IAE7D,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,SAAS;gBADT,gBAAgB,EAAE,gBAAgB,EAClC,SAAS,EAAE,SAAS;IAG9B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAmD;IAC7E,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAa;IAClD,OAAO,CAAC,SAAS,CAAkB;IAE5B,cAAc,IAAI,OAAO;IAInB,SAAS,CAAC,cAAc,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAkEtF,OAAO,CAAC,uBAAuB;CAWhC"}
+{"version":3,"file":"GroupByEndpointComponent.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAMhE,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6CAA6C,CAAC;AAS1F,cAAc;AACd,qBAAa,wBAAyB,YAAW,gBAAgB;IAE7D,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,SAAS;gBADT,gBAAgB,EAAE,gBAAgB,EAClC,SAAS,EAAE,SAAS;IAG9B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAmD;IAC7E,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAa;IAClD,OAAO,CAAC,SAAS,CAAkB;IAE5B,cAAc,IAAI,OAAO;IAInB,SAAS,CAAC,cAAc,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAsFtF,OAAO,CAAC,uBAAuB;CAehC"}

package/dist/browser/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.js

@@ -2,6 +2,7 @@ import { hashObject } from "../../utils/hashObject.js";
 import { createAggregator } from "../Aggregators/index.js";
 import { getInitialHeader, mergeHeaders } from "../headerUtils.js";
 import { emptyGroup, extractAggregateResult } from "./emptyGroup.js";
+import { createParallelQueryResult } from "../parallelQueryResult.js";
 /** @hidden */
 export class GroupByEndpointComponent {
     executionContext;
@@ -25,15 +26,27 @@ export class GroupByEndpointComponent {
         }
         const aggregateHeaders = getInitialHeader();
         const response = await this.executionContext.fetchMore(diagnosticNode);
+        if (!response) {
+            // If there are any groupings, consolidate and return them
+            if (this.groupings.size > 0) {
+                return this.consolidateGroupResults(aggregateHeaders);
+            }
+            return { result: undefined, headers: aggregateHeaders };
+        }
         mergeHeaders(aggregateHeaders, response.headers);
-        if (response === undefined || response.result === undefined) {
+        if (response.result === undefined ||
+            !Array.isArray(response.result.buffer) ||
+            response.result.buffer.length === 0) {
             // If there are any groupings, consolidate and return them
             if (this.groupings.size > 0) {
                 return this.consolidateGroupResults(aggregateHeaders);
             }
             return { result: undefined, headers: aggregateHeaders };
         }
-        for (const item of response.result) {
+        const parallelResult = response.result;
+        const dataToProcess = parallelResult.buffer;
+        // Process GROUP BY aggregation logic
+        for (const item of dataToProcess) {
             // If it exists, process it via aggregators
             if (item) {
                 const group = item.groupByItems ? await hashObject(item.groupByItems) : emptyGroup;
@@ -72,10 +85,10 @@ export class GroupByEndpointComponent {
             }
         }
         if (this.executionContext.hasMoreResults()) {
-            return {
-                result: [],
-                headers: aggregateHeaders,
-            };
+            // Return empty buffer but preserve the structure and pass-through fields
+            const result = createParallelQueryResult([], // empty buffer
+            new Map());
+            return { result, headers: aggregateHeaders };
         }
         else {
             return this.consolidateGroupResults(aggregateHeaders);
@@ -90,7 +103,9 @@ export class GroupByEndpointComponent {
             this.aggregateResultArray.push(groupResult);
         }
         this.completed = true;
-        return { result: this.aggregateResultArray, headers: aggregateHeaders };
+        // Return in the new structure format using the utility function
+        const result = createParallelQueryResult(this.aggregateResultArray, new Map());
+        return { result, headers: aggregateHeaders };
     }
 }
 //# sourceMappingURL=GroupByEndpointComponent.js.map