@azure/cosmos 4.9.1 → 4.9.2

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

@@ -1,115 +1,118 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GroupByEndpointComponent = void 0;
-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 */
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var GroupByEndpointComponent_exports = {};
+__export(GroupByEndpointComponent_exports, {
+  GroupByEndpointComponent: () => GroupByEndpointComponent
+});
+module.exports = __toCommonJS(GroupByEndpointComponent_exports);
+var import_hashObject = require("../../utils/hashObject.js");
+var import_Aggregators = require("../Aggregators/index.js");
+var import_headerUtils = require("../headerUtils.js");
+var import_emptyGroup = require("./emptyGroup.js");
+var import_parallelQueryResult = require("../parallelQueryResult.js");
 class GroupByEndpointComponent {
-    executionContext;
-    queryInfo;
-    constructor(executionContext, queryInfo) {
-        this.executionContext = executionContext;
-        this.queryInfo = queryInfo;
+  constructor(executionContext, queryInfo) {
+    this.executionContext = executionContext;
+    this.queryInfo = queryInfo;
+  }
+  groupings = /* @__PURE__ */ new Map();
+  aggregateResultArray = [];
+  completed = false;
+  hasMoreResults() {
+    return this.executionContext.hasMoreResults();
+  }
+  async fetchMore(diagnosticNode) {
+    if (this.completed) {
+      return {
+        result: void 0,
+        headers: (0, import_headerUtils.getInitialHeader)()
+      };
     }
-    groupings = new Map();
-    aggregateResultArray = [];
-    completed = false;
-    hasMoreResults() {
-        return this.executionContext.hasMoreResults();
+    const aggregateHeaders = (0, import_headerUtils.getInitialHeader)();
+    const response = await this.executionContext.fetchMore(diagnosticNode);
+    if (!response) {
+      if (this.groupings.size > 0) {
+        return this.consolidateGroupResults(aggregateHeaders);
+      }
+      return { result: void 0, headers: aggregateHeaders };
     }
-    async fetchMore(diagnosticNode) {
-        if (this.completed) {
-            return {
-                result: undefined,
-                headers: (0, headerUtils_js_1.getInitialHeader)(),
-            };
-        }
-        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.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 };
-        }
-        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;
-                const aggregators = this.groupings.get(group);
-                const payload = item.payload;
-                if (aggregators) {
-                    // Iterator over all results in the payload
-                    for (const key of Object.keys(payload)) {
-                        // in case the value of a group is null make sure we create a dummy payload with item2==null
-                        const effectiveGroupByValue = payload[key]
-                            ? payload[key]
-                            : new Map().set("item2", null);
-                        const aggregateResult = (0, emptyGroup_js_1.extractAggregateResult)(effectiveGroupByValue);
-                        aggregators.get(key).aggregate(aggregateResult);
-                    }
-                }
-                else {
-                    // This is the first time we have seen a grouping. Setup the initial result without aggregate values
-                    const grouping = new Map();
-                    this.groupings.set(group, grouping);
-                    // Iterator over all results in the payload
-                    for (const key of Object.keys(payload)) {
-                        const aggregateType = this.queryInfo.groupByAliasToAggregateType[key];
-                        // Create a new aggregator for this specific aggregate field
-                        const aggregator = (0, index_js_1.createAggregator)(aggregateType);
-                        grouping.set(key, aggregator);
-                        if (aggregateType) {
-                            const aggregateResult = (0, emptyGroup_js_1.extractAggregateResult)(payload[key]);
-                            aggregator.aggregate(aggregateResult);
-                        }
-                        else {
-                            aggregator.aggregate(payload[key]);
-                        }
-                    }
-                }
-            }
-        }
-        if (this.executionContext.hasMoreResults()) {
-            // 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);
-        }
+    (0, import_headerUtils.mergeHeaders)(aggregateHeaders, response.headers);
+    if (response.result === void 0 || !Array.isArray(response.result.buffer) || response.result.buffer.length === 0) {
+      if (this.groupings.size > 0) {
+        return this.consolidateGroupResults(aggregateHeaders);
+      }
+      return { result: void 0, headers: aggregateHeaders };
     }
-    consolidateGroupResults(aggregateHeaders) {
-        for (const grouping of this.groupings.values()) {
-            const groupResult = {};
-            for (const [aggregateKey, aggregator] of grouping.entries()) {
-                groupResult[aggregateKey] = aggregator.getResult();
+    const parallelResult = response.result;
+    const dataToProcess = parallelResult.buffer;
+    for (const item of dataToProcess) {
+      if (item) {
+        const group = item.groupByItems ? await (0, import_hashObject.hashObject)(item.groupByItems) : import_emptyGroup.emptyGroup;
+        const aggregators = this.groupings.get(group);
+        const payload = item.payload;
+        if (aggregators) {
+          for (const key of Object.keys(payload)) {
+            const effectiveGroupByValue = payload[key] ? payload[key] : (/* @__PURE__ */ new Map()).set("item2", null);
+            const aggregateResult = (0, import_emptyGroup.extractAggregateResult)(effectiveGroupByValue);
+            aggregators.get(key).aggregate(aggregateResult);
+          }
+        } else {
+          const grouping = /* @__PURE__ */ new Map();
+          this.groupings.set(group, grouping);
+          for (const key of Object.keys(payload)) {
+            const aggregateType = this.queryInfo.groupByAliasToAggregateType[key];
+            const aggregator = (0, import_Aggregators.createAggregator)(aggregateType);
+            grouping.set(key, aggregator);
+            if (aggregateType) {
+              const aggregateResult = (0, import_emptyGroup.extractAggregateResult)(payload[key]);
+              aggregator.aggregate(aggregateResult);
+            } else {
+              aggregator.aggregate(payload[key]);
             }
-            this.aggregateResultArray.push(groupResult);
+          }
         }
-        this.completed = true;
-        // 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 };
+      }
+    }
+    if (this.executionContext.hasMoreResults()) {
+      const result = (0, import_parallelQueryResult.createParallelQueryResult)(
+        [],
+        // empty buffer
+        /* @__PURE__ */ new Map()
+      );
+      return { result, headers: aggregateHeaders };
+    } else {
+      return this.consolidateGroupResults(aggregateHeaders);
+    }
+  }
+  consolidateGroupResults(aggregateHeaders) {
+    for (const grouping of this.groupings.values()) {
+      const groupResult = {};
+      for (const [aggregateKey, aggregator] of grouping.entries()) {
+        groupResult[aggregateKey] = aggregator.getResult();
+      }
+      this.aggregateResultArray.push(groupResult);
     }
+    this.completed = true;
+    const result = (0, import_parallelQueryResult.createParallelQueryResult)(this.aggregateResultArray, /* @__PURE__ */ new Map());
+    return { result, headers: aggregateHeaders };
+  }
 }
-exports.GroupByEndpointComponent = GroupByEndpointComponent;
-//# sourceMappingURL=GroupByEndpointComponent.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GroupByEndpointComponent
+});

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

