@azure/cosmos 4.8.0 → 4.9.0

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

@@ -0,0 +1,133 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OrderByQueryContinuationTokenManager = void 0;
+const BaseContinuationTokenManager_js_1 = require("./BaseContinuationTokenManager.js");
+const OrderByQueryContinuationToken_js_1 = require("../../documents/ContinuationToken/OrderByQueryContinuationToken.js");
+const CompositeQueryContinuationToken_js_1 = require("../../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
+ */
+class OrderByQueryContinuationTokenManager extends BaseContinuationTokenManager_js_1.BaseContinuationTokenManager {
+    continuationToken;
+    orderByItemsArray;
+    collectionLink;
+    constructor(collectionLink, initialContinuationToken) {
+        super(initialContinuationToken);
+        this.collectionLink = collectionLink;
+        this.orderByItemsArray = [];
+        if (initialContinuationToken) {
+            this.continuationToken = (0, OrderByQueryContinuationToken_js_1.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 = [
+                    (0, CompositeQueryContinuationToken_js_1.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 = (0, CompositeQueryContinuationToken_js_1.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 = (0, OrderByQueryContinuationToken_js_1.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 OrderByQueryContinuationToken_js_1.serializeOrderByQueryContinuationToken;
+    }
+}
+exports.OrderByQueryContinuationTokenManager = OrderByQueryContinuationTokenManager;
+//# sourceMappingURL=OrderByQueryContinuationTokenManager.js.map

package/dist/commonjs/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,uFAAiF;AAGjF,yHAI4E;AAC5E,6HAAuH;AAEvH;;;;GAIG;AACH,MAAa,oCAAqC,SAAQ,8DAA4B;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,IAAA,qEAAkC,EAAC,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,IAAA,oEAA+B,EAAC,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,IAAA,oEAA+B,EAAC,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,IAAA,sEAAmC,EAC1D,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,yEAAsC,CAAC;IAChD,CAAC;CACF;AA3ID,oFA2IC","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/commonjs/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.d.ts

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

package/dist/commonjs/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/commonjs/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.js

@@ -0,0 +1,70 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ParallelQueryContinuationTokenManager = void 0;
+const BaseContinuationTokenManager_js_1 = require("./BaseContinuationTokenManager.js");
+const CompositeQueryContinuationToken_js_1 = require("../../documents/ContinuationToken/CompositeQueryContinuationToken.js");
+/**
+ * Manages continuation tokens for parallel queries using multi-range aggregation.
+ * Uses CompositeQueryContinuationToken for range tracking across multiple partitions.
+ * @internal
+ */
+class ParallelQueryContinuationTokenManager extends BaseContinuationTokenManager_js_1.BaseContinuationTokenManager {
+    continuationToken;
+    collectionLink;
+    constructor(collectionLink, initialContinuationToken) {
+        super(initialContinuationToken);
+        this.collectionLink = collectionLink;
+        if (initialContinuationToken) {
+            this.continuationToken = (0, CompositeQueryContinuationToken_js_1.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) => (0, CompositeQueryContinuationToken_js_1.convertRangeMappingToQueryRange)(mapping));
+        if (!this.continuationToken) {
+            this.continuationToken = (0, CompositeQueryContinuationToken_js_1.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 CompositeQueryContinuationToken_js_1.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);
+            }
+        }
+    }
+}
+exports.ParallelQueryContinuationTokenManager = ParallelQueryContinuationTokenManager;
+//# sourceMappingURL=ParallelQueryContinuationTokenManager.js.map

package/dist/commonjs/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,uFAAiF;AAMjF,6HAK8E;AAG9E;;;;GAIG;AACH,MAAa,qCAAsC,SAAQ,8DAA4B;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,IAAA,yEAAoC,EAAC,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,IAAA,oEAA+B,EAAC,OAAO,CAAC,CACzE,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,IAAI,CAAC,iBAAiB,GAAG,IAAA,0EAAqC,EAC5D,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,4DAAuB,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;AA7ED,sFA6EC","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/commonjs/queryExecutionContext/ContinuationTokenParser.d.ts

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

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

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

package/dist/commonjs/queryExecutionContext/ContinuationTokenParser.js

@@ -0,0 +1,26 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseContinuationTokenFields = parseContinuationTokenFields;
+/**
+ * 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
+ */
+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/commonjs/queryExecutionContext/ContinuationTokenParser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ContinuationTokenParser.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/ContinuationTokenParser.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAmBlC,oEAcC;AArBD;;;;;;GAMG;AACH,SAAgB,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/commonjs/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/commonjs/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.js

@@ -5,6 +5,7 @@ const hashObject_js_1 = require("../../utils/hashObject.js");
 const index_js_1 = require("../Aggregators/index.js");
 const headerUtils_js_1 = require("../headerUtils.js");
 const emptyGroup_js_1 = require("./emptyGroup.js");
+const parallelQueryResult_js_1 = require("../parallelQueryResult.js");
 /** @hidden */
 class GroupByEndpointComponent {
     executionContext;
@@ -28,15 +29,27 @@ class GroupByEndpointComponent {
         }
         const aggregateHeaders = (0, headerUtils_js_1.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 };
+        }
         (0, headerUtils_js_1.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 (0, hashObject_js_1.hashObject)(item.groupByItems) : emptyGroup_js_1.emptyGroup;
@@ -75,10 +88,10 @@ class GroupByEndpointComponent {
             }
         }
         if (this.executionContext.hasMoreResults()) {
-            return {
-                result: [],
-                headers: aggregateHeaders,
-            };
+            // Return empty buffer but preserve the structure and pass-through fields
+            const result = (0, parallelQueryResult_js_1.createParallelQueryResult)([], // empty buffer
+            new Map());
+            return { result, headers: aggregateHeaders };
         }
         else {
             return this.consolidateGroupResults(aggregateHeaders);
@@ -93,7 +106,9 @@ 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 = (0, parallelQueryResult_js_1.createParallelQueryResult)(this.aggregateResultArray, new Map());
+        return { result, headers: aggregateHeaders };
     }
 }
 exports.GroupByEndpointComponent = GroupByEndpointComponent;

package/dist/commonjs/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.js.map

@@ -1 +1 @@
-{"version":3,"file":"GroupByEndpointComponent.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.ts"],"names":[],"mappings":";;;AAMA,6DAAuD;AAEvD,sDAA2D;AAC3D,sDAAmE;AACnE,mDAAqE;AAQrE,cAAc;AACd,MAAa,wBAAwB;IAEzB;IACA;IAFV,YACU,gBAAkC,EAClC,SAAoB;QADpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,cAAS,GAAT,SAAS,CAAW;IAC3B,CAAC;IAEa,SAAS,GAAyC,IAAI,GAAG,EAAE,CAAC;IAC5D,oBAAoB,GAAU,EAAE,CAAC;IAC1C,SAAS,GAAY,KAAK,CAAC;IAE5B,cAAc;QACnB,OAAO,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAsC;QAC3D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAA,iCAAgB,GAAE;aAC5B,CAAC;QACJ,CAAC;QACD,MAAM,gBAAgB,GAAG,IAAA,iCAAgB,GAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACvE,IAAA,6BAAY,EAAC,gBAAgB,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;QAEjD,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC5D,0DAA0D;YAC1D,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,OAAO,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;YACxD,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC1D,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,MAAyB,EAAE,CAAC;YACtD,2CAA2C;YAC3C,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,IAAA,0BAAU,EAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,0BAAU,CAAC;gBACnF,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;gBAC7B,IAAI,WAAW,EAAE,CAAC;oBAChB,2CAA2C;oBAC3C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;wBACvC,4FAA4F;wBAC5F,MAAM,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC;4BACxC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;4BACd,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;wBACjC,MAAM,eAAe,GAAG,IAAA,sCAAsB,EAAC,qBAAqB,CAAC,CAAC;wBACtE,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;oBAClD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,oGAAoG;oBACpG,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;oBAC3B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;oBACpC,2CAA2C;oBAC3C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;wBACvC,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,2BAA2B,CAAC,GAAG,CAAC,CAAC;wBACtE,4DAA4D;wBAC5D,MAAM,UAAU,GAAG,IAAA,2BAAgB,EAAC,aAAa,CAAC,CAAC;wBACnD,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;wBAC9B,IAAI,aAAa,EAAE,CAAC;4BAClB,MAAM,eAAe,GAAG,IAAA,sCAAsB,EAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;4BAC7D,UAAU,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;wBACxC,CAAC;6BAAM,CAAC;4BACN,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;wBACrC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3C,OAAO;gBACL,MAAM,EAAE,EAAE;gBACV,OAAO,EAAE,gBAAgB;aAC1B,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,uBAAuB,CAAC,gBAA+B;QAC7D,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,MAAM,WAAW,GAAQ,EAAE,CAAC;YAC5B,KAAK,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;gBAC5D,WAAW,CAAC,YAAY,CAAC,GAAG,UAAU,CAAC,SAAS,EAAE,CAAC;YACrD,CAAC;YACD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,oBAAoB,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAC1E,CAAC;CACF;AA3FD,4DA2FC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { Response } from \"../../request/index.js\";\nimport type { ExecutionContext } from \"../ExecutionContext.js\";\nimport type { CosmosHeaders } from \"../CosmosHeaders.js\";\nimport type { QueryInfo } from \"../../request/ErrorResponse.js\";\nimport { hashObject } from \"../../utils/hashObject.js\";\nimport type { Aggregator } from \"../Aggregators/index.js\";\nimport { createAggregator } from \"../Aggregators/index.js\";\nimport { getInitialHeader, mergeHeaders } from \"../headerUtils.js\";\nimport { emptyGroup, extractAggregateResult } from \"./emptyGroup.js\";\nimport type { DiagnosticNodeInternal } from \"../../diagnostics/DiagnosticNodeInternal.js\";\n\ninterface GroupByResult {\n  groupByItems: any[];\n  payload: any;\n}\n\n/** @hidden */\nexport class GroupByEndpointComponent implements ExecutionContext {\n  constructor(\n    private executionContext: ExecutionContext,\n    private queryInfo: QueryInfo,\n  ) {}\n\n  private readonly groupings: Map<string, Map<string, Aggregator>> = new Map();\n  private readonly aggregateResultArray: any[] = [];\n  private completed: boolean = false;\n\n  public hasMoreResults(): boolean {\n    return this.executionContext.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    if (this.completed) {\n      return {\n        result: undefined,\n        headers: getInitialHeader(),\n      };\n    }\n    const aggregateHeaders = getInitialHeader();\n    const response = await this.executionContext.fetchMore(diagnosticNode);\n    mergeHeaders(aggregateHeaders, response.headers);\n\n    if (response === undefined || response.result === undefined) {\n      // If there are any groupings, consolidate and return them\n      if (this.groupings.size > 0) {\n        return this.consolidateGroupResults(aggregateHeaders);\n      }\n      return { result: undefined, headers: aggregateHeaders };\n    }\n\n    for (const item of response.result as GroupByResult[]) {\n      // If it exists, process it via aggregators\n      if (item) {\n        const group = item.groupByItems ? await hashObject(item.groupByItems) : emptyGroup;\n        const aggregators = this.groupings.get(group);\n        const payload = item.payload;\n        if (aggregators) {\n          // Iterator over all results in the payload\n          for (const key of Object.keys(payload)) {\n            // in case the value of a group is null make sure we create a dummy payload with item2==null\n            const effectiveGroupByValue = payload[key]\n              ? payload[key]\n              : new Map().set(\"item2\", null);\n            const aggregateResult = extractAggregateResult(effectiveGroupByValue);\n            aggregators.get(key).aggregate(aggregateResult);\n          }\n        } else {\n          // This is the first time we have seen a grouping. Setup the initial result without aggregate values\n          const grouping = new Map();\n          this.groupings.set(group, grouping);\n          // Iterator over all results in the payload\n          for (const key of Object.keys(payload)) {\n            const aggregateType = this.queryInfo.groupByAliasToAggregateType[key];\n            // Create a new aggregator for this specific aggregate field\n            const aggregator = createAggregator(aggregateType);\n            grouping.set(key, aggregator);\n            if (aggregateType) {\n              const aggregateResult = extractAggregateResult(payload[key]);\n              aggregator.aggregate(aggregateResult);\n            } else {\n              aggregator.aggregate(payload[key]);\n            }\n          }\n        }\n      }\n    }\n\n    if (this.executionContext.hasMoreResults()) {\n      return {\n        result: [],\n        headers: aggregateHeaders,\n      };\n    } else {\n      return this.consolidateGroupResults(aggregateHeaders);\n    }\n  }\n\n  private consolidateGroupResults(aggregateHeaders: CosmosHeaders): Response<any> {\n    for (const grouping of this.groupings.values()) {\n      const groupResult: any = {};\n      for (const [aggregateKey, aggregator] of grouping.entries()) {\n        groupResult[aggregateKey] = aggregator.getResult();\n      }\n      this.aggregateResultArray.push(groupResult);\n    }\n    this.completed = true;\n    return { result: this.aggregateResultArray, headers: aggregateHeaders };\n  }\n}\n"]}
+{"version":3,"file":"GroupByEndpointComponent.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.ts"],"names":[],"mappings":";;;AAMA,6DAAuD;AAEvD,sDAA2D;AAC3D,sDAAmE;AACnE,mDAAqE;AAGrE,sEAAsE;AAOtE,cAAc;AACd,MAAa,wBAAwB;IAEzB;IACA;IAFV,YACU,gBAAkC,EAClC,SAAoB;QADpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,cAAS,GAAT,SAAS,CAAW;IAC3B,CAAC;IAEa,SAAS,GAAyC,IAAI,GAAG,EAAE,CAAC;IAC5D,oBAAoB,GAAU,EAAE,CAAC;IAC1C,SAAS,GAAY,KAAK,CAAC;IAE5B,cAAc;QACnB,OAAO,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAsC;QAC3D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAA,iCAAgB,GAAE;aAC5B,CAAC;QACJ,CAAC;QACD,MAAM,gBAAgB,GAAG,IAAA,iCAAgB,GAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAEvE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,0DAA0D;YAC1D,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,OAAO,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;YACxD,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC1D,CAAC;QAED,IAAA,6BAAY,EAAC,gBAAgB,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;QAEjD,IACE,QAAQ,CAAC,MAAM,KAAK,SAAS;YAC7B,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC;YACtC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EACnC,CAAC;YACD,0DAA0D;YAC1D,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,OAAO,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;YACxD,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC1D,CAAC;QAED,MAAM,cAAc,GAAG,QAAQ,CAAC,MAA6B,CAAC;QAC9D,MAAM,aAAa,GAAoB,cAAc,CAAC,MAAyB,CAAC;QAEhF,qCAAqC;QACrC,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,2CAA2C;YAC3C,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,IAAA,0BAAU,EAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,0BAAU,CAAC;gBACnF,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;gBAC7B,IAAI,WAAW,EAAE,CAAC;oBAChB,2CAA2C;oBAC3C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;wBACvC,4FAA4F;wBAC5F,MAAM,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC;4BACxC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;4BACd,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;wBACjC,MAAM,eAAe,GAAG,IAAA,sCAAsB,EAAC,qBAAqB,CAAC,CAAC;wBACtE,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;oBAClD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,oGAAoG;oBACpG,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;oBAC3B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;oBACpC,2CAA2C;oBAC3C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;wBACvC,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,2BAA2B,CAAC,GAAG,CAAC,CAAC;wBACtE,4DAA4D;wBAC5D,MAAM,UAAU,GAAG,IAAA,2BAAgB,EAAC,aAAa,CAAC,CAAC;wBACnD,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;wBAC9B,IAAI,aAAa,EAAE,CAAC;4BAClB,MAAM,eAAe,GAAG,IAAA,sCAAsB,EAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;4BAC7D,UAAU,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;wBACxC,CAAC;6BAAM,CAAC;4BACN,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;wBACrC,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3C,yEAAyE;YACzE,MAAM,MAAM,GAAG,IAAA,kDAAyB,EACtC,EAAE,EAAE,eAAe;YACnB,IAAI,GAAG,EAAE,CACV,CAAC;YAEF,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,uBAAuB,CAAC,gBAA+B;QAC7D,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,MAAM,WAAW,GAAQ,EAAE,CAAC;YAC5B,KAAK,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;gBAC5D,WAAW,CAAC,YAAY,CAAC,GAAG,UAAU,CAAC,SAAS,EAAE,CAAC;YACrD,CAAC;YACD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,gEAAgE;QAChE,MAAM,MAAM,GAAG,IAAA,kDAAyB,EAAC,IAAI,CAAC,oBAAoB,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QAE/E,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAC/C,CAAC;CACF;AAnHD,4DAmHC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { Response } from \"../../request/index.js\";\nimport type { ExecutionContext } from \"../ExecutionContext.js\";\nimport type { CosmosHeaders } from \"../CosmosHeaders.js\";\nimport type { QueryInfo } from \"../../request/ErrorResponse.js\";\nimport { hashObject } from \"../../utils/hashObject.js\";\nimport type { Aggregator } from \"../Aggregators/index.js\";\nimport { createAggregator } from \"../Aggregators/index.js\";\nimport { getInitialHeader, mergeHeaders } from \"../headerUtils.js\";\nimport { emptyGroup, extractAggregateResult } from \"./emptyGroup.js\";\nimport type { DiagnosticNodeInternal } from \"../../diagnostics/DiagnosticNodeInternal.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\nimport { createParallelQueryResult } from \"../parallelQueryResult.js\";\n\ninterface GroupByResult {\n  groupByItems: any[];\n  payload: any;\n}\n\n/** @hidden */\nexport class GroupByEndpointComponent implements ExecutionContext {\n  constructor(\n    private executionContext: ExecutionContext,\n    private queryInfo: QueryInfo,\n  ) {}\n\n  private readonly groupings: Map<string, Map<string, Aggregator>> = new Map();\n  private readonly aggregateResultArray: any[] = [];\n  private completed: boolean = false;\n\n  public hasMoreResults(): boolean {\n    return this.executionContext.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    if (this.completed) {\n      return {\n        result: undefined,\n        headers: getInitialHeader(),\n      };\n    }\n    const aggregateHeaders = getInitialHeader();\n    const response = await this.executionContext.fetchMore(diagnosticNode);\n\n    if (!response) {\n      // If there are any groupings, consolidate and return them\n      if (this.groupings.size > 0) {\n        return this.consolidateGroupResults(aggregateHeaders);\n      }\n      return { result: undefined, headers: aggregateHeaders };\n    }\n\n    mergeHeaders(aggregateHeaders, response.headers);\n\n    if (\n      response.result === undefined ||\n      !Array.isArray(response.result.buffer) ||\n      response.result.buffer.length === 0\n    ) {\n      // If there are any groupings, consolidate and return them\n      if (this.groupings.size > 0) {\n        return this.consolidateGroupResults(aggregateHeaders);\n      }\n      return { result: undefined, headers: aggregateHeaders };\n    }\n\n    const parallelResult = response.result as ParallelQueryResult;\n    const dataToProcess: GroupByResult[] = parallelResult.buffer as GroupByResult[];\n\n    // Process GROUP BY aggregation logic\n    for (const item of dataToProcess) {\n      // If it exists, process it via aggregators\n      if (item) {\n        const group = item.groupByItems ? await hashObject(item.groupByItems) : emptyGroup;\n        const aggregators = this.groupings.get(group);\n        const payload = item.payload;\n        if (aggregators) {\n          // Iterator over all results in the payload\n          for (const key of Object.keys(payload)) {\n            // in case the value of a group is null make sure we create a dummy payload with item2==null\n            const effectiveGroupByValue = payload[key]\n              ? payload[key]\n              : new Map().set(\"item2\", null);\n            const aggregateResult = extractAggregateResult(effectiveGroupByValue);\n            aggregators.get(key).aggregate(aggregateResult);\n          }\n        } else {\n          // This is the first time we have seen a grouping. Setup the initial result without aggregate values\n          const grouping = new Map();\n          this.groupings.set(group, grouping);\n          // Iterator over all results in the payload\n          for (const key of Object.keys(payload)) {\n            const aggregateType = this.queryInfo.groupByAliasToAggregateType[key];\n            // Create a new aggregator for this specific aggregate field\n            const aggregator = createAggregator(aggregateType);\n            grouping.set(key, aggregator);\n            if (aggregateType) {\n              const aggregateResult = extractAggregateResult(payload[key]);\n              aggregator.aggregate(aggregateResult);\n            } else {\n              aggregator.aggregate(payload[key]);\n            }\n          }\n        }\n      }\n    }\n\n    if (this.executionContext.hasMoreResults()) {\n      // Return empty buffer but preserve the structure and pass-through fields\n      const result = createParallelQueryResult(\n        [], // empty buffer\n        new Map(),\n      );\n\n      return { result, headers: aggregateHeaders };\n    } else {\n      return this.consolidateGroupResults(aggregateHeaders);\n    }\n  }\n\n  private consolidateGroupResults(aggregateHeaders: CosmosHeaders): Response<any> {\n    for (const grouping of this.groupings.values()) {\n      const groupResult: any = {};\n      for (const [aggregateKey, aggregator] of grouping.entries()) {\n        groupResult[aggregateKey] = aggregator.getResult();\n      }\n      this.aggregateResultArray.push(groupResult);\n    }\n    this.completed = true;\n\n    // Return in the new structure format using the utility function\n    const result = createParallelQueryResult(this.aggregateResultArray, new Map());\n\n    return { result, headers: aggregateHeaders };\n  }\n}\n"]}

package/dist/commonjs/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"GroupByValueEndpointComponent.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.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,EAAiB,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAM/E,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6CAA6C,CAAC;AAO1F,cAAc;AACd,qBAAa,6BAA8B,YAAW,gBAAgB;IAOlE,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,SAAS;IAPnB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAsC;IAClE,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAa;IAClD,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,SAAS,CAAkB;gBAGzB,gBAAgB,EAAE,gBAAgB,EAClC,SAAS,EAAE,SAAS;IAMvB,cAAc,IAAI,OAAO;IAInB,SAAS,CAAC,cAAc,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAiEtF,OAAO,CAAC,yBAAyB;CAalC"}
+{"version":3,"file":"GroupByValueEndpointComponent.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.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,EAAiB,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAM/E,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6CAA6C,CAAC;AAS1F,cAAc;AACd,qBAAa,6BAA8B,YAAW,gBAAgB;IAOlE,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,SAAS;IAPnB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAsC;IAClE,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAa;IAClD,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,SAAS,CAAkB;gBAGzB,gBAAgB,EAAE,gBAAgB,EAClC,SAAS,EAAE,SAAS;IAMvB,cAAc,IAAI,OAAO;IAInB,SAAS,CAAC,cAAc,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAwFtF,OAAO,CAAC,yBAAyB;CAgBlC"}

