@azure/cosmos 4.9.1 → 4.9.3

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

@@ -1,167 +1,181 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.NonStreamingOrderByDistinctEndpointComponent = void 0;
-const headerUtils_js_1 = require("../headerUtils.js");
-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.
- */
+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 NonStreamingOrderByDistinctEndpointComponent_exports = {};
+__export(NonStreamingOrderByDistinctEndpointComponent_exports, {
+  NonStreamingOrderByDistinctEndpointComponent: () => NonStreamingOrderByDistinctEndpointComponent
+});
+module.exports = __toCommonJS(NonStreamingOrderByDistinctEndpointComponent_exports);
+var import_headerUtils = require("../headerUtils.js");
+var import_hashObject = require("../../utils/hashObject.js");
+var import_fixedSizePriorityQueue = require("../../utils/fixedSizePriorityQueue.js");
+var import_nonStreamingOrderByMap = require("../../utils/nonStreamingOrderByMap.js");
+var import_orderByComparator = require("../orderByComparator.js");
+var import_parallelQueryResult = require("../parallelQueryResult.js");
 class NonStreamingOrderByDistinctEndpointComponent {
-    executionContext;
-    queryInfo;
-    priorityQueueBufferSize;
-    emitRawOrderByPayload;
-    /**
-     * A Map that holds the distinct values of the items before storing in priority queue.
-     */
-    aggregateMap;
-    /**
-     * A priority queue to compute the final sorted results.
-     */
-    nonStreamingOrderByPQ;
-    /**
-     * Array to store the final sorted results.
-     */
-    finalResultArray;
-    sortOrders;
-    /**
-     * Flag to determine if all results are fetched from backend and results can be returned.
-     */
-    isCompleted = false;
-    constructor(executionContext, queryInfo, priorityQueueBufferSize, emitRawOrderByPayload = false) {
-        this.executionContext = executionContext;
-        this.queryInfo = queryInfo;
-        this.priorityQueueBufferSize = priorityQueueBufferSize;
-        this.emitRawOrderByPayload = emitRawOrderByPayload;
-        this.sortOrders = this.queryInfo.orderBy;
-        const comparator = new orderByComparator_js_1.OrderByComparator(this.sortOrders);
-        this.aggregateMap = new nonStreamingOrderByMap_js_1.NonStreamingOrderByMap((a, b) => {
-            return comparator.compareItems(a, b);
-        });
-        this.nonStreamingOrderByPQ = new fixedSizePriorityQueue_js_1.FixedSizePriorityQueue((a, b) => {
-            return comparator.compareItems(b, a);
-        }, this.priorityQueueBufferSize);
+  constructor(executionContext, queryInfo, priorityQueueBufferSize, emitRawOrderByPayload = false) {
+    this.executionContext = executionContext;
+    this.queryInfo = queryInfo;
+    this.priorityQueueBufferSize = priorityQueueBufferSize;
+    this.emitRawOrderByPayload = emitRawOrderByPayload;
+    this.sortOrders = this.queryInfo.orderBy;
+    const comparator = new import_orderByComparator.OrderByComparator(this.sortOrders);
+    this.aggregateMap = new import_nonStreamingOrderByMap.NonStreamingOrderByMap(
+      (a, b) => {
+        return comparator.compareItems(a, b);
+      }
+    );
+    this.nonStreamingOrderByPQ = new import_fixedSizePriorityQueue.FixedSizePriorityQueue(
+      (a, b) => {
+        return comparator.compareItems(b, a);
+      },
+      this.priorityQueueBufferSize
+    );
+  }
+  executionContext;
+  queryInfo;
+  priorityQueueBufferSize;
+  emitRawOrderByPayload;
+  /**
+   * A Map that holds the distinct values of the items before storing in priority queue.
+   */
+  aggregateMap;
+  /**
+   * A priority queue to compute the final sorted results.
+   */
+  nonStreamingOrderByPQ;
+  /**
+   * Array to store the final sorted results.
+   */
+  finalResultArray;
+  sortOrders;
+  /**
+   * Flag to determine if all results are fetched from backend and results can be returned.
+   */
+  isCompleted = false;
+  /**
+   * Build final sorted result array from which responses will be served.
+   */
+  async buildFinalResultArray() {
+    const allValues = this.aggregateMap.getAllValuesAndReset();
+    for (const value of allValues) {
+      this.nonStreamingOrderByPQ.enqueue(value);
     }
-    /**
-     * Build final sorted result array from which responses will be served.
-     */
-    async buildFinalResultArray() {
-        // Fetch all distinct values from the map and store in priority queue.
-        const allValues = this.aggregateMap.getAllValuesAndReset();
-        for (const value of allValues) {
-            this.nonStreamingOrderByPQ.enqueue(value);
-        }
-        // Compute the final result array size based on offset and limit.
-        const offSet = this.queryInfo.offset ? this.queryInfo.offset : 0;
-        const queueSize = this.nonStreamingOrderByPQ.size();
-        const finalArraySize = queueSize - offSet;
-        if (finalArraySize <= 0) {
-            this.finalResultArray = [];
-        }
-        else {
-            this.finalResultArray = new Array(finalArraySize);
-            // Only keep the final result array size number of items in the final result array and discard the rest.
-            for (let count = finalArraySize - 1; count >= 0; count--) {
-                if (this.emitRawOrderByPayload) {
-                    this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue();
-                }
-                else {
-                    this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue()?.payload;
-                }
-            }
+    const offSet = this.queryInfo.offset ? this.queryInfo.offset : 0;
+    const queueSize = this.nonStreamingOrderByPQ.size();
+    const finalArraySize = queueSize - offSet;
+    if (finalArraySize <= 0) {
+      this.finalResultArray = [];
+    } else {
+      this.finalResultArray = new Array(finalArraySize);
+      for (let count = finalArraySize - 1; count >= 0; count--) {
+        if (this.emitRawOrderByPayload) {
+          this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue();
+        } else {
+          this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue()?.payload;
         }
+      }
     }
-    hasMoreResults() {
-        if (this.priorityQueueBufferSize === 0)
-            return false;
-        return this.executionContext.hasMoreResults();
+  }
+  hasMoreResults() {
+    if (this.priorityQueueBufferSize === 0) return false;
+    return this.executionContext.hasMoreResults();
+  }
+  async fetchMore(diagnosticNode) {
+    if (this.isCompleted) {
+      return {
+        result: void 0,
+        headers: (0, import_headerUtils.getInitialHeader)()
+      };
     }
-    async fetchMore(diagnosticNode) {
-        if (this.isCompleted) {
-            return {
-                result: undefined,
-                headers: (0, headerUtils_js_1.getInitialHeader)(),
-            };
-        }
-        let resHeaders = (0, headerUtils_js_1.getInitialHeader)();
-        // if size is 0, just return undefined to signal to more results. Valid if query is TOP 0 or LIMIT 0
-        if (this.priorityQueueBufferSize <= 0) {
-            return {
-                result: undefined,
-                headers: resHeaders,
-            };
+    let resHeaders = (0, import_headerUtils.getInitialHeader)();
+    if (this.priorityQueueBufferSize <= 0) {
+      return {
+        result: void 0,
+        headers: resHeaders
+      };
+    }
+    if (this.executionContext.hasMoreResults()) {
+      const response = await this.executionContext.fetchMore(diagnosticNode);
+      if (!response) {
+        this.isCompleted = true;
+        if (this.aggregateMap.size() > 0) {
+          await this.buildFinalResultArray();
+          const result2 = (0, import_parallelQueryResult.createParallelQueryResult)(this.finalResultArray, /* @__PURE__ */ new Map(), {}, void 0);
+          return {
+            result: result2,
+            headers: resHeaders
+          };
         }
-        // If there are more results in backend, keep filling map.
-        if (this.executionContext.hasMoreResults()) {
-            // Grab the next result
-            const response = await this.executionContext.fetchMore(diagnosticNode);
-            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,
-                        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;
-            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);
-                }
-            }
-            // 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,
-                    headers: resHeaders,
-                };
-            }
+        return { result: void 0, headers: resHeaders };
+      }
+      if (response.result === void 0 || !Array.isArray(response.result.buffer) || response.result.buffer.length === 0) {
+        this.isCompleted = true;
+        if (this.aggregateMap.size() > 0) {
+          await this.buildFinalResultArray();
+          const result2 = (0, import_parallelQueryResult.createParallelQueryResult)(this.finalResultArray, /* @__PURE__ */ new Map(), {}, void 0);
+          return {
+            result: result2,
+            headers: response.headers
+          };
         }
-        // If all results are fetched from backend, prepare final results
-        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,
-                headers: resHeaders,
-            };
+        return { result: void 0, headers: response.headers };
+      }
+      resHeaders = response.headers;
+      const parallelResult = response.result;
+      const dataToProcess = parallelResult.buffer;
+      for (const item of dataToProcess) {
+        if (item) {
+          const key = await (0, import_hashObject.hashObject)(item?.payload);
+          this.aggregateMap.set(key, item);
         }
-        // Signal that there are no more results.
-        const result = (0, parallelQueryResult_js_1.createParallelQueryResult)([], new Map());
+      }
+      if (this.executionContext.hasMoreResults()) {
+        const result2 = (0, import_parallelQueryResult.createParallelQueryResult)(
+          [],
+          // empty buffer
+          /* @__PURE__ */ new Map(),
+          void 0,
+          void 0
+        );
         return {
-            result,
-            headers: resHeaders,
+          result: result2,
+          headers: resHeaders
         };
+      }
+    }
+    if (!this.executionContext.hasMoreResults() && !this.isCompleted) {
+      this.isCompleted = true;
+      await this.buildFinalResultArray();
+      const result2 = (0, import_parallelQueryResult.createParallelQueryResult)(this.finalResultArray, /* @__PURE__ */ new Map());
+      return {
+        result: result2,
+        headers: resHeaders
+      };
     }
+    const result = (0, import_parallelQueryResult.createParallelQueryResult)([], /* @__PURE__ */ new Map());
+    return {
+      result,
+      headers: resHeaders
+    };
+  }
 }
-exports.NonStreamingOrderByDistinctEndpointComponent = NonStreamingOrderByDistinctEndpointComponent;
-//# sourceMappingURL=NonStreamingOrderByDistinctEndpointComponent.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  NonStreamingOrderByDistinctEndpointComponent
+});
+//# sourceMappingURL=NonStreamingOrderByDistinctEndpointComponent.js.map

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