@@ -1 +1,7 @@
-{"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"]}
+{
+  "version": 3,
+  "sources": ["/mnt/vss/_work/1/s/sdk/cosmosdb/cosmos/src/queryExecutionContext/EndpointComponent/GroupByEndpointComponent.ts"],
+  "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"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAMA,wBAA2B;AAE3B,yBAAiC;AACjC,yBAA+C;AAC/C,wBAAmD;AAGnD,iCAA0C;AAQnC,MAAM,yBAAqD;AAAA,EAChE,YACU,kBACA,WACR;AAFQ;AACA;AAAA,EACP;AAAA,EAEc,YAAkD,oBAAI,IAAI;AAAA,EAC1D,uBAA8B,CAAC;AAAA,EACxC,YAAqB;AAAA,EAEtB,iBAA0B;AAC/B,WAAO,KAAK,iBAAiB,eAAe;AAAA,EAC9C;AAAA,EAEA,MAAa,UAAU,gBAAgE;AACrF,QAAI,KAAK,WAAW;AAClB,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,aAAS,qCAAiB;AAAA,MAC5B;AAAA,IACF;AACA,UAAM,uBAAmB,qCAAiB;AAC1C,UAAM,WAAW,MAAM,KAAK,iBAAiB,UAAU,cAAc;AAErE,QAAI,CAAC,UAAU;AAEb,UAAI,KAAK,UAAU,OAAO,GAAG;AAC3B,eAAO,KAAK,wBAAwB,gBAAgB;AAAA,MACtD;AACA,aAAO,EAAE,QAAQ,QAAW,SAAS,iBAAiB;AAAA,IACxD;AAEA,yCAAa,kBAAkB,SAAS,OAAO;AAE/C,QACE,SAAS,WAAW,UACpB,CAAC,MAAM,QAAQ,SAAS,OAAO,MAAM,KACrC,SAAS,OAAO,OAAO,WAAW,GAClC;AAEA,UAAI,KAAK,UAAU,OAAO,GAAG;AAC3B,eAAO,KAAK,wBAAwB,gBAAgB;AAAA,MACtD;AACA,aAAO,EAAE,QAAQ,QAAW,SAAS,iBAAiB;AAAA,IACxD;AAEA,UAAM,iBAAiB,SAAS;AAChC,UAAM,gBAAiC,eAAe;AAGtD,eAAW,QAAQ,eAAe;AAEhC,UAAI,MAAM;AACR,cAAM,QAAQ,KAAK,eAAe,UAAM,8BAAW,KAAK,YAAY,IAAI;AACxE,cAAM,cAAc,KAAK,UAAU,IAAI,KAAK;AAC5C,cAAM,UAAU,KAAK;AACrB,YAAI,aAAa;AAEf,qBAAW,OAAO,OAAO,KAAK,OAAO,GAAG;AAEtC,kBAAM,wBAAwB,QAAQ,GAAG,IACrC,QAAQ,GAAG,KACX,oBAAI,IAAI,GAAE,IAAI,SAAS,IAAI;AAC/B,kBAAM,sBAAkB,0CAAuB,qBAAqB;AACpE,wBAAY,IAAI,GAAG,EAAE,UAAU,eAAe;AAAA,UAChD;AAAA,QACF,OAAO;AAEL,gBAAM,WAAW,oBAAI,IAAI;AACzB,eAAK,UAAU,IAAI,OAAO,QAAQ;AAElC,qBAAW,OAAO,OAAO,KAAK,OAAO,GAAG;AACtC,kBAAM,gBAAgB,KAAK,UAAU,4BAA4B,GAAG;AAEpE,kBAAM,iBAAa,qCAAiB,aAAa;AACjD,qBAAS,IAAI,KAAK,UAAU;AAC5B,gBAAI,eAAe;AACjB,oBAAM,sBAAkB,0CAAuB,QAAQ,GAAG,CAAC;AAC3D,yBAAW,UAAU,eAAe;AAAA,YACtC,OAAO;AACL,yBAAW,UAAU,QAAQ,GAAG,CAAC;AAAA,YACnC;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,QAAI,KAAK,iBAAiB,eAAe,GAAG;AAE1C,YAAM,aAAS;AAAA,QACb,CAAC;AAAA;AAAA,QACD,oBAAI,IAAI;AAAA,MACV;AAEA,aAAO,EAAE,QAAQ,SAAS,iBAAiB;AAAA,IAC7C,OAAO;AACL,aAAO,KAAK,wBAAwB,gBAAgB;AAAA,IACtD;AAAA,EACF;AAAA,EAEQ,wBAAwB,kBAAgD;AAC9E,eAAW,YAAY,KAAK,UAAU,OAAO,GAAG;AAC9C,YAAM,cAAmB,CAAC;AAC1B,iBAAW,CAAC,cAAc,UAAU,KAAK,SAAS,QAAQ,GAAG;AAC3D,oBAAY,YAAY,IAAI,WAAW,UAAU;AAAA,MACnD;AACA,WAAK,qBAAqB,KAAK,WAAW;AAAA,IAC5C;AACA,SAAK,YAAY;AAGjB,UAAM,aAAS,sDAA0B,KAAK,sBAAsB,oBAAI,IAAI,CAAC;AAE7E,WAAO,EAAE,QAAQ,SAAS,iBAAiB;AAAA,EAC7C;AACF;",
+  "names": []
+}

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