package/dist/commonjs/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.js

@@ -5,6 +5,7 @@ const hashObject_js_1 = require("../../utils/hashObject.js");
 const index_js_1 = require("../Aggregators/index.js");
 const headerUtils_js_1 = require("../headerUtils.js");
 const emptyGroup_js_1 = require("./emptyGroup.js");
+const parallelQueryResult_js_1 = require("../parallelQueryResult.js");
 /** @hidden */
 class GroupByValueEndpointComponent {
     executionContext;
@@ -31,14 +32,24 @@ class GroupByValueEndpointComponent {
         }
         const aggregateHeaders = (0, headerUtils_js_1.getInitialHeader)();
         const response = await this.executionContext.fetchMore(diagnosticNode);
+        if (!response) {
+            if (this.aggregators.size > 0) {
+                return this.generateAggregateResponse(aggregateHeaders);
+            }
+            return { result: undefined, headers: aggregateHeaders };
+        }
         (0, headerUtils_js_1.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 (this.aggregators.size > 0) {
                 return this.generateAggregateResponse(aggregateHeaders);
             }
             return { result: undefined, headers: aggregateHeaders };
         }
-        for (const item of response.result) {
+        const parallelResult = response.result;
+        const dataToProcess = parallelResult.buffer;
+        for (const item of dataToProcess) {
             if (item) {
                 let grouping = emptyGroup_js_1.emptyGroup;
                 let payload = item;
@@ -69,13 +80,17 @@ class GroupByValueEndpointComponent {
         }
         // We bail early since we got an undefined result back `[{}]`
         if (this.completed) {
+            const result = (0, parallelQueryResult_js_1.createParallelQueryResult)([], new Map());
             return {
-                result: undefined,
+                result,
                 headers: aggregateHeaders,
             };
         }
         if (this.executionContext.hasMoreResults()) {
-            return { result: [], headers: aggregateHeaders };
+            // Return empty buffer but preserve the structure and pass-through fields
+            const result = (0, parallelQueryResult_js_1.createParallelQueryResult)([], // empty buffer
+            new Map());
+            return { result, headers: aggregateHeaders };
         }
         else {
             // If no results are left in the underlying execution context, convert our aggregate results to an array
@@ -90,8 +105,9 @@ class GroupByValueEndpointComponent {
             }
         }
         this.completed = true;
+        const result = (0, parallelQueryResult_js_1.createParallelQueryResult)(this.aggregateResultArray, new Map());
         return {
-            result: this.aggregateResultArray,
+            result,
             headers: aggregateHeaders,
         };
     }

package/dist/commonjs/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.js.map

@@ -1 +1 @@
-{"version":3,"file":"GroupByValueEndpointComponent.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.ts"],"names":[],"mappings":";;;AAMA,6DAAuD;AAEvD,sDAA2D;AAC3D,sDAAmE;AACnE,mDAAqE;AAQrE,cAAc;AACd,MAAa,6BAA6B;IAO9B;IACA;IAPO,WAAW,GAA4B,IAAI,GAAG,EAAE,CAAC;IACjD,oBAAoB,GAAU,EAAE,CAAC;IAC1C,aAAa,CAAgB;IAC7B,SAAS,GAAY,KAAK,CAAC;IAEnC,YACU,gBAAkC,EAClC,SAAoB;QADpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,cAAS,GAAT,SAAS,CAAW;QAE5B,uDAAuD;QACvD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC;IAEM,cAAc;QACnB,OAAO,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAsC;QAC3D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAA,iCAAgB,GAAE;aAC5B,CAAC;QACJ,CAAC;QACD,MAAM,gBAAgB,GAAG,IAAA,iCAAgB,GAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACvE,IAAA,6BAAY,EAAC,gBAAgB,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;QAEjD,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC5D,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAC,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;YAC1D,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC1D,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,MAAyB,EAAE,CAAC;YACtD,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,QAAQ,GAAW,0BAAU,CAAC;gBAClC,IAAI,OAAO,GAAQ,IAAI,CAAC;gBACxB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,4FAA4F;oBAC5F,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;oBACvB,QAAQ,GAAG,MAAM,IAAA,0BAAU,EAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBACjD,CAAC;gBAED,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAClD,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,4EAA4E;oBAC5E,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAA,2BAAgB,EAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;gBACvE,CAAC;gBAED,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACvB,MAAM,eAAe,GAAG,IAAA,sCAAsB,EAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC3D,yFAAyF;oBACzF,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;wBAC7B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;oBACxB,CAAC;oBACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,yEAAyE;oBACzE,sDAAsD;oBACtD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;QACH,CAAC;QAED,6DAA6D;QAC7D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,gBAAgB;aAC1B,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3C,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QACnD,CAAC;aAAM,CAAC;YACN,wGAAwG;YACxG,OAAO,IAAI,CAAC,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,yBAAyB,CAAC,gBAA+B;QAC/D,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YACnD,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,oBAAoB;YACjC,OAAO,EAAE,gBAAgB;SAC1B,CAAC;IACJ,CAAC;CACF;AAhGD,sEAgGC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { Response } from \"../../request/index.js\";\nimport type { ExecutionContext } from \"../ExecutionContext.js\";\nimport type { CosmosHeaders } from \"../CosmosHeaders.js\";\nimport type { AggregateType, QueryInfo } from \"../../request/ErrorResponse.js\";\nimport { hashObject } from \"../../utils/hashObject.js\";\nimport type { Aggregator } from \"../Aggregators/index.js\";\nimport { createAggregator } from \"../Aggregators/index.js\";\nimport { getInitialHeader, mergeHeaders } from \"../headerUtils.js\";\nimport { emptyGroup, extractAggregateResult } from \"./emptyGroup.js\";\nimport type { DiagnosticNodeInternal } from \"../../diagnostics/DiagnosticNodeInternal.js\";\n\ninterface GroupByResult {\n  groupByItems: any[];\n  payload: any;\n}\n\n/** @hidden */\nexport class GroupByValueEndpointComponent implements ExecutionContext {\n  private readonly aggregators: Map<string, Aggregator> = new Map();\n  private readonly aggregateResultArray: any[] = [];\n  private aggregateType: AggregateType;\n  private completed: boolean = false;\n\n  constructor(\n    private executionContext: ExecutionContext,\n    private queryInfo: QueryInfo,\n  ) {\n    // VALUE queries will only every have a single grouping\n    this.aggregateType = this.queryInfo.aggregates[0];\n  }\n\n  public hasMoreResults(): boolean {\n    return this.executionContext.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    if (this.completed) {\n      return {\n        result: undefined,\n        headers: getInitialHeader(),\n      };\n    }\n    const aggregateHeaders = getInitialHeader();\n    const response = await this.executionContext.fetchMore(diagnosticNode);\n    mergeHeaders(aggregateHeaders, response.headers);\n\n    if (response === undefined || response.result === undefined) {\n      if (this.aggregators.size > 0) {\n        return this.generateAggregateResponse(aggregateHeaders);\n      }\n      return { result: undefined, headers: aggregateHeaders };\n    }\n\n    for (const item of response.result as GroupByResult[]) {\n      if (item) {\n        let grouping: string = emptyGroup;\n        let payload: any = item;\n        if (item.groupByItems) {\n          // If the query contains a GROUP BY clause, it will have a payload property and groupByItems\n          payload = item.payload;\n          grouping = await hashObject(item.groupByItems);\n        }\n\n        const aggregator = this.aggregators.get(grouping);\n        if (!aggregator) {\n          // This is the first time we have seen a grouping so create a new aggregator\n          this.aggregators.set(grouping, createAggregator(this.aggregateType));\n        }\n\n        if (this.aggregateType) {\n          const aggregateResult = extractAggregateResult(payload[0]);\n          // if aggregate result is null, we need to short circuit aggregation and return undefined\n          if (aggregateResult === null) {\n            this.completed = true;\n          }\n          this.aggregators.get(grouping).aggregate(aggregateResult);\n        } else {\n          // Queries with no aggregates pass the payload directly to the aggregator\n          // Example: SELECT VALUE c.team FROM c GROUP BY c.team\n          this.aggregators.get(grouping).aggregate(payload);\n        }\n      }\n    }\n\n    // We bail early since we got an undefined result back `[{}]`\n    if (this.completed) {\n      return {\n        result: undefined,\n        headers: aggregateHeaders,\n      };\n    }\n\n    if (this.executionContext.hasMoreResults()) {\n      return { result: [], headers: aggregateHeaders };\n    } else {\n      // If no results are left in the underlying execution context, convert our aggregate results to an array\n      return this.generateAggregateResponse(aggregateHeaders);\n    }\n  }\n\n  private generateAggregateResponse(aggregateHeaders: CosmosHeaders): Response<any> {\n    for (const aggregator of this.aggregators.values()) {\n      const result = aggregator.getResult();\n      if (result !== undefined) {\n        this.aggregateResultArray.push(result);\n      }\n    }\n    this.completed = true;\n    return {\n      result: this.aggregateResultArray,\n      headers: aggregateHeaders,\n    };\n  }\n}\n"]}
+{"version":3,"file":"GroupByValueEndpointComponent.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.ts"],"names":[],"mappings":";;;AAMA,6DAAuD;AAEvD,sDAA2D;AAC3D,sDAAmE;AACnE,mDAAqE;AAGrE,sEAAsE;AAOtE,cAAc;AACd,MAAa,6BAA6B;IAO9B;IACA;IAPO,WAAW,GAA4B,IAAI,GAAG,EAAE,CAAC;IACjD,oBAAoB,GAAU,EAAE,CAAC;IAC1C,aAAa,CAAgB;IAC7B,SAAS,GAAY,KAAK,CAAC;IAEnC,YACU,gBAAkC,EAClC,SAAoB;QADpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,cAAS,GAAT,SAAS,CAAW;QAE5B,uDAAuD;QACvD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC;IAEM,cAAc;QACnB,OAAO,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAsC;QAC3D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAA,iCAAgB,GAAE;aAC5B,CAAC;QACJ,CAAC;QACD,MAAM,gBAAgB,GAAG,IAAA,iCAAgB,GAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAEvE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAC,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;YAC1D,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC1D,CAAC;QAED,IAAA,6BAAY,EAAC,gBAAgB,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;QAEjD,IACE,QAAQ,CAAC,MAAM,KAAK,SAAS;YAC7B,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC;YACtC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EACnC,CAAC;YACD,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC9B,OAAO,IAAI,CAAC,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;YAC1D,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC1D,CAAC;QAED,MAAM,cAAc,GAAG,QAAQ,CAAC,MAA6B,CAAC;QAC9D,MAAM,aAAa,GAAoB,cAAc,CAAC,MAAyB,CAAC;QAEhF,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,QAAQ,GAAW,0BAAU,CAAC;gBAClC,IAAI,OAAO,GAAQ,IAAI,CAAC;gBACxB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,4FAA4F;oBAC5F,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;oBACvB,QAAQ,GAAG,MAAM,IAAA,0BAAU,EAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBACjD,CAAC;gBAED,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAClD,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,4EAA4E;oBAC5E,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAA,2BAAgB,EAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;gBACvE,CAAC;gBAED,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACvB,MAAM,eAAe,GAAG,IAAA,sCAAsB,EAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC3D,yFAAyF;oBACzF,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;wBAC7B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;oBACxB,CAAC;oBACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,yEAAyE;oBACzE,sDAAsD;oBACtD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;QACH,CAAC;QAED,6DAA6D;QAC7D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,MAAM,GAAG,IAAA,kDAAyB,EAAC,EAAE,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YAExD,OAAO;gBACL,MAAM;gBACN,OAAO,EAAE,gBAAgB;aAC1B,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3C,yEAAyE;YACzE,MAAM,MAAM,GAAG,IAAA,kDAAyB,EACtC,EAAE,EAAE,eAAe;YACnB,IAAI,GAAG,EAAE,CACV,CAAC;YAEF,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,wGAAwG;YACxG,OAAO,IAAI,CAAC,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,yBAAyB,CAAC,gBAA+B;QAC/D,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YACnD,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,MAAM,MAAM,GAAG,IAAA,kDAAyB,EAAC,IAAI,CAAC,oBAAoB,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QAE/E,OAAO;YACL,MAAM;YACN,OAAO,EAAE,gBAAgB;SAC1B,CAAC;IACJ,CAAC;CACF;AA1HD,sEA0HC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { Response } from \"../../request/index.js\";\nimport type { ExecutionContext } from \"../ExecutionContext.js\";\nimport type { CosmosHeaders } from \"../CosmosHeaders.js\";\nimport type { AggregateType, QueryInfo } from \"../../request/ErrorResponse.js\";\nimport { hashObject } from \"../../utils/hashObject.js\";\nimport type { Aggregator } from \"../Aggregators/index.js\";\nimport { createAggregator } from \"../Aggregators/index.js\";\nimport { getInitialHeader, mergeHeaders } from \"../headerUtils.js\";\nimport { emptyGroup, extractAggregateResult } from \"./emptyGroup.js\";\nimport type { DiagnosticNodeInternal } from \"../../diagnostics/DiagnosticNodeInternal.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\nimport { createParallelQueryResult } from \"../parallelQueryResult.js\";\n\ninterface GroupByResult {\n  groupByItems: any[];\n  payload: any;\n}\n\n/** @hidden */\nexport class GroupByValueEndpointComponent implements ExecutionContext {\n  private readonly aggregators: Map<string, Aggregator> = new Map();\n  private readonly aggregateResultArray: any[] = [];\n  private aggregateType: AggregateType;\n  private completed: boolean = false;\n\n  constructor(\n    private executionContext: ExecutionContext,\n    private queryInfo: QueryInfo,\n  ) {\n    // VALUE queries will only every have a single grouping\n    this.aggregateType = this.queryInfo.aggregates[0];\n  }\n\n  public hasMoreResults(): boolean {\n    return this.executionContext.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>> {\n    if (this.completed) {\n      return {\n        result: undefined,\n        headers: getInitialHeader(),\n      };\n    }\n    const aggregateHeaders = getInitialHeader();\n    const response = await this.executionContext.fetchMore(diagnosticNode);\n\n    if (!response) {\n      if (this.aggregators.size > 0) {\n        return this.generateAggregateResponse(aggregateHeaders);\n      }\n      return { result: undefined, headers: aggregateHeaders };\n    }\n\n    mergeHeaders(aggregateHeaders, response.headers);\n\n    if (\n      response.result === undefined ||\n      !Array.isArray(response.result.buffer) ||\n      response.result.buffer.length === 0\n    ) {\n      if (this.aggregators.size > 0) {\n        return this.generateAggregateResponse(aggregateHeaders);\n      }\n      return { result: undefined, headers: aggregateHeaders };\n    }\n\n    const parallelResult = response.result as ParallelQueryResult;\n    const dataToProcess: GroupByResult[] = parallelResult.buffer as GroupByResult[];\n\n    for (const item of dataToProcess) {\n      if (item) {\n        let grouping: string = emptyGroup;\n        let payload: any = item;\n        if (item.groupByItems) {\n          // If the query contains a GROUP BY clause, it will have a payload property and groupByItems\n          payload = item.payload;\n          grouping = await hashObject(item.groupByItems);\n        }\n\n        const aggregator = this.aggregators.get(grouping);\n        if (!aggregator) {\n          // This is the first time we have seen a grouping so create a new aggregator\n          this.aggregators.set(grouping, createAggregator(this.aggregateType));\n        }\n\n        if (this.aggregateType) {\n          const aggregateResult = extractAggregateResult(payload[0]);\n          // if aggregate result is null, we need to short circuit aggregation and return undefined\n          if (aggregateResult === null) {\n            this.completed = true;\n          }\n          this.aggregators.get(grouping).aggregate(aggregateResult);\n        } else {\n          // Queries with no aggregates pass the payload directly to the aggregator\n          // Example: SELECT VALUE c.team FROM c GROUP BY c.team\n          this.aggregators.get(grouping).aggregate(payload);\n        }\n      }\n    }\n\n    // We bail early since we got an undefined result back `[{}]`\n    if (this.completed) {\n      const result = createParallelQueryResult([], new Map());\n\n      return {\n        result,\n        headers: aggregateHeaders,\n      };\n    }\n\n    if (this.executionContext.hasMoreResults()) {\n      // Return empty buffer but preserve the structure and pass-through fields\n      const result = createParallelQueryResult(\n        [], // empty buffer\n        new Map(),\n      );\n\n      return { result, headers: aggregateHeaders };\n    } else {\n      // If no results are left in the underlying execution context, convert our aggregate results to an array\n      return this.generateAggregateResponse(aggregateHeaders);\n    }\n  }\n\n  private generateAggregateResponse(aggregateHeaders: CosmosHeaders): Response<any> {\n    for (const aggregator of this.aggregators.values()) {\n      const result = aggregator.getResult();\n      if (result !== undefined) {\n        this.aggregateResultArray.push(result);\n      }\n    }\n    this.completed = true;\n\n    const result = createParallelQueryResult(this.aggregateResultArray, new Map());\n\n    return {\n      result,\n      headers: aggregateHeaders,\n    };\n  }\n}\n"]}

