@azure/cosmos 4.3.0 → 4.4.0

package/dist/browser/queryExecutionContext/documentProducer.js

@@ -0,0 +1,269 @@
+import { Constants, getIdFromLink, getPathFromLink, ResourceType, StatusCodes, SubStatusCodes, } from "../common/index.js";
+import { DefaultQueryExecutionContext } from "./defaultQueryExecutionContext.js";
+import { FetchResult, FetchResultType } from "./FetchResult.js";
+import { getInitialHeader, mergeHeaders } from "./headerUtils.js";
+/** @hidden */
+export class DocumentProducer {
+    /**
+     * Provides the Target Partition Range Query Execution Context.
+     * @param clientContext  - The service endpoint to use to create the client.
+     * @param collectionLink - Represents collection link
+     * @param query          - A SQL query.
+     * @param targetPartitionKeyRange - Query Target Partition key Range
+     * @hidden
+     */
+    constructor(clientContext, collectionLink, query, targetPartitionKeyRange, options, correlatedActivityId, startEpk, endEpk, populateEpkRangeHeaders = false) {
+        this.clientContext = clientContext;
+        this.generation = 0;
+        this.fetchFunction = async (diagnosticNode, options, correlatedActivityId) => {
+            const path = getPathFromLink(this.collectionLink, ResourceType.item);
+            diagnosticNode.addData({ partitionKeyRangeId: this.targetPartitionKeyRange.id });
+            const id = getIdFromLink(this.collectionLink);
+            const startEpk = this.populateEpkRangeHeaders ? this.startEpk : undefined;
+            const endEpk = this.populateEpkRangeHeaders ? this.endEpk : undefined;
+            return this.clientContext.queryFeed({
+                path,
+                resourceType: ResourceType.item,
+                resourceId: id,
+                resultFn: (result) => result.Documents,
+                query: this.query,
+                options,
+                diagnosticNode,
+                partitionKeyRangeId: this.targetPartitionKeyRange["id"],
+                correlatedActivityId: correlatedActivityId,
+                startEpk: startEpk,
+                endEpk: endEpk,
+            });
+        };
+        // TODO: any options
+        this.collectionLink = collectionLink;
+        this.query = query;
+        this.targetPartitionKeyRange = targetPartitionKeyRange;
+        this.fetchResults = [];
+        this.allFetched = false;
+        this.err = undefined;
+        this.previousContinuationToken = undefined;
+        this.continuationToken = undefined;
+        this.respHeaders = getInitialHeader();
+        this.internalExecutionContext = new DefaultQueryExecutionContext(options, this.fetchFunction, correlatedActivityId);
+        this.startEpk = startEpk;
+        this.endEpk = endEpk;
+        this.populateEpkRangeHeaders = populateEpkRangeHeaders;
+    }
+    peekBufferedItems() {
+        const bufferedResults = [];
+        for (let i = 0, done = false; i < this.fetchResults.length && !done; i++) {
+            const fetchResult = this.fetchResults[i];
+            switch (fetchResult.fetchResultType) {
+                case FetchResultType.Done:
+                    done = true;
+                    break;
+                case FetchResultType.Exception:
+                    done = true;
+                    break;
+                case FetchResultType.Result:
+                    bufferedResults.push(fetchResult.feedResponse);
+                    break;
+            }
+        }
+        return bufferedResults;
+    }
+    hasMoreResults() {
+        return this.internalExecutionContext.hasMoreResults() || this.fetchResults.length !== 0;
+    }
+    gotSplit() {
+        if (this.fetchResults.length !== 0) {
+            const fetchResult = this.fetchResults[0];
+            if (fetchResult.fetchResultType === FetchResultType.Exception) {
+                if (DocumentProducer._needPartitionKeyRangeCacheRefresh(fetchResult.error)) {
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+    _getAndResetActiveResponseHeaders() {
+        const ret = this.respHeaders;
+        this.respHeaders = getInitialHeader();
+        return ret;
+    }
+    _updateStates(err, allFetched) {
+        if (err) {
+            this.err = err;
+            return;
+        }
+        if (allFetched) {
+            this.allFetched = true;
+        }
+        if (this.internalExecutionContext.continuationToken === this.continuationToken) {
+            // nothing changed
+            return;
+        }
+        this.previousContinuationToken = this.continuationToken;
+        this.continuationToken = this.internalExecutionContext.continuationToken;
+    }
+    static _needPartitionKeyRangeCacheRefresh(error) {
+        // TODO: error
+        return (error.code === StatusCodes.Gone &&
+            "substatus" in error &&
+            error["substatus"] === SubStatusCodes.PartitionKeyRangeGone);
+    }
+    /**
+     * Fetches and bufferes the next page of results in internal buffer
+     */
+    async bufferMore(diagnosticNode) {
+        if (this.err) {
+            throw this.err;
+        }
+        try {
+            const { result: resources, headers: headerResponse } = await this.internalExecutionContext.fetchMore(diagnosticNode);
+            ++this.generation;
+            this._updateStates(undefined, resources === undefined);
+            if (resources !== undefined) {
+                // add fetched header to the 1st element in the buffer
+                let addHeaderToFetchResult = true;
+                resources.forEach((element) => {
+                    this.fetchResults.push(new FetchResult(element, undefined, addHeaderToFetchResult ? headerResponse : getInitialHeader()));
+                    addHeaderToFetchResult = false;
+                });
+            }
+            // need to modify the header response so that the query metrics are per partition
+            if (headerResponse != null && Constants.HttpHeaders.QueryMetrics in headerResponse) {
+                // "0" is the default partition before one is actually assigned.
+                const queryMetrics = headerResponse[Constants.HttpHeaders.QueryMetrics]["0"];
+                // Wraping query metrics in a object where the keys are the partition key range.
+                headerResponse[Constants.HttpHeaders.QueryMetrics] = {};
+                headerResponse[Constants.HttpHeaders.QueryMetrics][this.targetPartitionKeyRange.id] =
+                    queryMetrics;
+            }
+            mergeHeaders(this.respHeaders, headerResponse);
+        }
+        catch (err) {
+            if (DocumentProducer._needPartitionKeyRangeCacheRefresh(err)) {
+                // Split just happend
+                // Buffer the error so the execution context can still get the feedResponses in the itemBuffer
+                const bufferedError = new FetchResult(undefined, err);
+                this.fetchResults.push(bufferedError);
+                mergeHeaders(this.respHeaders, err.headers);
+            }
+            else {
+                this._updateStates(err, err.resources === undefined);
+                throw err;
+            }
+        }
+    }
+    getTargetParitionKeyRange() {
+        return this.targetPartitionKeyRange;
+    }
+    /**
+     * Peak the next item in the buffer
+     */
+    peakNextItem() {
+        if (this.err) {
+            throw this.err;
+        }
+        if (this.allFetched || this.fetchResults.length === 0) {
+            return undefined;
+        }
+        const fetchResult = this.fetchResults[0];
+        switch (fetchResult.fetchResultType) {
+            case FetchResultType.Done:
+                return undefined;
+            case FetchResultType.Exception: // do not throw this error
+                return undefined;
+            case FetchResultType.Result:
+                return fetchResult.feedResponse;
+        }
+    }
+    /**
+     * Returns the first item in the buffered results if any, or [] otherwise.
+     */
+    async fetchNextItem() {
+        if (this.err) {
+            this._updateStates(this.err, undefined);
+            throw this.err;
+        }
+        if (this.allFetched) {
+            return { result: undefined, headers: this._getAndResetActiveResponseHeaders() };
+        }
+        try {
+            const { result, headers } = this.current();
+            this._updateStates(undefined, result === undefined);
+            if (result === undefined || result.length === 0) {
+                return { result: undefined, headers };
+            }
+            return { result, headers }; //
+        }
+        catch (err) {
+            this._updateStates(err, err.item === undefined);
+            throw err;
+        }
+    }
+    /**
+     * Fetches all the buffered results
+     */
+    async fetchBufferedItems() {
+        if (this.err) {
+            this._updateStates(this.err, undefined);
+            throw this.err;
+        }
+        if (this.allFetched) {
+            return { result: undefined, headers: this._getAndResetActiveResponseHeaders() };
+        }
+        const resources = [];
+        const resHeaders = getInitialHeader();
+        try {
+            while (this.fetchResults.length > 0) {
+                const { result, headers } = this.current();
+                this._updateStates(undefined, result === undefined);
+                mergeHeaders(resHeaders, headers);
+                if (result === undefined) {
+                    return { result: resources.length > 0 ? resources : undefined, headers: resHeaders };
+                }
+                else {
+                    resources.push(result);
+                }
+            }
+            return { result: resources, headers: resHeaders };
+        }
+        catch (err) {
+            this._updateStates(err, err.item === undefined);
+            throw err;
+        }
+    }
+    /**
+     * Retrieve the current element on the DocumentProducer.
+     */
+    current() {
+        // If something is buffered just give that
+        if (this.fetchResults.length > 0) {
+            const fetchResult = this.fetchResults.shift();
+            // Need to unwrap fetch results
+            switch (fetchResult.fetchResultType) {
+                case FetchResultType.Done:
+                    return {
+                        result: undefined,
+                        headers: this._getAndResetActiveResponseHeaders(),
+                    };
+                case FetchResultType.Exception:
+                    fetchResult.error.headers = this._getAndResetActiveResponseHeaders();
+                    throw fetchResult.error;
+                case FetchResultType.Result:
+                    return {
+                        result: fetchResult.feedResponse,
+                        headers: this._getAndResetActiveResponseHeaders(),
+                    };
+            }
+        }
+        // If there isn't anymore items left to fetch then let the user know.
+        if (this.allFetched) {
+            return {
+                result: undefined,
+                headers: this._getAndResetActiveResponseHeaders(),
+            };
+        }
+        // If the internal buffer is empty, return empty result
+        return { result: [], headers: this._getAndResetActiveResponseHeaders() };
+    }
+}
+//# sourceMappingURL=documentProducer.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"documentProducer.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/documentProducer.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,SAAS,EACT,aAAa,EACb,eAAe,EACf,YAAY,EACZ,WAAW,EACX,cAAc,GACf,MAAM,oBAAoB,CAAC;AAI5B,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAEjF,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAIlE,cAAc;AACd,MAAM,OAAO,gBAAgB;IAgB3B;;;;;;;OAOG;IACH,YACU,aAA4B,EACpC,cAAsB,EACtB,KAAmB,EACnB,uBAA0C,EAC1C,OAAoB,EACpB,oBAA4B,EAC5B,QAAiB,EACjB,MAAe,EACf,0BAAmC,KAAK;QARhC,kBAAa,GAAb,aAAa,CAAe;QAhB/B,eAAU,GAAW,CAAC,CAAC;QAmEvB,kBAAa,GAA0B,KAAK,EACjD,cAAsC,EACtC,OAAoB,EACpB,oBAA4B,EACC,EAAE;YAC/B,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,cAAc,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YACrE,cAAc,CAAC,OAAO,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,uBAAuB,CAAC,EAAE,EAAE,CAAC,CAAC;YACjF,MAAM,EAAE,GAAG,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAEtE,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;gBAClC,IAAI;gBACJ,YAAY,EAAE,YAAY,CAAC,IAAI;gBAC/B,UAAU,EAAE,EAAE;gBACd,QAAQ,EAAE,CAAC,MAAW,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS;gBAC3C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,OAAO;gBACP,cAAc;gBACd,mBAAmB,EAAE,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC;gBACvD,oBAAoB,EAAE,oBAAoB;gBAC1C,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE,MAAM;aACf,CAAC,CAAC;QACL,CAAC,CAAC;QAjEA,oBAAoB;QACpB,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,uBAAuB,GAAG,uBAAuB,CAAC;QACvD,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QAEvB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QACxB,IAAI,CAAC,GAAG,GAAG,SAAS,CAAC;QAErB,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;QAC3C,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,gBAAgB,EAAE,CAAC;QAEtC,IAAI,CAAC,wBAAwB,GAAG,IAAI,4BAA4B,CAC9D,OAAO,EACP,IAAI,CAAC,aAAa,EAClB,oBAAoB,CACrB,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,uBAAuB,GAAG,uBAAuB,CAAC;IACzD,CAAC;IACM,iBAAiB;QACtB,MAAM,eAAe,GAAG,EAAE,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,GAAG,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YACzE,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,QAAQ,WAAW,CAAC,eAAe,EAAE,CAAC;gBACpC,KAAK,eAAe,CAAC,IAAI;oBACvB,IAAI,GAAG,IAAI,CAAC;oBACZ,MAAM;gBACR,KAAK,eAAe,CAAC,SAAS;oBAC5B,IAAI,GAAG,IAAI,CAAC;oBACZ,MAAM;gBACR,KAAK,eAAe,CAAC,MAAM;oBACzB,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;oBAC/C,MAAM;YACV,CAAC;QACH,CAAC;QACD,OAAO,eAAe,CAAC;IACzB,CAAC;IA4BM,cAAc;QACnB,OAAO,IAAI,CAAC,wBAAwB,CAAC,cAAc,EAAE,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,CAAC;IAC1F,CAAC;IAEM,QAAQ;QACb,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,IAAI,WAAW,CAAC,eAAe,KAAK,eAAe,CAAC,SAAS,EAAE,CAAC;gBAC9D,IAAI,gBAAgB,CAAC,kCAAkC,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC3E,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,iCAAiC;QACvC,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,gBAAgB,EAAE,CAAC;QACtC,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,aAAa,CAAC,GAAQ,EAAE,UAAmB;QACjD,IAAI,GAAG,EAAE,CAAC;YACR,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;YACf,OAAO;QACT,CAAC;QACD,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;QACD,IAAI,IAAI,CAAC,wBAAwB,CAAC,iBAAiB,KAAK,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC/E,kBAAkB;YAClB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC,iBAAiB,CAAC;QACxD,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,wBAAwB,CAAC,iBAAiB,CAAC;IAC3E,CAAC;IAEO,MAAM,CAAC,kCAAkC,CAAC,KAAU;QAC1D,cAAc;QACd,OAAO,CACL,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC,IAAI;YAC/B,WAAW,IAAI,KAAK;YACpB,KAAK,CAAC,WAAW,CAAC,KAAK,cAAc,CAAC,qBAAqB,CAC5D,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU,CAAC,cAAsC;QAC5D,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,GAAG,CAAC;QACjB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,GAClD,MAAM,IAAI,CAAC,wBAAwB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAChE,EAAE,IAAI,CAAC,UAAU,CAAC;YAClB,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,KAAK,SAAS,CAAC,CAAC;YACvD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,sDAAsD;gBACtD,IAAI,sBAAsB,GAAG,IAAI,CAAC;gBAClC,SAAS,CAAC,OAAO,CAAC,CAAC,OAAY,EAAE,EAAE;oBACjC,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,IAAI,WAAW,CACb,OAAO,EACP,SAAS,EACT,sBAAsB,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAC7D,CACF,CAAC;oBACF,sBAAsB,GAAG,KAAK,CAAC;gBACjC,CAAC,CAAC,CAAC;YACL,CAAC;YAED,iFAAiF;YACjF,IAAI,cAAc,IAAI,IAAI,IAAI,SAAS,CAAC,WAAW,CAAC,YAAY,IAAI,cAAc,EAAE,CAAC;gBACnF,gEAAgE;gBAChE,MAAM,YAAY,GAAG,cAAc,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC;gBAE7E,gFAAgF;gBAChF,cAAc,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;gBACxD,cAAc,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,EAAE,CAAC;oBACjF,YAAY,CAAC;YACjB,CAAC;YACD,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;QACjD,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,IAAI,gBAAgB,CAAC,kCAAkC,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7D,qBAAqB;gBACrB,8FAA8F;gBAC9F,MAAM,aAAa,GAAG,IAAI,WAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;gBACtD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACtC,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC;gBACrD,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC;IAEM,yBAAyB;QAC9B,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IACD;;OAEG;IACI,YAAY;QACjB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,GAAG,CAAC;QACjB,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAEzC,QAAQ,WAAW,CAAC,eAAe,EAAE,CAAC;YACpC,KAAK,eAAe,CAAC,IAAI;gBACvB,OAAO,SAAS,CAAC;YAEnB,KAAK,eAAe,CAAC,SAAS,EAAE,0BAA0B;gBACxD,OAAO,SAAS,CAAC;YAEnB,KAAK,eAAe,CAAC,MAAM;gBACzB,OAAO,WAAW,CAAC,YAAY,CAAC;QACpC,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,aAAa;QACxB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACxC,MAAM,IAAI,CAAC,GAAG,CAAC;QACjB,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,iCAAiC,EAAE,EAAE,CAAC;QAClF,CAAC;QACD,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;YAC3C,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;YACpD,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAChD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;YACxC,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE;QAChC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YAChD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IACD;;OAEG;IACI,KAAK,CAAC,kBAAkB;QAC7B,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACxC,MAAM,IAAI,CAAC,GAAG,CAAC;QACjB,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,iCAAiC,EAAE,EAAE,CAAC;QAClF,CAAC;QACD,MAAM,SAAS,GAAU,EAAE,CAAC;QAC5B,MAAM,UAAU,GAAkB,gBAAgB,EAAE,CAAC;QACrD,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC3C,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;gBACpD,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAClC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;oBACzB,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;gBACvF,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;QACpD,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YAChD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,OAAO;QACb,0CAA0C;QAC1C,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC9C,+BAA+B;YAC/B,QAAQ,WAAW,CAAC,eAAe,EAAE,CAAC;gBACpC,KAAK,eAAe,CAAC,IAAI;oBACvB,OAAO;wBACL,MAAM,EAAE,SAAS;wBACjB,OAAO,EAAE,IAAI,CAAC,iCAAiC,EAAE;qBAClD,CAAC;gBACJ,KAAK,eAAe,CAAC,SAAS;oBAC5B,WAAW,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,iCAAiC,EAAE,CAAC;oBACrE,MAAM,WAAW,CAAC,KAAK,CAAC;gBAC1B,KAAK,eAAe,CAAC,MAAM;oBACzB,OAAO;wBACL,MAAM,EAAE,WAAW,CAAC,YAAY;wBAChC,OAAO,EAAE,IAAI,CAAC,iCAAiC,EAAE;qBAClD,CAAC;YACN,CAAC;QACH,CAAC;QAED,qEAAqE;QACrE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAI,CAAC,iCAAiC,EAAE;aAClD,CAAC;QACJ,CAAC;QAED,uDAAuD;QACvD,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,iCAAiC,EAAE,EAAE,CAAC;IAC3E,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { PartitionKeyRange, Resource } from \"../client/index.js\";\nimport type { ClientContext } from \"../ClientContext.js\";\nimport {\n  Constants,\n  getIdFromLink,\n  getPathFromLink,\n  ResourceType,\n  StatusCodes,\n  SubStatusCodes,\n} from \"../common/index.js\";\nimport type { DiagnosticNodeInternal } from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport type { FeedOptions } from \"../request/index.js\";\nimport type { Response } from \"../request/index.js\";\nimport { DefaultQueryExecutionContext } from \"./defaultQueryExecutionContext.js\";\nimport type { FetchFunctionCallback } from \"./defaultQueryExecutionContext.js\";\nimport { FetchResult, FetchResultType } from \"./FetchResult.js\";\nimport { getInitialHeader, mergeHeaders } from \"./headerUtils.js\";\nimport type { CosmosHeaders } from \"./headerUtils.js\";\nimport type { SqlQuerySpec } from \"./index.js\";\n\n/** @hidden */\nexport class DocumentProducer {\n  private collectionLink: string;\n  private query: string | SqlQuerySpec;\n  public targetPartitionKeyRange: PartitionKeyRange;\n  public fetchResults: FetchResult[];\n  public allFetched: boolean;\n  private err: Error;\n  public previousContinuationToken: string;\n  public continuationToken: string;\n  public generation: number = 0;\n  private respHeaders: CosmosHeaders;\n  private internalExecutionContext: DefaultQueryExecutionContext;\n  public startEpk: string;\n  public endEpk: string;\n  public populateEpkRangeHeaders: boolean;\n\n  /**\n   * Provides the Target Partition Range Query Execution Context.\n   * @param clientContext  - The service endpoint to use to create the client.\n   * @param collectionLink - Represents collection link\n   * @param query          - A SQL query.\n   * @param targetPartitionKeyRange - Query Target Partition key Range\n   * @hidden\n   */\n  constructor(\n    private clientContext: ClientContext,\n    collectionLink: string,\n    query: SqlQuerySpec,\n    targetPartitionKeyRange: PartitionKeyRange,\n    options: FeedOptions,\n    correlatedActivityId: string,\n    startEpk?: string,\n    endEpk?: string,\n    populateEpkRangeHeaders: boolean = false,\n  ) {\n    // TODO: any options\n    this.collectionLink = collectionLink;\n    this.query = query;\n    this.targetPartitionKeyRange = targetPartitionKeyRange;\n    this.fetchResults = [];\n\n    this.allFetched = false;\n    this.err = undefined;\n\n    this.previousContinuationToken = undefined;\n    this.continuationToken = undefined;\n    this.respHeaders = getInitialHeader();\n\n    this.internalExecutionContext = new DefaultQueryExecutionContext(\n      options,\n      this.fetchFunction,\n      correlatedActivityId,\n    );\n    this.startEpk = startEpk;\n    this.endEpk = endEpk;\n    this.populateEpkRangeHeaders = populateEpkRangeHeaders;\n  }\n  public peekBufferedItems(): any[] {\n    const bufferedResults = [];\n    for (let i = 0, done = false; i < this.fetchResults.length && !done; i++) {\n      const fetchResult = this.fetchResults[i];\n      switch (fetchResult.fetchResultType) {\n        case FetchResultType.Done:\n          done = true;\n          break;\n        case FetchResultType.Exception:\n          done = true;\n          break;\n        case FetchResultType.Result:\n          bufferedResults.push(fetchResult.feedResponse);\n          break;\n      }\n    }\n    return bufferedResults;\n  }\n\n  public fetchFunction: FetchFunctionCallback = async (\n    diagnosticNode: DiagnosticNodeInternal,\n    options: FeedOptions,\n    correlatedActivityId: string,\n  ): Promise<Response<Resource>> => {\n    const path = getPathFromLink(this.collectionLink, ResourceType.item);\n    diagnosticNode.addData({ partitionKeyRangeId: this.targetPartitionKeyRange.id });\n    const id = getIdFromLink(this.collectionLink);\n    const startEpk = this.populateEpkRangeHeaders ? this.startEpk : undefined;\n    const endEpk = this.populateEpkRangeHeaders ? this.endEpk : undefined;\n\n    return this.clientContext.queryFeed({\n      path,\n      resourceType: ResourceType.item,\n      resourceId: id,\n      resultFn: (result: any) => result.Documents,\n      query: this.query,\n      options,\n      diagnosticNode,\n      partitionKeyRangeId: this.targetPartitionKeyRange[\"id\"],\n      correlatedActivityId: correlatedActivityId,\n      startEpk: startEpk,\n      endEpk: endEpk,\n    });\n  };\n\n  public hasMoreResults(): boolean {\n    return this.internalExecutionContext.hasMoreResults() || this.fetchResults.length !== 0;\n  }\n\n  public gotSplit(): boolean {\n    if (this.fetchResults.length !== 0) {\n      const fetchResult = this.fetchResults[0];\n      if (fetchResult.fetchResultType === FetchResultType.Exception) {\n        if (DocumentProducer._needPartitionKeyRangeCacheRefresh(fetchResult.error)) {\n          return true;\n        }\n      }\n    }\n    return false;\n  }\n\n  private _getAndResetActiveResponseHeaders(): CosmosHeaders {\n    const ret = this.respHeaders;\n    this.respHeaders = getInitialHeader();\n    return ret;\n  }\n\n  private _updateStates(err: any, allFetched: boolean): void {\n    if (err) {\n      this.err = err;\n      return;\n    }\n    if (allFetched) {\n      this.allFetched = true;\n    }\n    if (this.internalExecutionContext.continuationToken === this.continuationToken) {\n      // nothing changed\n      return;\n    }\n    this.previousContinuationToken = this.continuationToken;\n    this.continuationToken = this.internalExecutionContext.continuationToken;\n  }\n\n  private static _needPartitionKeyRangeCacheRefresh(error: any): boolean {\n    // TODO: error\n    return (\n      error.code === StatusCodes.Gone &&\n      \"substatus\" in error &&\n      error[\"substatus\"] === SubStatusCodes.PartitionKeyRangeGone\n    );\n  }\n\n  /**\n   * Fetches and bufferes the next page of results in internal buffer\n   */\n  public async bufferMore(diagnosticNode: DiagnosticNodeInternal): Promise<void> {\n    if (this.err) {\n      throw this.err;\n    }\n\n    try {\n      const { result: resources, headers: headerResponse } =\n        await this.internalExecutionContext.fetchMore(diagnosticNode);\n      ++this.generation;\n      this._updateStates(undefined, resources === undefined);\n      if (resources !== undefined) {\n        // add fetched header to the 1st element in the buffer\n        let addHeaderToFetchResult = true;\n        resources.forEach((element: any) => {\n          this.fetchResults.push(\n            new FetchResult(\n              element,\n              undefined,\n              addHeaderToFetchResult ? headerResponse : getInitialHeader(),\n            ),\n          );\n          addHeaderToFetchResult = false;\n        });\n      }\n\n      // need to modify the header response so that the query metrics are per partition\n      if (headerResponse != null && Constants.HttpHeaders.QueryMetrics in headerResponse) {\n        // \"0\" is the default partition before one is actually assigned.\n        const queryMetrics = headerResponse[Constants.HttpHeaders.QueryMetrics][\"0\"];\n\n        // Wraping query metrics in a object where the keys are the partition key range.\n        headerResponse[Constants.HttpHeaders.QueryMetrics] = {};\n        headerResponse[Constants.HttpHeaders.QueryMetrics][this.targetPartitionKeyRange.id] =\n          queryMetrics;\n      }\n      mergeHeaders(this.respHeaders, headerResponse);\n    } catch (err: any) {\n      if (DocumentProducer._needPartitionKeyRangeCacheRefresh(err)) {\n        // Split just happend\n        // Buffer the error so the execution context can still get the feedResponses in the itemBuffer\n        const bufferedError = new FetchResult(undefined, err);\n        this.fetchResults.push(bufferedError);\n        mergeHeaders(this.respHeaders, err.headers);\n      } else {\n        this._updateStates(err, err.resources === undefined);\n        throw err;\n      }\n    }\n  }\n\n  public getTargetParitionKeyRange(): PartitionKeyRange {\n    return this.targetPartitionKeyRange;\n  }\n  /**\n   * Peak the next item in the buffer\n   */\n  public peakNextItem(): any {\n    if (this.err) {\n      throw this.err;\n    }\n    if (this.allFetched || this.fetchResults.length === 0) {\n      return undefined;\n    }\n    const fetchResult = this.fetchResults[0];\n\n    switch (fetchResult.fetchResultType) {\n      case FetchResultType.Done:\n        return undefined;\n\n      case FetchResultType.Exception: // do not throw this error\n        return undefined;\n\n      case FetchResultType.Result:\n        return fetchResult.feedResponse;\n    }\n  }\n\n  /**\n   * Returns the first item in the buffered results if any, or [] otherwise.\n   */\n  public async fetchNextItem(): Promise<Response<any>> {\n    if (this.err) {\n      this._updateStates(this.err, undefined);\n      throw this.err;\n    }\n    if (this.allFetched) {\n      return { result: undefined, headers: this._getAndResetActiveResponseHeaders() };\n    }\n    try {\n      const { result, headers } = this.current();\n      this._updateStates(undefined, result === undefined);\n      if (result === undefined || result.length === 0) {\n        return { result: undefined, headers };\n      }\n      return { result, headers }; //\n    } catch (err: any) {\n      this._updateStates(err, err.item === undefined);\n      throw err;\n    }\n  }\n  /**\n   * Fetches all the buffered results\n   */\n  public async fetchBufferedItems(): Promise<Response<any[]>> {\n    if (this.err) {\n      this._updateStates(this.err, undefined);\n      throw this.err;\n    }\n    if (this.allFetched) {\n      return { result: undefined, headers: this._getAndResetActiveResponseHeaders() };\n    }\n    const resources: any[] = [];\n    const resHeaders: CosmosHeaders = getInitialHeader();\n    try {\n      while (this.fetchResults.length > 0) {\n        const { result, headers } = this.current();\n        this._updateStates(undefined, result === undefined);\n        mergeHeaders(resHeaders, headers);\n        if (result === undefined) {\n          return { result: resources.length > 0 ? resources : undefined, headers: resHeaders };\n        } else {\n          resources.push(result);\n        }\n      }\n      return { result: resources, headers: resHeaders };\n    } catch (err: any) {\n      this._updateStates(err, err.item === undefined);\n      throw err;\n    }\n  }\n\n  /**\n   * Retrieve the current element on the DocumentProducer.\n   */\n  private current(): Response<any> {\n    // If something is buffered just give that\n    if (this.fetchResults.length > 0) {\n      const fetchResult = this.fetchResults.shift();\n      // Need to unwrap fetch results\n      switch (fetchResult.fetchResultType) {\n        case FetchResultType.Done:\n          return {\n            result: undefined,\n            headers: this._getAndResetActiveResponseHeaders(),\n          };\n        case FetchResultType.Exception:\n          fetchResult.error.headers = this._getAndResetActiveResponseHeaders();\n          throw fetchResult.error;\n        case FetchResultType.Result:\n          return {\n            result: fetchResult.feedResponse,\n            headers: this._getAndResetActiveResponseHeaders(),\n          };\n      }\n    }\n\n    // If there isn't anymore items left to fetch then let the user know.\n    if (this.allFetched) {\n      return {\n        result: undefined,\n        headers: this._getAndResetActiveResponseHeaders(),\n      };\n    }\n\n    // If the internal buffer is empty, return empty result\n    return { result: [], headers: this._getAndResetActiveResponseHeaders() };\n  }\n}\n"]}

package/dist/browser/queryExecutionContext/headerUtils.js

@@ -0,0 +1,94 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { Constants } from "../common/index.js";
+import { QueryMetrics } from "../queryMetrics/queryMetrics.js";
+/** @hidden */
+// TODO: docs
+export function getRequestChargeIfAny(headers) {
+    if (typeof headers === "number") {
+        return headers;
+    }
+    else if (typeof headers === "string") {
+        return parseFloat(headers);
+    }
+    if (headers) {
+        const rc = headers[Constants.HttpHeaders.RequestCharge];
+        if (rc) {
+            return parseFloat(rc);
+        }
+        else {
+            return 0;
+        }
+    }
+    else {
+        return 0;
+    }
+}
+/**
+ * @hidden
+ */
+export function getInitialHeader() {
+    const headers = {};
+    headers[Constants.HttpHeaders.RequestCharge] = 0;
+    headers[Constants.HttpHeaders.QueryMetrics] = {};
+    return headers;
+}
+/**
+ * @hidden
+ */
+// TODO: The name of this method isn't very accurate to what it does
+export function mergeHeaders(headers, toBeMergedHeaders) {
+    if (headers[Constants.HttpHeaders.RequestCharge] === undefined) {
+        headers[Constants.HttpHeaders.RequestCharge] = 0;
+    }
+    if (headers[Constants.HttpHeaders.QueryMetrics] === undefined) {
+        headers[Constants.HttpHeaders.QueryMetrics] = QueryMetrics.zero;
+    }
+    if (!toBeMergedHeaders) {
+        return;
+    }
+    headers[Constants.HttpHeaders.RequestCharge] += getRequestChargeIfAny(toBeMergedHeaders);
+    if (toBeMergedHeaders[Constants.HttpHeaders.IsRUPerMinuteUsed]) {
+        headers[Constants.HttpHeaders.IsRUPerMinuteUsed] =
+            toBeMergedHeaders[Constants.HttpHeaders.IsRUPerMinuteUsed];
+    }
+    if (Constants.HttpHeaders.QueryMetrics in toBeMergedHeaders) {
+        const headerQueryMetrics = headers[Constants.HttpHeaders.QueryMetrics];
+        const toBeMergedHeaderQueryMetrics = toBeMergedHeaders[Constants.HttpHeaders.QueryMetrics];
+        for (const partitionId in toBeMergedHeaderQueryMetrics) {
+            if (headerQueryMetrics[partitionId]) {
+                const combinedQueryMetrics = headerQueryMetrics[partitionId].add([
+                    toBeMergedHeaderQueryMetrics[partitionId],
+                ]);
+                headerQueryMetrics[partitionId] = combinedQueryMetrics;
+            }
+            else {
+                headerQueryMetrics[partitionId] = toBeMergedHeaderQueryMetrics[partitionId];
+            }
+        }
+    }
+    if (Constants.HttpHeaders.IndexUtilization in toBeMergedHeaders) {
+        headers[Constants.HttpHeaders.IndexUtilization] =
+            toBeMergedHeaders[Constants.HttpHeaders.IndexUtilization];
+    }
+    if (Constants.HttpHeaders.CorrelatedActivityId in toBeMergedHeaders) {
+        headers[Constants.HttpHeaders.CorrelatedActivityId] =
+            toBeMergedHeaders[Constants.HttpHeaders.CorrelatedActivityId];
+    }
+}
+/** @hidden */
+export function decodeAndParseJSONString(inputString) {
+    try {
+        if (!inputString || inputString === "") {
+            return "{}";
+        }
+        const decodedString = decodeURIComponent(inputString);
+        const parsedString = JSON.parse(decodedString);
+        const indexMetrics = JSON.stringify(parsedString);
+        return indexMetrics;
+    }
+    catch (e) {
+        console.error("Error parsing JSON file:", e.message);
+    }
+}
+//# sourceMappingURL=headerUtils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"headerUtils.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/headerUtils.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAM/D,cAAc;AACd,aAAa;AACb,MAAM,UAAU,qBAAqB,CAAC,OAA+B;IACnE,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,OAAO,CAAC;IACjB,CAAC;SAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QACvC,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,OAAO,EAAE,CAAC;QACZ,MAAM,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;QACxD,IAAI,EAAE,EAAE,CAAC;YACP,OAAO,UAAU,CAAC,EAAY,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB;IAC9B,MAAM,OAAO,GAAkB,EAAE,CAAC;IAClC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACjD,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;IACjD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,oEAAoE;AACpE,MAAM,UAAU,YAAY,CAAC,OAAsB,EAAE,iBAAgC;IACnF,IAAI,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,KAAK,SAAS,EAAE,CAAC;QAC/D,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,SAAS,EAAE,CAAC;QAC9D,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC;IAClE,CAAC;IAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,OAAO;IACT,CAAC;IAED,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,IAAI,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;IACzF,IAAI,iBAAiB,CAAC,SAAS,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC/D,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,iBAAiB,CAAC;YAC9C,iBAAiB,CAAC,SAAS,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAC/D,CAAC;IAED,IAAI,SAAS,CAAC,WAAW,CAAC,YAAY,IAAI,iBAAiB,EAAE,CAAC;QAC5D,MAAM,kBAAkB,GAAG,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QACvE,MAAM,4BAA4B,GAAG,iBAAiB,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QAE3F,KAAK,MAAM,WAAW,IAAI,4BAA4B,EAAE,CAAC;YACvD,IAAI,kBAAkB,CAAC,WAAW,CAAC,EAAE,CAAC;gBACpC,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC;oBAC/D,4BAA4B,CAAC,WAAW,CAAC;iBAC1C,CAAC,CAAC;gBACH,kBAAkB,CAAC,WAAW,CAAC,GAAG,oBAAoB,CAAC;YACzD,CAAC;iBAAM,CAAC;gBACN,kBAAkB,CAAC,WAAW,CAAC,GAAG,4BAA4B,CAAC,WAAW,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,SAAS,CAAC,WAAW,CAAC,gBAAgB,IAAI,iBAAiB,EAAE,CAAC;QAChE,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,gBAAgB,CAAC;YAC7C,iBAAiB,CAAC,SAAS,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC;IAC9D,CAAC;IAED,IAAI,SAAS,CAAC,WAAW,CAAC,oBAAoB,IAAI,iBAAiB,EAAE,CAAC;QACpE,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,oBAAoB,CAAC;YACjD,iBAAiB,CAAC,SAAS,CAAC,WAAW,CAAC,oBAAoB,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED,cAAc;AACd,MAAM,UAAU,wBAAwB,CAAC,WAAmB;IAC1D,IAAI,CAAC;QACH,IAAI,CAAC,WAAW,IAAI,WAAW,KAAK,EAAE,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,aAAa,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAClD,OAAO,YAAY,CAAC;IACtB,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { Constants } from \"../common/index.js\";\nimport { QueryMetrics } from \"../queryMetrics/queryMetrics.js\";\n\nexport interface CosmosHeaders {\n  [key: string]: any;\n}\n\n/** @hidden */\n// TODO: docs\nexport function getRequestChargeIfAny(headers: CosmosHeaders | number): number {\n  if (typeof headers === \"number\") {\n    return headers;\n  } else if (typeof headers === \"string\") {\n    return parseFloat(headers);\n  }\n\n  if (headers) {\n    const rc = headers[Constants.HttpHeaders.RequestCharge];\n    if (rc) {\n      return parseFloat(rc as string);\n    } else {\n      return 0;\n    }\n  } else {\n    return 0;\n  }\n}\n\n/**\n * @hidden\n */\nexport function getInitialHeader(): CosmosHeaders {\n  const headers: CosmosHeaders = {};\n  headers[Constants.HttpHeaders.RequestCharge] = 0;\n  headers[Constants.HttpHeaders.QueryMetrics] = {};\n  return headers;\n}\n\n/**\n * @hidden\n */\n// TODO: The name of this method isn't very accurate to what it does\nexport function mergeHeaders(headers: CosmosHeaders, toBeMergedHeaders: CosmosHeaders): void {\n  if (headers[Constants.HttpHeaders.RequestCharge] === undefined) {\n    headers[Constants.HttpHeaders.RequestCharge] = 0;\n  }\n\n  if (headers[Constants.HttpHeaders.QueryMetrics] === undefined) {\n    headers[Constants.HttpHeaders.QueryMetrics] = QueryMetrics.zero;\n  }\n\n  if (!toBeMergedHeaders) {\n    return;\n  }\n\n  headers[Constants.HttpHeaders.RequestCharge] += getRequestChargeIfAny(toBeMergedHeaders);\n  if (toBeMergedHeaders[Constants.HttpHeaders.IsRUPerMinuteUsed]) {\n    headers[Constants.HttpHeaders.IsRUPerMinuteUsed] =\n      toBeMergedHeaders[Constants.HttpHeaders.IsRUPerMinuteUsed];\n  }\n\n  if (Constants.HttpHeaders.QueryMetrics in toBeMergedHeaders) {\n    const headerQueryMetrics = headers[Constants.HttpHeaders.QueryMetrics];\n    const toBeMergedHeaderQueryMetrics = toBeMergedHeaders[Constants.HttpHeaders.QueryMetrics];\n\n    for (const partitionId in toBeMergedHeaderQueryMetrics) {\n      if (headerQueryMetrics[partitionId]) {\n        const combinedQueryMetrics = headerQueryMetrics[partitionId].add([\n          toBeMergedHeaderQueryMetrics[partitionId],\n        ]);\n        headerQueryMetrics[partitionId] = combinedQueryMetrics;\n      } else {\n        headerQueryMetrics[partitionId] = toBeMergedHeaderQueryMetrics[partitionId];\n      }\n    }\n  }\n\n  if (Constants.HttpHeaders.IndexUtilization in toBeMergedHeaders) {\n    headers[Constants.HttpHeaders.IndexUtilization] =\n      toBeMergedHeaders[Constants.HttpHeaders.IndexUtilization];\n  }\n\n  if (Constants.HttpHeaders.CorrelatedActivityId in toBeMergedHeaders) {\n    headers[Constants.HttpHeaders.CorrelatedActivityId] =\n      toBeMergedHeaders[Constants.HttpHeaders.CorrelatedActivityId];\n  }\n}\n\n/** @hidden */\nexport function decodeAndParseJSONString(inputString: string): string {\n  try {\n    if (!inputString || inputString === \"\") {\n      return \"{}\";\n    }\n    const decodedString = decodeURIComponent(inputString);\n    const parsedString = JSON.parse(decodedString);\n    const indexMetrics = JSON.stringify(parsedString);\n    return indexMetrics;\n  } catch (e) {\n    console.error(\"Error parsing JSON file:\", e.message);\n  }\n}\n"]}

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

@@ -0,0 +1,59 @@
+import type { ClientContext } from "../ClientContext.js";
+import type { DiagnosticNodeInternal } from "../diagnostics/DiagnosticNodeInternal.js";
+import type { FeedOptions, PartitionedQueryExecutionInfo, QueryRange, Response } from "../request/index.js";
+import type { ExecutionContext } from "./ExecutionContext.js";
+/** @hidden */
+export declare enum HybridQueryExecutionContextBaseStates {
+    uninitialized = "uninitialized",
+    initialized = "initialized",
+    draining = "draining",
+    done = "done"
+}
+export declare class HybridQueryExecutionContext implements ExecutionContext {
+    private clientContext;
+    private collectionLink;
+    private options;
+    private partitionedQueryExecutionInfo;
+    private correlatedActivityId;
+    private allPartitionsRanges;
+    private globalStatisticsExecutionContext;
+    private componentsExecutionContext;
+    private pageSize;
+    private state;
+    private globalStatisticsAggregator;
+    private emitRawOrderByPayload;
+    private buffer;
+    private DEFAULT_PAGE_SIZE;
+    private TOTAL_WORD_COUNT_PLACEHOLDER;
+    private HIT_COUNTS_ARRAY_PLACEHOLDER;
+    private TOTAL_DOCUMENT_COUNT_PLACEHOLDER;
+    private RRF_CONSTANT;
+    private logger;
+    private hybridSearchResult;
+    private uniqueItems;
+    private isSingleComponent;
+    constructor(clientContext: ClientContext, collectionLink: string, options: FeedOptions, partitionedQueryExecutionInfo: PartitionedQueryExecutionInfo, correlatedActivityId: string, allPartitionsRanges: QueryRange[]);
+    nextItem(diagnosticNode: DiagnosticNodeInternal): Promise<Response<any>>;
+    hasMoreResults(): boolean;
+    fetchMore(diagnosticNode?: DiagnosticNodeInternal): Promise<Response<any>>;
+    private fetchMoreInternal;
+    private initialize;
+    private executeComponentQueries;
+    private processUniqueItems;
+    private applySkipAndTakeToBuffer;
+    private drain;
+    private drainOne;
+    private done;
+    private sortHybridSearchResultByRRFScore;
+    private drainSingleComponent;
+    private createComponentExecutionContexts;
+    private processComponentQueries;
+    private replacePlaceholdersWorkaroud;
+    private computeRRFScore;
+    private extractComponentWeights;
+}
+export interface ComponentWeight {
+    weight: number;
+    comparator: (x: number, y: number) => number;
+}
+//# sourceMappingURL=hybridQueryExecutionContext.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"hybridQueryExecutionContext.d.ts","sourceRoot":"","sources":["../../../src/queryExecutionContext/hybridQueryExecutionContext.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,0CAA0C,CAAC;AACvF,OAAO,KAAK,EACV,WAAW,EAEX,6BAA6B,EAE7B,UAAU,EACV,QAAQ,EACT,MAAM,qBAAqB,CAAC;AAI7B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAK9D,cAAc;AACd,oBAAY,qCAAqC;IAC/C,aAAa,kBAAkB;IAC/B,WAAW,gBAAgB;IAC3B,QAAQ,aAAa;IACrB,IAAI,SAAS;CACd;AACD,qBAAa,2BAA4B,YAAW,gBAAgB;IAoBhE,OAAO,CAAC,aAAa;IACrB,OAAO,CAAC,cAAc;IACtB,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,6BAA6B;IACrC,OAAO,CAAC,oBAAoB;IAC5B,OAAO,CAAC,mBAAmB;IAxB7B,OAAO,CAAC,gCAAgC,CAAmB;IAC3D,OAAO,CAAC,0BAA0B,CAA0B;IAC5D,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,KAAK,CAAwC;IACrD,OAAO,CAAC,0BAA0B,CAA6B;IAC/D,OAAO,CAAC,qBAAqB,CAAiB;IAC9C,OAAO,CAAC,MAAM,CAAiC;IAC/C,OAAO,CAAC,iBAAiB,CAAM;IAC/B,OAAO,CAAC,4BAA4B,CAA4D;IAChG,OAAO,CAAC,4BAA4B,CAA4D;IAChG,OAAO,CAAC,gCAAgC,CACuB;IAC/D,OAAO,CAAC,YAAY,CAAM;IAC1B,OAAO,CAAC,MAAM,CAAkE;IAChF,OAAO,CAAC,kBAAkB,CAAiC;IAC3D,OAAO,CAAC,WAAW,CAA8C;IACjE,OAAO,CAAC,iBAAiB,CAAkB;gBAGjC,aAAa,EAAE,aAAa,EAC5B,cAAc,EAAE,MAAM,EACtB,OAAO,EAAE,WAAW,EACpB,6BAA6B,EAAE,6BAA6B,EAC5D,oBAAoB,EAAE,MAAM,EAC5B,mBAAmB,EAAE,UAAU,EAAE;IAsC9B,QAAQ,CAAC,cAAc,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAiB9E,cAAc,IAAI,OAAO;IAenB,SAAS,CAAC,cAAc,CAAC,EAAE,sBAAsB,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YAKzE,iBAAiB;YA2BjB,UAAU;YA4BV,uBAAuB;IAyDrC,OAAO,CAAC,kBAAkB;IAsB1B,OAAO,CAAC,wBAAwB;YAUlB,KAAK;YAqBL,QAAQ;IAoBtB,OAAO,CAAC,IAAI;IAOZ,OAAO,CAAC,gCAAgC;YAiD1B,oBAAoB;IAoDlC,OAAO,CAAC,gCAAgC;IA8BxC,OAAO,CAAC,uBAAuB;IAmD/B,OAAO,CAAC,4BAA4B;IAmCpC,OAAO,CAAC,eAAe,CAerB;IAEF,OAAO,CAAC,uBAAuB;CAmChC;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC;CAC9C"}