@@ -1 +1,7 @@
-{"version":3,"file":"NonStreamingOrderByDistinctEndpointComponent.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.ts"],"names":[],"mappings":";;;AAIA,sDAAqD;AAErD,6DAAuD;AAEvD,qFAA+E;AAC/E,qFAA+E;AAC/E,kEAA4D;AAE5D,sEAAsE;AAEtE;;;GAGG;AACH,MAAa,4CAA4C;IAqB7C;IACA;IACA;IACA;IAvBV;;OAEG;IACK,YAAY,CAAoD;IACxE;;OAEG;IACK,qBAAqB,CAAoD;IACjF;;OAEG;IACK,gBAAgB,CAA8B;IAE9C,UAAU,CAAW;IAC7B;;OAEG;IACK,WAAW,GAAY,KAAK,CAAC;IAErC,YACU,gBAAkC,EAClC,SAAoB,EACpB,uBAA+B,EAC/B,wBAAiC,KAAK;QAHtC,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,cAAS,GAAT,SAAS,CAAW;QACpB,4BAAuB,GAAvB,uBAAuB,CAAQ;QAC/B,0BAAqB,GAArB,qBAAqB,CAAiB;QAE9C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;QACzC,MAAM,UAAU,GAAG,IAAI,wCAAiB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1D,IAAI,CAAC,YAAY,GAAG,IAAI,kDAAsB,CAC5C,CAAC,CAA4B,EAAE,CAA4B,EAAE,EAAE;YAC7D,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC,CACF,CAAC;QACF,IAAI,CAAC,qBAAqB,GAAG,IAAI,kDAAsB,CACrD,CAAC,CAA4B,EAAE,CAA4B,EAAE,EAAE;YAC7D,OAAO,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC,EACD,IAAI,CAAC,uBAAuB,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB;QACjC,sEAAsE;QACtE,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,oBAAoB,EAAE,CAAC;QAC3D,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;YAC9B,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC;QAED,iEAAiE;QACjE,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,SAAS,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,EAAE,CAAC;QACpD,MAAM,cAAc,GAAG,SAAS,GAAG,MAAM,CAAC;QAE1C,IAAI,cAAc,IAAI,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,GAAG,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;YAClD,wGAAwG;YACxG,KAAK,IAAI,KAAK,GAAG,cAAc,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;gBACzD,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAC/B,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAAC;gBACtE,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC;gBAC/E,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAEM,cAAc;QACnB,IAAI,IAAI,CAAC,uBAAuB,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACrD,OAAO,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,cAAuC;QAC5D,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAA,iCAAgB,GAAE;aAC5B,CAAC;QACJ,CAAC;QACD,IAAI,UAAU,GAAG,IAAA,iCAAgB,GAAE,CAAC;QACpC,oGAAoG;QACpG,IAAI,IAAI,CAAC,uBAAuB,IAAI,CAAC,EAAE,CAAC;YACtC,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,UAAU;aACpB,CAAC;QACJ,CAAC;QAED,0DAA0D;QAC1D,IAAI,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3C,uBAAuB;YACvB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAEvE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;gBACxB,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;oBACjC,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBACnC,MAAM,MAAM,GAAG,IAAA,kDAAyB,EAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,GAAG,EAAE,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;oBAE1F,OAAO;wBACL,MAAM;wBACN,OAAO,EAAE,UAAU;qBACpB,CAAC;gBACJ,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;YACpD,CAAC;YAED,IACE,QAAQ,CAAC,MAAM,KAAK,SAAS;gBAC7B,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC;gBACtC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EACnC,CAAC;gBACD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;gBACxB,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;oBACjC,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBACnC,MAAM,MAAM,GAAG,IAAA,kDAAyB,EAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,GAAG,EAAE,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;oBAE1F,OAAO;wBACL,MAAM;wBACN,OAAO,EAAE,QAAQ,CAAC,OAAO;qBAC1B,CAAC;gBACJ,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC1D,CAAC;YACD,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC;YAE9B,MAAM,cAAc,GAAG,QAAQ,CAAC,MAA6B,CAAC;YAC9D,MAAM,aAAa,GACjB,cAAc,CAAC,MAAqC,CAAC;YAEvD,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;gBACjC,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,GAAG,GAAG,MAAM,IAAA,0BAAU,EAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBAC5C,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;YAED,4DAA4D;YAC5D,IAAI,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,EAAE,CAAC;gBAC3C,MAAM,MAAM,GAAG,IAAA,kDAAyB,EACtC,EAAE,EAAE,eAAe;gBACnB,IAAI,GAAG,EAAE,EACT,SAAS,EACT,SAAS,CACV,CAAC;gBAEF,OAAO;oBACL,MAAM;oBACN,OAAO,EAAE,UAAU;iBACpB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,iEAAiE;QACjE,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjE,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,IAAI,CAAC,qBAAqB,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,IAAA,kDAAyB,EAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;YAE3E,OAAO;gBACL,MAAM;gBACN,OAAO,EAAE,UAAU;aACpB,CAAC;QACJ,CAAC;QACD,yCAAyC;QACzC,MAAM,MAAM,GAAG,IAAA,kDAAyB,EAAC,EAAE,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QAExD,OAAO;YACL,MAAM;YACN,OAAO,EAAE,UAAU;SACpB,CAAC;IACJ,CAAC;CACF;AAhLD,oGAgLC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { QueryInfo, Response } from \"../../request/index.js\";\nimport type { ExecutionContext } from \"../ExecutionContext.js\";\nimport { getInitialHeader } from \"../headerUtils.js\";\nimport type { DiagnosticNodeInternal } from \"../../diagnostics/DiagnosticNodeInternal.js\";\nimport { hashObject } from \"../../utils/hashObject.js\";\nimport type { NonStreamingOrderByResult } from \"../nonStreamingOrderByResult.js\";\nimport { FixedSizePriorityQueue } from \"../../utils/fixedSizePriorityQueue.js\";\nimport { NonStreamingOrderByMap } from \"../../utils/nonStreamingOrderByMap.js\";\nimport { OrderByComparator } from \"../orderByComparator.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\nimport { createParallelQueryResult } from \"../parallelQueryResult.js\";\n\n/**\n * @hidden\n * Represents an endpoint in handling an non-streaming order by distinct query.\n */\nexport class NonStreamingOrderByDistinctEndpointComponent implements ExecutionContext {\n  /**\n   * A Map that holds the distinct values of the items before storing in priority queue.\n   */\n  private aggregateMap: NonStreamingOrderByMap<NonStreamingOrderByResult>;\n  /**\n   * A priority queue to compute the final sorted results.\n   */\n  private nonStreamingOrderByPQ: FixedSizePriorityQueue<NonStreamingOrderByResult>;\n  /**\n   * Array to store the final sorted results.\n   */\n  private finalResultArray: NonStreamingOrderByResult[];\n\n  private sortOrders: string[];\n  /**\n   * Flag to determine if all results are fetched from backend and results can be returned.\n   */\n  private isCompleted: boolean = false;\n\n  constructor(\n    private executionContext: ExecutionContext,\n    private queryInfo: QueryInfo,\n    private priorityQueueBufferSize: number,\n    private emitRawOrderByPayload: boolean = false,\n  ) {\n    this.sortOrders = this.queryInfo.orderBy;\n    const comparator = new OrderByComparator(this.sortOrders);\n    this.aggregateMap = new NonStreamingOrderByMap<NonStreamingOrderByResult>(\n      (a: NonStreamingOrderByResult, b: NonStreamingOrderByResult) => {\n        return comparator.compareItems(a, b);\n      },\n    );\n    this.nonStreamingOrderByPQ = new FixedSizePriorityQueue<NonStreamingOrderByResult>(\n      (a: NonStreamingOrderByResult, b: NonStreamingOrderByResult) => {\n        return comparator.compareItems(b, a);\n      },\n      this.priorityQueueBufferSize,\n    );\n  }\n\n  /**\n   * Build final sorted result array from which responses will be served.\n   */\n  private async buildFinalResultArray(): Promise<void> {\n    // Fetch all distinct values from the map and store in priority queue.\n    const allValues = this.aggregateMap.getAllValuesAndReset();\n    for (const value of allValues) {\n      this.nonStreamingOrderByPQ.enqueue(value);\n    }\n\n    // Compute the final result array size based on offset and limit.\n    const offSet = this.queryInfo.offset ? this.queryInfo.offset : 0;\n    const queueSize = this.nonStreamingOrderByPQ.size();\n    const finalArraySize = queueSize - offSet;\n\n    if (finalArraySize <= 0) {\n      this.finalResultArray = [];\n    } else {\n      this.finalResultArray = new Array(finalArraySize);\n      // Only keep the final result array size number of items in the final result array and discard the rest.\n      for (let count = finalArraySize - 1; count >= 0; count--) {\n        if (this.emitRawOrderByPayload) {\n          this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue();\n        } else {\n          this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue()?.payload;\n        }\n      }\n    }\n  }\n\n  public hasMoreResults(): boolean {\n    if (this.priorityQueueBufferSize === 0) return false;\n    return this.executionContext.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode?: DiagnosticNodeInternal): Promise<Response<any>> {\n    if (this.isCompleted) {\n      return {\n        result: undefined,\n        headers: getInitialHeader(),\n      };\n    }\n    let resHeaders = getInitialHeader();\n    // if size is 0, just return undefined to signal to more results. Valid if query is TOP 0 or LIMIT 0\n    if (this.priorityQueueBufferSize <= 0) {\n      return {\n        result: undefined,\n        headers: resHeaders,\n      };\n    }\n\n    // If there are more results in backend, keep filling map.\n    if (this.executionContext.hasMoreResults()) {\n      // Grab the next result\n      const response = await this.executionContext.fetchMore(diagnosticNode);\n\n      if (!response) {\n        this.isCompleted = true;\n        if (this.aggregateMap.size() > 0) {\n          await this.buildFinalResultArray();\n          const result = createParallelQueryResult(this.finalResultArray, new Map(), {}, undefined);\n\n          return {\n            result,\n            headers: resHeaders,\n          };\n        }\n        return { result: undefined, headers: resHeaders };\n      }\n\n      if (\n        response.result === undefined ||\n        !Array.isArray(response.result.buffer) ||\n        response.result.buffer.length === 0\n      ) {\n        this.isCompleted = true;\n        if (this.aggregateMap.size() > 0) {\n          await this.buildFinalResultArray();\n          const result = createParallelQueryResult(this.finalResultArray, new Map(), {}, undefined);\n\n          return {\n            result,\n            headers: response.headers,\n          };\n        }\n        return { result: undefined, headers: response.headers };\n      }\n      resHeaders = response.headers;\n\n      const parallelResult = response.result as ParallelQueryResult;\n      const dataToProcess: NonStreamingOrderByResult[] =\n        parallelResult.buffer as NonStreamingOrderByResult[];\n\n      for (const item of dataToProcess) {\n        if (item) {\n          const key = await hashObject(item?.payload);\n          this.aggregateMap.set(key, item);\n        }\n      }\n\n      // return [] to signal that there are more results to fetch.\n      if (this.executionContext.hasMoreResults()) {\n        const result = createParallelQueryResult(\n          [], // empty buffer\n          new Map(),\n          undefined,\n          undefined,\n        );\n\n        return {\n          result,\n          headers: resHeaders,\n        };\n      }\n    }\n\n    // If all results are fetched from backend, prepare final results\n    if (!this.executionContext.hasMoreResults() && !this.isCompleted) {\n      this.isCompleted = true;\n      await this.buildFinalResultArray();\n      const result = createParallelQueryResult(this.finalResultArray, new Map());\n\n      return {\n        result,\n        headers: resHeaders,\n      };\n    }\n    // Signal that there are no more results.\n    const result = createParallelQueryResult([], new Map());\n\n    return {\n      result,\n      headers: resHeaders,\n    };\n  }\n}\n"]}
+{
+  "version": 3,
+  "sources": ["../../../../src/queryExecutionContext/EndpointComponent/NonStreamingOrderByDistinctEndpointComponent.ts"],
+  "sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { QueryInfo, Response } from \"../../request/index.js\";\nimport type { ExecutionContext } from \"../ExecutionContext.js\";\nimport { getInitialHeader } from \"../headerUtils.js\";\nimport type { DiagnosticNodeInternal } from \"../../diagnostics/DiagnosticNodeInternal.js\";\nimport { hashObject } from \"../../utils/hashObject.js\";\nimport type { NonStreamingOrderByResult } from \"../nonStreamingOrderByResult.js\";\nimport { FixedSizePriorityQueue } from \"../../utils/fixedSizePriorityQueue.js\";\nimport { NonStreamingOrderByMap } from \"../../utils/nonStreamingOrderByMap.js\";\nimport { OrderByComparator } from \"../orderByComparator.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\nimport { createParallelQueryResult } from \"../parallelQueryResult.js\";\n\n/**\n * @hidden\n * Represents an endpoint in handling an non-streaming order by distinct query.\n */\nexport class NonStreamingOrderByDistinctEndpointComponent implements ExecutionContext {\n  /**\n   * A Map that holds the distinct values of the items before storing in priority queue.\n   */\n  private aggregateMap: NonStreamingOrderByMap<NonStreamingOrderByResult>;\n  /**\n   * A priority queue to compute the final sorted results.\n   */\n  private nonStreamingOrderByPQ: FixedSizePriorityQueue<NonStreamingOrderByResult>;\n  /**\n   * Array to store the final sorted results.\n   */\n  private finalResultArray: NonStreamingOrderByResult[];\n\n  private sortOrders: string[];\n  /**\n   * Flag to determine if all results are fetched from backend and results can be returned.\n   */\n  private isCompleted: boolean = false;\n\n  constructor(\n    private executionContext: ExecutionContext,\n    private queryInfo: QueryInfo,\n    private priorityQueueBufferSize: number,\n    private emitRawOrderByPayload: boolean = false,\n  ) {\n    this.sortOrders = this.queryInfo.orderBy;\n    const comparator = new OrderByComparator(this.sortOrders);\n    this.aggregateMap = new NonStreamingOrderByMap<NonStreamingOrderByResult>(\n      (a: NonStreamingOrderByResult, b: NonStreamingOrderByResult) => {\n        return comparator.compareItems(a, b);\n      },\n    );\n    this.nonStreamingOrderByPQ = new FixedSizePriorityQueue<NonStreamingOrderByResult>(\n      (a: NonStreamingOrderByResult, b: NonStreamingOrderByResult) => {\n        return comparator.compareItems(b, a);\n      },\n      this.priorityQueueBufferSize,\n    );\n  }\n\n  /**\n   * Build final sorted result array from which responses will be served.\n   */\n  private async buildFinalResultArray(): Promise<void> {\n    // Fetch all distinct values from the map and store in priority queue.\n    const allValues = this.aggregateMap.getAllValuesAndReset();\n    for (const value of allValues) {\n      this.nonStreamingOrderByPQ.enqueue(value);\n    }\n\n    // Compute the final result array size based on offset and limit.\n    const offSet = this.queryInfo.offset ? this.queryInfo.offset : 0;\n    const queueSize = this.nonStreamingOrderByPQ.size();\n    const finalArraySize = queueSize - offSet;\n\n    if (finalArraySize <= 0) {\n      this.finalResultArray = [];\n    } else {\n      this.finalResultArray = new Array(finalArraySize);\n      // Only keep the final result array size number of items in the final result array and discard the rest.\n      for (let count = finalArraySize - 1; count >= 0; count--) {\n        if (this.emitRawOrderByPayload) {\n          this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue();\n        } else {\n          this.finalResultArray[count] = this.nonStreamingOrderByPQ.dequeue()?.payload;\n        }\n      }\n    }\n  }\n\n  public hasMoreResults(): boolean {\n    if (this.priorityQueueBufferSize === 0) return false;\n    return this.executionContext.hasMoreResults();\n  }\n\n  public async fetchMore(diagnosticNode?: DiagnosticNodeInternal): Promise<Response<any>> {\n    if (this.isCompleted) {\n      return {\n        result: undefined,\n        headers: getInitialHeader(),\n      };\n    }\n    let resHeaders = getInitialHeader();\n    // if size is 0, just return undefined to signal to more results. Valid if query is TOP 0 or LIMIT 0\n    if (this.priorityQueueBufferSize <= 0) {\n      return {\n        result: undefined,\n        headers: resHeaders,\n      };\n    }\n\n    // If there are more results in backend, keep filling map.\n    if (this.executionContext.hasMoreResults()) {\n      // Grab the next result\n      const response = await this.executionContext.fetchMore(diagnosticNode);\n\n      if (!response) {\n        this.isCompleted = true;\n        if (this.aggregateMap.size() > 0) {\n          await this.buildFinalResultArray();\n          const result = createParallelQueryResult(this.finalResultArray, new Map(), {}, undefined);\n\n          return {\n            result,\n            headers: resHeaders,\n          };\n        }\n        return { result: undefined, headers: resHeaders };\n      }\n\n      if (\n        response.result === undefined ||\n        !Array.isArray(response.result.buffer) ||\n        response.result.buffer.length === 0\n      ) {\n        this.isCompleted = true;\n        if (this.aggregateMap.size() > 0) {\n          await this.buildFinalResultArray();\n          const result = createParallelQueryResult(this.finalResultArray, new Map(), {}, undefined);\n\n          return {\n            result,\n            headers: response.headers,\n          };\n        }\n        return { result: undefined, headers: response.headers };\n      }\n      resHeaders = response.headers;\n\n      const parallelResult = response.result as ParallelQueryResult;\n      const dataToProcess: NonStreamingOrderByResult[] =\n        parallelResult.buffer as NonStreamingOrderByResult[];\n\n      for (const item of dataToProcess) {\n        if (item) {\n          const key = await hashObject(item?.payload);\n          this.aggregateMap.set(key, item);\n        }\n      }\n\n      // return [] to signal that there are more results to fetch.\n      if (this.executionContext.hasMoreResults()) {\n        const result = createParallelQueryResult(\n          [], // empty buffer\n          new Map(),\n          undefined,\n          undefined,\n        );\n\n        return {\n          result,\n          headers: resHeaders,\n        };\n      }\n    }\n\n    // If all results are fetched from backend, prepare final results\n    if (!this.executionContext.hasMoreResults() && !this.isCompleted) {\n      this.isCompleted = true;\n      await this.buildFinalResultArray();\n      const result = createParallelQueryResult(this.finalResultArray, new Map());\n\n      return {\n        result,\n        headers: resHeaders,\n      };\n    }\n    // Signal that there are no more results.\n    const result = createParallelQueryResult([], new Map());\n\n    return {\n      result,\n      headers: resHeaders,\n    };\n  }\n}\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAIA,yBAAiC;AAEjC,wBAA2B;AAE3B,oCAAuC;AACvC,oCAAuC;AACvC,+BAAkC;AAElC,iCAA0C;AAMnC,MAAM,6CAAyE;AAAA,EAoBpF,YACU,kBACA,WACA,yBACA,wBAAiC,OACzC;AAJQ;AACA;AACA;AACA;AAER,SAAK,aAAa,KAAK,UAAU;AACjC,UAAM,aAAa,IAAI,2CAAkB,KAAK,UAAU;AACxD,SAAK,eAAe,IAAI;AAAA,MACtB,CAAC,GAA8B,MAAiC;AAC9D,eAAO,WAAW,aAAa,GAAG,CAAC;AAAA,MACrC;AAAA,IACF;AACA,SAAK,wBAAwB,IAAI;AAAA,MAC/B,CAAC,GAA8B,MAAiC;AAC9D,eAAO,WAAW,aAAa,GAAG,CAAC;AAAA,MACrC;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAAA,EAlBU;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EApBF;AAAA;AAAA;AAAA;AAAA,EAIA;AAAA;AAAA;AAAA;AAAA,EAIA;AAAA,EAEA;AAAA;AAAA;AAAA;AAAA,EAIA,cAAuB;AAAA;AAAA;AAAA;AAAA,EA0B/B,MAAc,wBAAuC;AAEnD,UAAM,YAAY,KAAK,aAAa,qBAAqB;AACzD,eAAW,SAAS,WAAW;AAC7B,WAAK,sBAAsB,QAAQ,KAAK;AAAA,IAC1C;AAGA,UAAM,SAAS,KAAK,UAAU,SAAS,KAAK,UAAU,SAAS;AAC/D,UAAM,YAAY,KAAK,sBAAsB,KAAK;AAClD,UAAM,iBAAiB,YAAY;AAEnC,QAAI,kBAAkB,GAAG;AACvB,WAAK,mBAAmB,CAAC;AAAA,IAC3B,OAAO;AACL,WAAK,mBAAmB,IAAI,MAAM,cAAc;AAEhD,eAAS,QAAQ,iBAAiB,GAAG,SAAS,GAAG,SAAS;AACxD,YAAI,KAAK,uBAAuB;AAC9B,eAAK,iBAAiB,KAAK,IAAI,KAAK,sBAAsB,QAAQ;AAAA,QACpE,OAAO;AACL,eAAK,iBAAiB,KAAK,IAAI,KAAK,sBAAsB,QAAQ,GAAG;AAAA,QACvE;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEO,iBAA0B;AAC/B,QAAI,KAAK,4BAA4B,EAAG,QAAO;AAC/C,WAAO,KAAK,iBAAiB,eAAe;AAAA,EAC9C;AAAA,EAEA,MAAa,UAAU,gBAAiE;AACtF,QAAI,KAAK,aAAa;AACpB,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,aAAS,qCAAiB;AAAA,MAC5B;AAAA,IACF;AACA,QAAI,iBAAa,qCAAiB;AAElC,QAAI,KAAK,2BAA2B,GAAG;AACrC,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAGA,QAAI,KAAK,iBAAiB,eAAe,GAAG;AAE1C,YAAM,WAAW,MAAM,KAAK,iBAAiB,UAAU,cAAc;AAErE,UAAI,CAAC,UAAU;AACb,aAAK,cAAc;AACnB,YAAI,KAAK,aAAa,KAAK,IAAI,GAAG;AAChC,gBAAM,KAAK,sBAAsB;AACjC,gBAAMA,cAAS,sDAA0B,KAAK,kBAAkB,oBAAI,IAAI,GAAG,CAAC,GAAG,MAAS;AAExF,iBAAO;AAAA,YACL,QAAAA;AAAA,YACA,SAAS;AAAA,UACX;AAAA,QACF;AACA,eAAO,EAAE,QAAQ,QAAW,SAAS,WAAW;AAAA,MAClD;AAEA,UACE,SAAS,WAAW,UACpB,CAAC,MAAM,QAAQ,SAAS,OAAO,MAAM,KACrC,SAAS,OAAO,OAAO,WAAW,GAClC;AACA,aAAK,cAAc;AACnB,YAAI,KAAK,aAAa,KAAK,IAAI,GAAG;AAChC,gBAAM,KAAK,sBAAsB;AACjC,gBAAMA,cAAS,sDAA0B,KAAK,kBAAkB,oBAAI,IAAI,GAAG,CAAC,GAAG,MAAS;AAExF,iBAAO;AAAA,YACL,QAAAA;AAAA,YACA,SAAS,SAAS;AAAA,UACpB;AAAA,QACF;AACA,eAAO,EAAE,QAAQ,QAAW,SAAS,SAAS,QAAQ;AAAA,MACxD;AACA,mBAAa,SAAS;AAEtB,YAAM,iBAAiB,SAAS;AAChC,YAAM,gBACJ,eAAe;AAEjB,iBAAW,QAAQ,eAAe;AAChC,YAAI,MAAM;AACR,gBAAM,MAAM,UAAM,8BAAW,MAAM,OAAO;AAC1C,eAAK,aAAa,IAAI,KAAK,IAAI;AAAA,QACjC;AAAA,MACF;AAGA,UAAI,KAAK,iBAAiB,eAAe,GAAG;AAC1C,cAAMA,cAAS;AAAA,UACb,CAAC;AAAA;AAAA,UACD,oBAAI,IAAI;AAAA,UACR;AAAA,UACA;AAAA,QACF;AAEA,eAAO;AAAA,UACL,QAAAA;AAAA,UACA,SAAS;AAAA,QACX;AAAA,MACF;AAAA,IACF;AAGA,QAAI,CAAC,KAAK,iBAAiB,eAAe,KAAK,CAAC,KAAK,aAAa;AAChE,WAAK,cAAc;AACnB,YAAM,KAAK,sBAAsB;AACjC,YAAMA,cAAS,sDAA0B,KAAK,kBAAkB,oBAAI,IAAI,CAAC;AAEzE,aAAO;AAAA,QACL,QAAAA;AAAA,QACA,SAAS;AAAA,MACX;AAAA,IACF;AAEA,UAAM,aAAS,sDAA0B,CAAC,GAAG,oBAAI,IAAI,CAAC;AAEtD,WAAO;AAAA,MACL;AAAA,MACA,SAAS;AAAA,IACX;AAAA,EACF;AACF;",
+  "names": ["result"]
+}