package/dist/commonjs/queryExecutionContext/EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"NonStreamingOrderByDistinctEndpointComponent.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6CAA6C,CAAC;AAO1F;;;GAGG;AACH,qBAAa,4CAA6C,YAAW,gBAAgB;IAqBjF,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,uBAAuB;IAC/B,OAAO,CAAC,qBAAqB;IAvB/B;;OAEG;IACH,OAAO,CAAC,YAAY,CAAoD;IACxE;;OAEG;IACH,OAAO,CAAC,qBAAqB,CAAoD;IACjF;;OAEG;IACH,OAAO,CAAC,gBAAgB,CAA8B;IAEtD,OAAO,CAAC,UAAU,CAAW;IAC7B;;OAEG;IACH,OAAO,CAAC,WAAW,CAAkB;gBAG3B,gBAAgB,EAAE,gBAAgB,EAClC,SAAS,EAAE,SAAS,EACpB,uBAAuB,EAAE,MAAM,EAC/B,qBAAqB,GAAE,OAAe;IAiBhD;;OAEG;YACW,qBAAqB;IA2B5B,cAAc,IAAI,OAAO;IAKnB,SAAS,CAAC,cAAc,CAAC,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;CA+DxF"}
+{"version":3,"file":"NonStreamingOrderByDistinctEndpointComponent.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,6CAA6C,CAAC;AAS1F;;;GAGG;AACH,qBAAa,4CAA6C,YAAW,gBAAgB;IAqBjF,OAAO,CAAC,gBAAgB;IACxB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,uBAAuB;IAC/B,OAAO,CAAC,qBAAqB;IAvB/B;;OAEG;IACH,OAAO,CAAC,YAAY,CAAoD;IACxE;;OAEG;IACH,OAAO,CAAC,qBAAqB,CAAoD;IACjF;;OAEG;IACH,OAAO,CAAC,gBAAgB,CAA8B;IAEtD,OAAO,CAAC,UAAU,CAAW;IAC7B;;OAEG;IACH,OAAO,CAAC,WAAW,CAAkB;gBAG3B,gBAAgB,EAAE,gBAAgB,EAClC,SAAS,EAAE,SAAS,EACpB,uBAAuB,EAAE,MAAM,EAC/B,qBAAqB,GAAE,OAAe;IAiBhD;;OAEG;YACW,qBAAqB;IA2B5B,cAAc,IAAI,OAAO;IAKnB,SAAS,CAAC,cAAc,CAAC,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;CAoGxF"}