@@ -1,116 +1,124 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GroupByValueEndpointComponent = void 0;
-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 */
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var GroupByValueEndpointComponent_exports = {};
+__export(GroupByValueEndpointComponent_exports, {
+  GroupByValueEndpointComponent: () => GroupByValueEndpointComponent
+});
+module.exports = __toCommonJS(GroupByValueEndpointComponent_exports);
+var import_hashObject = require("../../utils/hashObject.js");
+var import_Aggregators = require("../Aggregators/index.js");
+var import_headerUtils = require("../headerUtils.js");
+var import_emptyGroup = require("./emptyGroup.js");
+var import_parallelQueryResult = require("../parallelQueryResult.js");
 class GroupByValueEndpointComponent {
-    executionContext;
-    queryInfo;
-    aggregators = new Map();
-    aggregateResultArray = [];
-    aggregateType;
-    completed = false;
-    constructor(executionContext, queryInfo) {
-        this.executionContext = executionContext;
-        this.queryInfo = queryInfo;
-        // VALUE queries will only every have a single grouping
-        this.aggregateType = this.queryInfo.aggregates[0];
+  constructor(executionContext, queryInfo) {
+    this.executionContext = executionContext;
+    this.queryInfo = queryInfo;
+    this.aggregateType = this.queryInfo.aggregates[0];
+  }
+  aggregators = /* @__PURE__ */ new Map();
+  aggregateResultArray = [];
+  aggregateType;
+  completed = false;
+  hasMoreResults() {
+    return this.executionContext.hasMoreResults();
+  }
+  async fetchMore(diagnosticNode) {
+    if (this.completed) {
+      return {
+        result: void 0,
+        headers: (0, import_headerUtils.getInitialHeader)()
+      };
     }
-    hasMoreResults() {
-        return this.executionContext.hasMoreResults();
+    const aggregateHeaders = (0, import_headerUtils.getInitialHeader)();
+    const response = await this.executionContext.fetchMore(diagnosticNode);
+    if (!response) {
+      if (this.aggregators.size > 0) {
+        return this.generateAggregateResponse(aggregateHeaders);
+      }
+      return { result: void 0, headers: aggregateHeaders };
     }
-    async fetchMore(diagnosticNode) {
-        if (this.completed) {
-            return {
-                result: undefined,
-                headers: (0, headerUtils_js_1.getInitialHeader)(),
-            };
-        }
-        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.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 };
-        }
-        const parallelResult = response.result;
-        const dataToProcess = parallelResult.buffer;
-        for (const item of dataToProcess) {
-            if (item) {
-                let grouping = emptyGroup_js_1.emptyGroup;
-                let payload = item;
-                if (item.groupByItems) {
-                    // If the query contains a GROUP BY clause, it will have a payload property and groupByItems
-                    payload = item.payload;
-                    grouping = await (0, hashObject_js_1.hashObject)(item.groupByItems);
-                }
-                const aggregator = this.aggregators.get(grouping);
-                if (!aggregator) {
-                    // This is the first time we have seen a grouping so create a new aggregator
-                    this.aggregators.set(grouping, (0, index_js_1.createAggregator)(this.aggregateType));
-                }
-                if (this.aggregateType) {
-                    const aggregateResult = (0, emptyGroup_js_1.extractAggregateResult)(payload[0]);
-                    // if aggregate result is null, we need to short circuit aggregation and return undefined
-                    if (aggregateResult === null) {
-                        this.completed = true;
-                    }
-                    this.aggregators.get(grouping).aggregate(aggregateResult);
-                }
-                else {
-                    // Queries with no aggregates pass the payload directly to the aggregator
-                    // Example: SELECT VALUE c.team FROM c GROUP BY c.team
-                    this.aggregators.get(grouping).aggregate(payload);
-                }
-            }
-        }
-        // We bail early since we got an undefined result back `[{}]`
-        if (this.completed) {
-            const result = (0, parallelQueryResult_js_1.createParallelQueryResult)([], new Map());
-            return {
-                result,
-                headers: aggregateHeaders,
-            };
+    (0, import_headerUtils.mergeHeaders)(aggregateHeaders, response.headers);
+    if (response.result === void 0 || !Array.isArray(response.result.buffer) || response.result.buffer.length === 0) {
+      if (this.aggregators.size > 0) {
+        return this.generateAggregateResponse(aggregateHeaders);
+      }
+      return { result: void 0, headers: aggregateHeaders };
+    }
+    const parallelResult = response.result;
+    const dataToProcess = parallelResult.buffer;
+    for (const item of dataToProcess) {
+      if (item) {
+        let grouping = import_emptyGroup.emptyGroup;
+        let payload = item;
+        if (item.groupByItems) {
+          payload = item.payload;
+          grouping = await (0, import_hashObject.hashObject)(item.groupByItems);
         }
-        if (this.executionContext.hasMoreResults()) {
-            // 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 };
+        const aggregator = this.aggregators.get(grouping);
+        if (!aggregator) {
+          this.aggregators.set(grouping, (0, import_Aggregators.createAggregator)(this.aggregateType));
         }
-        else {
-            // If no results are left in the underlying execution context, convert our aggregate results to an array
-            return this.generateAggregateResponse(aggregateHeaders);
+        if (this.aggregateType) {
+          const aggregateResult = (0, import_emptyGroup.extractAggregateResult)(payload[0]);
+          if (aggregateResult === null) {
+            this.completed = true;
+          }
+          this.aggregators.get(grouping).aggregate(aggregateResult);
+        } else {
+          this.aggregators.get(grouping).aggregate(payload);
         }
+      }
     }
-    generateAggregateResponse(aggregateHeaders) {
-        for (const aggregator of this.aggregators.values()) {
-            const result = aggregator.getResult();
-            if (result !== undefined) {
-                this.aggregateResultArray.push(result);
-            }
-        }
-        this.completed = true;
-        const result = (0, parallelQueryResult_js_1.createParallelQueryResult)(this.aggregateResultArray, new Map());
-        return {
-            result,
-            headers: aggregateHeaders,
-        };
+    if (this.completed) {
+      const result = (0, import_parallelQueryResult.createParallelQueryResult)([], /* @__PURE__ */ new Map());
+      return {
+        result,
+        headers: aggregateHeaders
+      };
+    }
+    if (this.executionContext.hasMoreResults()) {
+      const result = (0, import_parallelQueryResult.createParallelQueryResult)(
+        [],
+        // empty buffer
+        /* @__PURE__ */ new Map()
+      );
+      return { result, headers: aggregateHeaders };
+    } else {
+      return this.generateAggregateResponse(aggregateHeaders);
+    }
+  }
+  generateAggregateResponse(aggregateHeaders) {
+    for (const aggregator of this.aggregators.values()) {
+      const result2 = aggregator.getResult();
+      if (result2 !== void 0) {
+        this.aggregateResultArray.push(result2);
+      }
     }
+    this.completed = true;
+    const result = (0, import_parallelQueryResult.createParallelQueryResult)(this.aggregateResultArray, /* @__PURE__ */ new Map());
+    return {
+      result,
+      headers: aggregateHeaders
+    };
+  }
 }
-exports.GroupByValueEndpointComponent = GroupByValueEndpointComponent;
-//# sourceMappingURL=GroupByValueEndpointComponent.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  GroupByValueEndpointComponent
+});

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

@@ -1 +1,7 @@
-{"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"]}
+{
+  "version": 3,
+  "sources": ["/mnt/vss/_work/1/s/sdk/cosmosdb/cosmos/src/queryExecutionContext/EndpointComponent/GroupByValueEndpointComponent.ts"],
+  "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"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAMA,wBAA2B;AAE3B,yBAAiC;AACjC,yBAA+C;AAC/C,wBAAmD;AAGnD,iCAA0C;AAQnC,MAAM,8BAA0D;AAAA,EAMrE,YACU,kBACA,WACR;AAFQ;AACA;AAGR,SAAK,gBAAgB,KAAK,UAAU,WAAW,CAAC;AAAA,EAClD;AAAA,EAXiB,cAAuC,oBAAI,IAAI;AAAA,EAC/C,uBAA8B,CAAC;AAAA,EACxC;AAAA,EACA,YAAqB;AAAA,EAUtB,iBAA0B;AAC/B,WAAO,KAAK,iBAAiB,eAAe;AAAA,EAC9C;AAAA,EAEA,MAAa,UAAU,gBAAgE;AACrF,QAAI,KAAK,WAAW;AAClB,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,aAAS,qCAAiB;AAAA,MAC5B;AAAA,IACF;AACA,UAAM,uBAAmB,qCAAiB;AAC1C,UAAM,WAAW,MAAM,KAAK,iBAAiB,UAAU,cAAc;AAErE,QAAI,CAAC,UAAU;AACb,UAAI,KAAK,YAAY,OAAO,GAAG;AAC7B,eAAO,KAAK,0BAA0B,gBAAgB;AAAA,MACxD;AACA,aAAO,EAAE,QAAQ,QAAW,SAAS,iBAAiB;AAAA,IACxD;AAEA,yCAAa,kBAAkB,SAAS,OAAO;AAE/C,QACE,SAAS,WAAW,UACpB,CAAC,MAAM,QAAQ,SAAS,OAAO,MAAM,KACrC,SAAS,OAAO,OAAO,WAAW,GAClC;AACA,UAAI,KAAK,YAAY,OAAO,GAAG;AAC7B,eAAO,KAAK,0BAA0B,gBAAgB;AAAA,MACxD;AACA,aAAO,EAAE,QAAQ,QAAW,SAAS,iBAAiB;AAAA,IACxD;AAEA,UAAM,iBAAiB,SAAS;AAChC,UAAM,gBAAiC,eAAe;AAEtD,eAAW,QAAQ,eAAe;AAChC,UAAI,MAAM;AACR,YAAI,WAAmB;AACvB,YAAI,UAAe;AACnB,YAAI,KAAK,cAAc;AAErB,oBAAU,KAAK;AACf,qBAAW,UAAM,8BAAW,KAAK,YAAY;AAAA,QAC/C;AAEA,cAAM,aAAa,KAAK,YAAY,IAAI,QAAQ;AAChD,YAAI,CAAC,YAAY;AAEf,eAAK,YAAY,IAAI,cAAU,qCAAiB,KAAK,aAAa,CAAC;AAAA,QACrE;AAEA,YAAI,KAAK,eAAe;AACtB,gBAAM,sBAAkB,0CAAuB,QAAQ,CAAC,CAAC;AAEzD,cAAI,oBAAoB,MAAM;AAC5B,iBAAK,YAAY;AAAA,UACnB;AACA,eAAK,YAAY,IAAI,QAAQ,EAAE,UAAU,eAAe;AAAA,QAC1D,OAAO;AAGL,eAAK,YAAY,IAAI,QAAQ,EAAE,UAAU,OAAO;AAAA,QAClD;AAAA,MACF;AAAA,IACF;AAGA,QAAI,KAAK,WAAW;AAClB,YAAM,aAAS,sDAA0B,CAAC,GAAG,oBAAI,IAAI,CAAC;AAEtD,aAAO;AAAA,QACL;AAAA,QACA,SAAS;AAAA,MACX;AAAA,IACF;AAEA,QAAI,KAAK,iBAAiB,eAAe,GAAG;AAE1C,YAAM,aAAS;AAAA,QACb,CAAC;AAAA;AAAA,QACD,oBAAI,IAAI;AAAA,MACV;AAEA,aAAO,EAAE,QAAQ,SAAS,iBAAiB;AAAA,IAC7C,OAAO;AAEL,aAAO,KAAK,0BAA0B,gBAAgB;AAAA,IACxD;AAAA,EACF;AAAA,EAEQ,0BAA0B,kBAAgD;AAChF,eAAW,cAAc,KAAK,YAAY,OAAO,GAAG;AAClD,YAAMA,UAAS,WAAW,UAAU;AACpC,UAAIA,YAAW,QAAW;AACxB,aAAK,qBAAqB,KAAKA,OAAM;AAAA,MACvC;AAAA,IACF;AACA,SAAK,YAAY;AAEjB,UAAM,aAAS,sDAA0B,KAAK,sBAAsB,oBAAI,IAAI,CAAC;AAE7E,WAAO;AAAA,MACL;AAAA,MACA,SAAS;AAAA,IACX;AAAA,EACF;AACF;",
+  "names": ["result"]
+}