package/dist/commonjs/queryExecutionContext/EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.js

@@ -6,6 +6,7 @@ const hashObject_js_1 = require("../../utils/hashObject.js");
 const fixedSizePriorityQueue_js_1 = require("../../utils/fixedSizePriorityQueue.js");
 const nonStreamingOrderByMap_js_1 = require("../../utils/nonStreamingOrderByMap.js");
 const orderByComparator_js_1 = require("../orderByComparator.js");
+const parallelQueryResult_js_1 = require("../parallelQueryResult.js");
 /**
  * @hidden
  * Represents an endpoint in handling an non-streaming order by distinct query.
@@ -99,19 +100,36 @@ class NonStreamingOrderByDistinctEndpointComponent {
         if (this.executionContext.hasMoreResults()) {
             // Grab the next result
             const response = await this.executionContext.fetchMore(diagnosticNode);
-            if (response === undefined || response.result === undefined) {
+            if (!response) {
                 this.isCompleted = true;
                 if (this.aggregateMap.size() > 0) {
                     await this.buildFinalResultArray();
+                    const result = (0, parallelQueryResult_js_1.createParallelQueryResult)(this.finalResultArray, new Map(), {}, undefined);
                     return {
-                        result: this.finalResultArray,
+                        result,
+                        headers: resHeaders,
+                    };
+                }
+                return { result: undefined, headers: resHeaders };
+            }
+            if (response.result === undefined ||
+                !Array.isArray(response.result.buffer) ||
+                response.result.buffer.length === 0) {
+                this.isCompleted = true;
+                if (this.aggregateMap.size() > 0) {
+                    await this.buildFinalResultArray();
+                    const result = (0, parallelQueryResult_js_1.createParallelQueryResult)(this.finalResultArray, new Map(), {}, undefined);
+                    return {
+                        result,
                         headers: response.headers,
                     };
                 }
                 return { result: undefined, headers: response.headers };
             }
             resHeaders = response.headers;
-            for (const item of response.result) {
+            const parallelResult = response.result;
+            const dataToProcess = parallelResult.buffer;
+            for (const item of dataToProcess) {
                 if (item) {
                     const key = await (0, hashObject_js_1.hashObject)(item?.payload);
                     this.aggregateMap.set(key, item);
@@ -119,8 +137,10 @@ class NonStreamingOrderByDistinctEndpointComponent {
             }
             // return [] to signal that there are more results to fetch.
             if (this.executionContext.hasMoreResults()) {
+                const result = (0, parallelQueryResult_js_1.createParallelQueryResult)([], // empty buffer
+                new Map(), undefined, undefined);
                 return {
-                    result: [],
+                    result,
                     headers: resHeaders,
                 };
             }
@@ -129,14 +149,16 @@ class NonStreamingOrderByDistinctEndpointComponent {
         if (!this.executionContext.hasMoreResults() && !this.isCompleted) {
             this.isCompleted = true;
             await this.buildFinalResultArray();
+            const result = (0, parallelQueryResult_js_1.createParallelQueryResult)(this.finalResultArray, new Map());
             return {
-                result: this.finalResultArray,
+                result,
                 headers: resHeaders,
             };
         }
         // Signal that there are no more results.
+        const result = (0, parallelQueryResult_js_1.createParallelQueryResult)([], new Map());
         return {
-            result: undefined,
+            result,
             headers: resHeaders,
         };
     }