@azure/cosmos 4.9.1 → 4.9.2

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

@@ -1 +1,7 @@
-{"version":3,"file":"documentProducer.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/documentProducer.ts"],"names":[],"mappings":";;;AAIA,iDAO4B;AAI5B,uFAAiF;AAEjF,qDAAgE;AAChE,qDAAoD;AAIpD,cAAc;AACd,MAAa,gBAAgB;IA0BjB;IAzBF,cAAc,CAAS;IACvB,KAAK,CAAwB;IAC9B,uBAAuB,CAAoB;IAC3C,YAAY,CAAgB;IAC5B,UAAU,CAAU;IACnB,GAAG,CAAQ;IACZ,yBAAyB,CAAS;IAClC,iBAAiB,CAAS;IAC1B,UAAU,GAAW,CAAC,CAAC;IACtB,wBAAwB,CAA+B;IACxD,QAAQ,CAAS;IACjB,MAAM,CAAS;IACf,uBAAuB,CAAU;IAChC,MAAM,CAAkB;IACxB,kBAAkB,CAA6D;IAEvF;;;;;;;OAOG;IACH,YACU,aAA4B,EACpC,cAAsB,EACtB,KAAmB,EACnB,uBAA0C,EAC1C,OAAoB,EACpB,oBAA4B,EAC5B,QAAiB,EACjB,MAAe,EACf,0BAAmC,KAAK,EACxC,MAAuB;QATf,kBAAa,GAAb,aAAa,CAAe;QAWpC,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;QAEnC,IAAI,CAAC,wBAAwB,GAAG,IAAI,8DAA4B,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;QACvD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,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,gCAAe,CAAC,IAAI;oBACvB,IAAI,GAAG,IAAI,CAAC;oBACZ,MAAM;gBACR,KAAK,gCAAe,CAAC,SAAS;oBAC5B,IAAI,GAAG,IAAI,CAAC;oBACZ,MAAM;gBACR,KAAK,gCAAe,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;IAEM,aAAa,GAA0B,KAAK,EACjD,cAAsC,EACtC,OAAoB,EACpB,oBAA4B,EACC,EAAE;QAC/B,MAAM,IAAI,GAAG,IAAA,0BAAe,EAAC,IAAI,CAAC,cAAc,EAAE,uBAAY,CAAC,IAAI,CAAC,CAAC;QACrE,cAAc,CAAC,OAAO,CAAC,EAAE,mBAAmB,EAAE,IAAI,CAAC,uBAAuB,CAAC,EAAE,EAAE,CAAC,CAAC;QACjF,MAAM,EAAE,GAAG,IAAA,wBAAa,EAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;QAEtE,OAAO,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;YAClC,IAAI;YACJ,YAAY,EAAE,uBAAY,CAAC,IAAI;YAC/B,UAAU,EAAE,EAAE;YACd,QAAQ,EAAE,CAAC,MAAW,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS;YAC3C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO;YACP,cAAc;YACd,mBAAmB,EAAE,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC;YACvD,oBAAoB,EAAE,oBAAoB;YAC1C,QAAQ,EAAE,QAAQ;YAClB,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;IACL,CAAC,CAAC;IAEK,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,gCAAe,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,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,sBAAW,CAAC,IAAI;YAC/B,WAAW,IAAI,KAAK;YACpB,KAAK,CAAC,WAAW,CAAC,KAAK,yBAAc,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,eAAe,EAAE,OAAO,EAAE,cAAc,EAAE,GACxD,MAAM,IAAI,CAAC,wBAAwB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAChE,IAAI,SAAS,GAAG,eAAe,CAAC;YAChC,EAAE,IAAI,CAAC,UAAU,CAAC;YAClB,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,KAAK,SAAS,CAAC,CAAC;YACvD,IAAI,cAAc,IAAI,cAAc,CAAC,kCAAkC,CAAC,EAAE,CAAC;gBACzE,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,kCAAkC,CAAC,CAAC,CAAC;YAC3F,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;gBAC7B,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;YACjD,CAAC;YAED,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,4BAAW,CACb,OAAO,EACP,SAAS,EACT,sBAAsB,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAA,iCAAgB,GAAE,CAC7D,CACF,CAAC;oBACF,sBAAsB,GAAG,KAAK,CAAC;gBACjC,CAAC,CAAC,CAAC;YACL,CAAC;YAED,iFAAiF;YACjF,IAAI,cAAc,IAAI,IAAI,IAAI,oBAAS,CAAC,WAAW,CAAC,YAAY,IAAI,cAAc,EAAE,CAAC;gBACnF,gEAAgE;gBAChE,MAAM,YAAY,GAAG,cAAc,CAAC,oBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC;gBAE7E,gFAAgF;gBAChF,cAAc,CAAC,oBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;gBACxD,cAAc,CAAC,oBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,EAAE,CAAC;oBACjF,YAAY,CAAC;YACjB,CAAC;YACD,OAAO,cAAc,CAAC;QACxB,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,4BAAW,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;gBACtD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACtC,OAAO,GAAG,CAAC,OAAO,CAAC;YACrB,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,0BAA0B;QAC/B,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,gCAAe,CAAC,IAAI;gBACvB,OAAO,SAAS,CAAC;YAEnB,KAAK,gCAAe,CAAC,SAAS,EAAE,0BAA0B;gBACxD,OAAO,SAAS,CAAC;YAEnB,KAAK,gCAAe,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,IAAA,iCAAgB,GAAE,EAAE,CAAC;QAC5D,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;QAC7B,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,IAAA,iCAAgB,GAAE,EAAE,CAAC;QAC5D,CAAC;QACD,MAAM,SAAS,GAAU,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;gBAClC,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;gBACpD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;oBACzB,OAAO;wBACL,MAAM,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;wBACpD,OAAO,EAAE,IAAA,iCAAgB,GAAE;qBAC5B,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAA,iCAAgB,GAAE,EAAE,CAAC;QAC5D,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,gCAAe,CAAC,IAAI;oBACvB,OAAO;wBACL,MAAM,EAAE,SAAS;wBACjB,OAAO,EAAE,IAAA,iCAAgB,GAAE;qBAC5B,CAAC;gBACJ,KAAK,gCAAe,CAAC,SAAS;oBAC5B,WAAW,CAAC,KAAK,CAAC,OAAO,GAAG,IAAA,iCAAgB,GAAE,CAAC;oBAC/C,MAAM,WAAW,CAAC,KAAK,CAAC;gBAC1B,KAAK,gCAAe,CAAC,MAAM;oBACzB,OAAO;wBACL,MAAM,EAAE,WAAW,CAAC,YAAY;wBAChC,OAAO,EAAE,IAAA,iCAAgB,GAAE;qBAC5B,CAAC;YACN,CAAC;QACH,CAAC;QAED,qEAAqE;QACrE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAA,iCAAgB,GAAE;aAC5B,CAAC;QACJ,CAAC;QAED,uDAAuD;QACvD,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,IAAA,iCAAgB,GAAE,EAAE,CAAC;IACrD,CAAC;IAEM,qBAAqB;QAG1B,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;CACF;AA3UD,4CA2UC","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 } from \"./headerUtils.js\";\nimport type { CosmosHeaders } from \"./headerUtils.js\";\nimport type { SqlQuerySpec, FilterStrategy } 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 internalExecutionContext: DefaultQueryExecutionContext;\n  public startEpk: string;\n  public endEpk: string;\n  public populateEpkRangeHeaders: boolean;\n  private filter?: FilterStrategy;\n  private queryExecutionInfo?: { reverseRidEnabled: boolean; reverseIndexScan: 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    filter?: FilterStrategy,\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\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    this.filter = filter;\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 _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<CosmosHeaders> {\n    if (this.err) {\n      throw this.err;\n    }\n\n    try {\n      const { result: resourcesResult, headers: headerResponse } =\n        await this.internalExecutionContext.fetchMore(diagnosticNode);\n      let resources = resourcesResult;\n      ++this.generation;\n      this._updateStates(undefined, resources === undefined);\n      if (headerResponse && headerResponse[\"x-ms-cosmos-query-execution-info\"]) {\n        this.queryExecutionInfo = JSON.parse(headerResponse[\"x-ms-cosmos-query-execution-info\"]);\n      }\n\n      if (this.filter && resources) {\n        resources = this.filter.applyFilter(resources);\n      }\n\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      return 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        return err.headers;\n      } else {\n        this._updateStates(err, err.resources === undefined);\n        throw err;\n      }\n    }\n  }\n\n  public getTargetPartitionKeyRange(): 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: getInitialHeader() };\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: getInitialHeader() };\n    }\n    const resources: any[] = [];\n    try {\n      while (this.fetchResults.length > 0) {\n        const { result } = this.current();\n        this._updateStates(undefined, result === undefined);\n        if (result === undefined) {\n          return {\n            result: resources.length > 0 ? resources : undefined,\n            headers: getInitialHeader(),\n          };\n        } else {\n          resources.push(result);\n        }\n      }\n      return { result: resources, headers: getInitialHeader() };\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: getInitialHeader(),\n          };\n        case FetchResultType.Exception:\n          fetchResult.error.headers = getInitialHeader();\n          throw fetchResult.error;\n        case FetchResultType.Result:\n          return {\n            result: fetchResult.feedResponse,\n            headers: getInitialHeader(),\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: getInitialHeader(),\n      };\n    }\n\n    // If the internal buffer is empty, return empty result\n    return { result: [], headers: getInitialHeader() };\n  }\n\n  public getQueryExecutionInfo():\n    | { reverseRidEnabled: boolean; reverseIndexScan: boolean }\n    | undefined {\n    return this.queryExecutionInfo;\n  }\n}\n"]}
+{
+  "version": 3,
+  "sources": ["/mnt/vss/_work/1/s/sdk/cosmosdb/cosmos/src/queryExecutionContext/documentProducer.ts"],
+  "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 } from \"./headerUtils.js\";\nimport type { CosmosHeaders } from \"./headerUtils.js\";\nimport type { SqlQuerySpec, FilterStrategy } 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 internalExecutionContext: DefaultQueryExecutionContext;\n  public startEpk: string;\n  public endEpk: string;\n  public populateEpkRangeHeaders: boolean;\n  private filter?: FilterStrategy;\n  private queryExecutionInfo?: { reverseRidEnabled: boolean; reverseIndexScan: 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    filter?: FilterStrategy,\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\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    this.filter = filter;\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 _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<CosmosHeaders> {\n    if (this.err) {\n      throw this.err;\n    }\n\n    try {\n      const { result: resourcesResult, headers: headerResponse } =\n        await this.internalExecutionContext.fetchMore(diagnosticNode);\n      let resources = resourcesResult;\n      ++this.generation;\n      this._updateStates(undefined, resources === undefined);\n      if (headerResponse && headerResponse[\"x-ms-cosmos-query-execution-info\"]) {\n        this.queryExecutionInfo = JSON.parse(headerResponse[\"x-ms-cosmos-query-execution-info\"]);\n      }\n\n      if (this.filter && resources) {\n        resources = this.filter.applyFilter(resources);\n      }\n\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      return 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        return err.headers;\n      } else {\n        this._updateStates(err, err.resources === undefined);\n        throw err;\n      }\n    }\n  }\n\n  public getTargetPartitionKeyRange(): 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: getInitialHeader() };\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: getInitialHeader() };\n    }\n    const resources: any[] = [];\n    try {\n      while (this.fetchResults.length > 0) {\n        const { result } = this.current();\n        this._updateStates(undefined, result === undefined);\n        if (result === undefined) {\n          return {\n            result: resources.length > 0 ? resources : undefined,\n            headers: getInitialHeader(),\n          };\n        } else {\n          resources.push(result);\n        }\n      }\n      return { result: resources, headers: getInitialHeader() };\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: getInitialHeader(),\n          };\n        case FetchResultType.Exception:\n          fetchResult.error.headers = getInitialHeader();\n          throw fetchResult.error;\n        case FetchResultType.Result:\n          return {\n            result: fetchResult.feedResponse,\n            headers: getInitialHeader(),\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: getInitialHeader(),\n      };\n    }\n\n    // If the internal buffer is empty, return empty result\n    return { result: [], headers: getInitialHeader() };\n  }\n\n  public getQueryExecutionInfo():\n    | { reverseRidEnabled: boolean; reverseIndexScan: boolean }\n    | undefined {\n    return this.queryExecutionInfo;\n  }\n}\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAIA,oBAOO;AAIP,0CAA6C;AAE7C,yBAA6C;AAC7C,yBAAiC;AAK1B,MAAM,iBAAiB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAyB5B,YACU,eACR,gBACA,OACA,yBACA,SACA,sBACA,UACA,QACA,0BAAmC,OACnC,QACA;AAVQ;AAYR,SAAK,iBAAiB;AACtB,SAAK,QAAQ;AACb,SAAK,0BAA0B;AAC/B,SAAK,eAAe,CAAC;AAErB,SAAK,aAAa;AAClB,SAAK,MAAM;AAEX,SAAK,4BAA4B;AACjC,SAAK,oBAAoB;AAEzB,SAAK,2BAA2B,IAAI;AAAA,MAClC;AAAA,MACA,KAAK;AAAA,MACL;AAAA,IACF;AACA,SAAK,WAAW;AAChB,SAAK,SAAS;AACd,SAAK,0BAA0B;AAC/B,SAAK,SAAS;AAAA,EAChB;AAAA,EAzDQ;AAAA,EACA;AAAA,EACD;AAAA,EACA;AAAA,EACA;AAAA,EACC;AAAA,EACD;AAAA,EACA;AAAA,EACA,aAAqB;AAAA,EACpB;AAAA,EACD;AAAA,EACA;AAAA,EACA;AAAA,EACC;AAAA,EACA;AAAA,EA4CD,oBAA2B;AAChC,UAAM,kBAAkB,CAAC;AACzB,aAAS,IAAI,GAAG,OAAO,OAAO,IAAI,KAAK,aAAa,UAAU,CAAC,MAAM,KAAK;AACxE,YAAM,cAAc,KAAK,aAAa,CAAC;AACvC,cAAQ,YAAY,iBAAiB;AAAA,QACnC,KAAK,mCAAgB;AACnB,iBAAO;AACP;AAAA,QACF,KAAK,mCAAgB;AACnB,iBAAO;AACP;AAAA,QACF,KAAK,mCAAgB;AACnB,0BAAgB,KAAK,YAAY,YAAY;AAC7C;AAAA,MACJ;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEO,gBAAuC,OAC5C,gBACA,SACA,yBACgC;AAChC,UAAM,WAAO,+BAAgB,KAAK,gBAAgB,2BAAa,IAAI;AACnE,mBAAe,QAAQ,EAAE,qBAAqB,KAAK,wBAAwB,GAAG,CAAC;AAC/E,UAAM,SAAK,6BAAc,KAAK,cAAc;AAC5C,UAAM,WAAW,KAAK,0BAA0B,KAAK,WAAW;AAChE,UAAM,SAAS,KAAK,0BAA0B,KAAK,SAAS;AAE5D,WAAO,KAAK,cAAc,UAAU;AAAA,MAClC;AAAA,MACA,cAAc,2BAAa;AAAA,MAC3B,YAAY;AAAA,MACZ,UAAU,CAAC,WAAgB,OAAO;AAAA,MAClC,OAAO,KAAK;AAAA,MACZ;AAAA,MACA;AAAA,MACA,qBAAqB,KAAK,wBAAwB,IAAI;AAAA,MACtD;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEO,iBAA0B;AAC/B,WAAO,KAAK,yBAAyB,eAAe,KAAK,KAAK,aAAa,WAAW;AAAA,EACxF;AAAA,EAEO,WAAoB;AACzB,QAAI,KAAK,aAAa,WAAW,GAAG;AAClC,YAAM,cAAc,KAAK,aAAa,CAAC;AACvC,UAAI,YAAY,oBAAoB,mCAAgB,WAAW;AAC7D,YAAI,iBAAiB,mCAAmC,YAAY,KAAK,GAAG;AAC1E,iBAAO;AAAA,QACT;AAAA,MACF;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEQ,cAAc,KAAU,YAA2B;AACzD,QAAI,KAAK;AACP,WAAK,MAAM;AACX;AAAA,IACF;AACA,QAAI,YAAY;AACd,WAAK,aAAa;AAAA,IACpB;AACA,QAAI,KAAK,yBAAyB,sBAAsB,KAAK,mBAAmB;AAE9E;AAAA,IACF;AACA,SAAK,4BAA4B,KAAK;AACtC,SAAK,oBAAoB,KAAK,yBAAyB;AAAA,EACzD;AAAA,EAEA,OAAe,mCAAmC,OAAqB;AAErE,WACE,MAAM,SAAS,0BAAY,QAC3B,eAAe,SACf,MAAM,WAAW,MAAM,6BAAe;AAAA,EAE1C;AAAA;AAAA;AAAA;AAAA,EAKA,MAAa,WAAW,gBAAgE;AACtF,QAAI,KAAK,KAAK;AACZ,YAAM,KAAK;AAAA,IACb;AAEA,QAAI;AACF,YAAM,EAAE,QAAQ,iBAAiB,SAAS,eAAe,IACvD,MAAM,KAAK,yBAAyB,UAAU,cAAc;AAC9D,UAAI,YAAY;AAChB,QAAE,KAAK;AACP,WAAK,cAAc,QAAW,cAAc,MAAS;AACrD,UAAI,kBAAkB,eAAe,kCAAkC,GAAG;AACxE,aAAK,qBAAqB,KAAK,MAAM,eAAe,kCAAkC,CAAC;AAAA,MACzF;AAEA,UAAI,KAAK,UAAU,WAAW;AAC5B,oBAAY,KAAK,OAAO,YAAY,SAAS;AAAA,MAC/C;AAEA,UAAI,cAAc,QAAW;AAE3B,YAAI,yBAAyB;AAC7B,kBAAU,QAAQ,CAAC,YAAiB;AAClC,eAAK,aAAa;AAAA,YAChB,IAAI;AAAA,cACF;AAAA,cACA;AAAA,cACA,yBAAyB,qBAAiB,qCAAiB;AAAA,YAC7D;AAAA,UACF;AACA,mCAAyB;AAAA,QAC3B,CAAC;AAAA,MACH;AAGA,UAAI,kBAAkB,QAAQ,wBAAU,YAAY,gBAAgB,gBAAgB;AAElF,cAAM,eAAe,eAAe,wBAAU,YAAY,YAAY,EAAE,GAAG;AAG3E,uBAAe,wBAAU,YAAY,YAAY,IAAI,CAAC;AACtD,uBAAe,wBAAU,YAAY,YAAY,EAAE,KAAK,wBAAwB,EAAE,IAChF;AAAA,MACJ;AACA,aAAO;AAAA,IACT,SAAS,KAAU;AACjB,UAAI,iBAAiB,mCAAmC,GAAG,GAAG;AAG5D,cAAM,gBAAgB,IAAI,+BAAY,QAAW,GAAG;AACpD,aAAK,aAAa,KAAK,aAAa;AACpC,eAAO,IAAI;AAAA,MACb,OAAO;AACL,aAAK,cAAc,KAAK,IAAI,cAAc,MAAS;AACnD,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAAA,EAEO,6BAAgD;AACrD,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAIO,eAAoB;AACzB,QAAI,KAAK,KAAK;AACZ,YAAM,KAAK;AAAA,IACb;AACA,QAAI,KAAK,cAAc,KAAK,aAAa,WAAW,GAAG;AACrD,aAAO;AAAA,IACT;AACA,UAAM,cAAc,KAAK,aAAa,CAAC;AAEvC,YAAQ,YAAY,iBAAiB;AAAA,MACnC,KAAK,mCAAgB;AACnB,eAAO;AAAA,MAET,KAAK,mCAAgB;AACnB,eAAO;AAAA,MAET,KAAK,mCAAgB;AACnB,eAAO,YAAY;AAAA,IACvB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAa,gBAAwC;AACnD,QAAI,KAAK,KAAK;AACZ,WAAK,cAAc,KAAK,KAAK,MAAS;AACtC,YAAM,KAAK;AAAA,IACb;AACA,QAAI,KAAK,YAAY;AACnB,aAAO,EAAE,QAAQ,QAAW,aAAS,qCAAiB,EAAE;AAAA,IAC1D;AACA,QAAI;AACF,YAAM,EAAE,QAAQ,QAAQ,IAAI,KAAK,QAAQ;AACzC,WAAK,cAAc,QAAW,WAAW,MAAS;AAClD,UAAI,WAAW,UAAa,OAAO,WAAW,GAAG;AAC/C,eAAO,EAAE,QAAQ,QAAW,QAAQ;AAAA,MACtC;AACA,aAAO,EAAE,QAAQ,QAAQ;AAAA,IAC3B,SAAS,KAAU;AACjB,WAAK,cAAc,KAAK,IAAI,SAAS,MAAS;AAC9C,YAAM;AAAA,IACR;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAIA,MAAa,qBAA+C;AAC1D,QAAI,KAAK,KAAK;AACZ,WAAK,cAAc,KAAK,KAAK,MAAS;AACtC,YAAM,KAAK;AAAA,IACb;AACA,QAAI,KAAK,YAAY;AACnB,aAAO,EAAE,QAAQ,QAAW,aAAS,qCAAiB,EAAE;AAAA,IAC1D;AACA,UAAM,YAAmB,CAAC;AAC1B,QAAI;AACF,aAAO,KAAK,aAAa,SAAS,GAAG;AACnC,cAAM,EAAE,OAAO,IAAI,KAAK,QAAQ;AAChC,aAAK,cAAc,QAAW,WAAW,MAAS;AAClD,YAAI,WAAW,QAAW;AACxB,iBAAO;AAAA,YACL,QAAQ,UAAU,SAAS,IAAI,YAAY;AAAA,YAC3C,aAAS,qCAAiB;AAAA,UAC5B;AAAA,QACF,OAAO;AACL,oBAAU,KAAK,MAAM;AAAA,QACvB;AAAA,MACF;AACA,aAAO,EAAE,QAAQ,WAAW,aAAS,qCAAiB,EAAE;AAAA,IAC1D,SAAS,KAAU;AACjB,WAAK,cAAc,KAAK,IAAI,SAAS,MAAS;AAC9C,YAAM;AAAA,IACR;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,UAAyB;AAE/B,QAAI,KAAK,aAAa,SAAS,GAAG;AAChC,YAAM,cAAc,KAAK,aAAa,MAAM;AAE5C,cAAQ,YAAY,iBAAiB;AAAA,QACnC,KAAK,mCAAgB;AACnB,iBAAO;AAAA,YACL,QAAQ;AAAA,YACR,aAAS,qCAAiB;AAAA,UAC5B;AAAA,QACF,KAAK,mCAAgB;AACnB,sBAAY,MAAM,cAAU,qCAAiB;AAC7C,gBAAM,YAAY;AAAA,QACpB,KAAK,mCAAgB;AACnB,iBAAO;AAAA,YACL,QAAQ,YAAY;AAAA,YACpB,aAAS,qCAAiB;AAAA,UAC5B;AAAA,MACJ;AAAA,IACF;AAGA,QAAI,KAAK,YAAY;AACnB,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,aAAS,qCAAiB;AAAA,MAC5B;AAAA,IACF;AAGA,WAAO,EAAE,QAAQ,CAAC,GAAG,aAAS,qCAAiB,EAAE;AAAA,EACnD;AAAA,EAEO,wBAEO;AACZ,WAAO,KAAK;AAAA,EACd;AACF;",
+  "names": []
+}

package/dist/commonjs/queryExecutionContext/headerUtils.js

@@ -1,100 +1,105 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getRequestChargeIfAny = getRequestChargeIfAny;
-exports.getInitialHeader = getInitialHeader;
-exports.mergeHeaders = mergeHeaders;
-exports.decodeAndParseJSONString = decodeAndParseJSONString;
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
-const index_js_1 = require("../common/index.js");
-const queryMetrics_js_1 = require("../queryMetrics/queryMetrics.js");
-/** @hidden */
-// TODO: docs
+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 headerUtils_exports = {};
+__export(headerUtils_exports, {
+  decodeAndParseJSONString: () => decodeAndParseJSONString,
+  getInitialHeader: () => getInitialHeader,
+  getRequestChargeIfAny: () => getRequestChargeIfAny,
+  mergeHeaders: () => mergeHeaders
+});
+module.exports = __toCommonJS(headerUtils_exports);
+var import_common = require("../common/index.js");
+var import_queryMetrics = require("../queryMetrics/queryMetrics.js");
 function getRequestChargeIfAny(headers) {
-    if (typeof headers === "number") {
-        return headers;
-    }
-    else if (typeof headers === "string") {
-        return parseFloat(headers);
-    }
-    if (headers) {
-        const rc = headers[index_js_1.Constants.HttpHeaders.RequestCharge];
-        if (rc) {
-            return parseFloat(rc);
-        }
-        else {
-            return 0;
-        }
-    }
-    else {
-        return 0;
+  if (typeof headers === "number") {
+    return headers;
+  } else if (typeof headers === "string") {
+    return parseFloat(headers);
+  }
+  if (headers) {
+    const rc = headers[import_common.Constants.HttpHeaders.RequestCharge];
+    if (rc) {
+      return parseFloat(rc);
+    } else {
+      return 0;
     }
+  } else {
+    return 0;
+  }
 }
-/**
- * @hidden
- */
 function getInitialHeader() {
-    const headers = {};
-    headers[index_js_1.Constants.HttpHeaders.RequestCharge] = 0;
-    headers[index_js_1.Constants.HttpHeaders.QueryMetrics] = {};
-    return headers;
+  const headers = {};
+  headers[import_common.Constants.HttpHeaders.RequestCharge] = 0;
+  headers[import_common.Constants.HttpHeaders.QueryMetrics] = {};
+  return headers;
 }
-/**
- * @hidden
- */
-// TODO: The name of this method isn't very accurate to what it does
 function mergeHeaders(headers, toBeMergedHeaders) {
-    if (headers[index_js_1.Constants.HttpHeaders.RequestCharge] === undefined) {
-        headers[index_js_1.Constants.HttpHeaders.RequestCharge] = 0;
-    }
-    if (headers[index_js_1.Constants.HttpHeaders.QueryMetrics] === undefined) {
-        headers[index_js_1.Constants.HttpHeaders.QueryMetrics] = queryMetrics_js_1.QueryMetrics.zero;
-    }
-    if (!toBeMergedHeaders) {
-        return;
-    }
-    headers[index_js_1.Constants.HttpHeaders.RequestCharge] += getRequestChargeIfAny(toBeMergedHeaders);
-    if (toBeMergedHeaders[index_js_1.Constants.HttpHeaders.IsRUPerMinuteUsed]) {
-        headers[index_js_1.Constants.HttpHeaders.IsRUPerMinuteUsed] =
-            toBeMergedHeaders[index_js_1.Constants.HttpHeaders.IsRUPerMinuteUsed];
-    }
-    if (index_js_1.Constants.HttpHeaders.QueryMetrics in toBeMergedHeaders) {
-        const headerQueryMetrics = headers[index_js_1.Constants.HttpHeaders.QueryMetrics];
-        const toBeMergedHeaderQueryMetrics = toBeMergedHeaders[index_js_1.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 (index_js_1.Constants.HttpHeaders.IndexUtilization in toBeMergedHeaders) {
-        headers[index_js_1.Constants.HttpHeaders.IndexUtilization] =
-            toBeMergedHeaders[index_js_1.Constants.HttpHeaders.IndexUtilization];
-    }
-    if (index_js_1.Constants.HttpHeaders.CorrelatedActivityId in toBeMergedHeaders) {
-        headers[index_js_1.Constants.HttpHeaders.CorrelatedActivityId] =
-            toBeMergedHeaders[index_js_1.Constants.HttpHeaders.CorrelatedActivityId];
+  if (headers[import_common.Constants.HttpHeaders.RequestCharge] === void 0) {
+    headers[import_common.Constants.HttpHeaders.RequestCharge] = 0;
+  }
+  if (headers[import_common.Constants.HttpHeaders.QueryMetrics] === void 0) {
+    headers[import_common.Constants.HttpHeaders.QueryMetrics] = import_queryMetrics.QueryMetrics.zero;
+  }
+  if (!toBeMergedHeaders) {
+    return;
+  }
+  headers[import_common.Constants.HttpHeaders.RequestCharge] += getRequestChargeIfAny(toBeMergedHeaders);
+  if (toBeMergedHeaders[import_common.Constants.HttpHeaders.IsRUPerMinuteUsed]) {
+    headers[import_common.Constants.HttpHeaders.IsRUPerMinuteUsed] = toBeMergedHeaders[import_common.Constants.HttpHeaders.IsRUPerMinuteUsed];
+  }
+  if (import_common.Constants.HttpHeaders.QueryMetrics in toBeMergedHeaders) {
+    const headerQueryMetrics = headers[import_common.Constants.HttpHeaders.QueryMetrics];
+    const toBeMergedHeaderQueryMetrics = toBeMergedHeaders[import_common.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 (import_common.Constants.HttpHeaders.IndexUtilization in toBeMergedHeaders) {
+    headers[import_common.Constants.HttpHeaders.IndexUtilization] = toBeMergedHeaders[import_common.Constants.HttpHeaders.IndexUtilization];
+  }
+  if (import_common.Constants.HttpHeaders.CorrelatedActivityId in toBeMergedHeaders) {
+    headers[import_common.Constants.HttpHeaders.CorrelatedActivityId] = toBeMergedHeaders[import_common.Constants.HttpHeaders.CorrelatedActivityId];
+  }
 }
-/** @hidden */
 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);
+  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
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  decodeAndParseJSONString,
+  getInitialHeader,
+  getRequestChargeIfAny,
+  mergeHeaders
+});

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

@@ -1 +1,7 @@
-{"version":3,"file":"headerUtils.js","sourceRoot":"","sources":["../../../src/queryExecutionContext/headerUtils.ts"],"names":[],"mappings":";;AAWA,sDAiBC;AAKD,4CAKC;AAMD,oCA4CC;AAGD,4DAYC;AAvGD,uCAAuC;AACvC,kCAAkC;AAClC,iDAA+C;AAC/C,qEAA+D;AAM/D,cAAc;AACd,aAAa;AACb,SAAgB,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,oBAAS,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,SAAgB,gBAAgB;IAC9B,MAAM,OAAO,GAAkB,EAAE,CAAC;IAClC,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACjD,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;IACjD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,oEAAoE;AACpE,SAAgB,YAAY,CAAC,OAAsB,EAAE,iBAAgC;IACnF,IAAI,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,aAAa,CAAC,KAAK,SAAS,EAAE,CAAC;QAC/D,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,KAAK,SAAS,EAAE,CAAC;QAC9D,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,8BAAY,CAAC,IAAI,CAAC;IAClE,CAAC;IAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,OAAO;IACT,CAAC;IAED,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,aAAa,CAAC,IAAI,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;IACzF,IAAI,iBAAiB,CAAC,oBAAS,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC/D,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,iBAAiB,CAAC;YAC9C,iBAAiB,CAAC,oBAAS,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC;IAC/D,CAAC;IAED,IAAI,oBAAS,CAAC,WAAW,CAAC,YAAY,IAAI,iBAAiB,EAAE,CAAC;QAC5D,MAAM,kBAAkB,GAAG,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QACvE,MAAM,4BAA4B,GAAG,iBAAiB,CAAC,oBAAS,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,oBAAS,CAAC,WAAW,CAAC,gBAAgB,IAAI,iBAAiB,EAAE,CAAC;QAChE,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,gBAAgB,CAAC;YAC7C,iBAAiB,CAAC,oBAAS,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC;IAC9D,CAAC;IAED,IAAI,oBAAS,CAAC,WAAW,CAAC,oBAAoB,IAAI,iBAAiB,EAAE,CAAC;QACpE,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,oBAAoB,CAAC;YACjD,iBAAiB,CAAC,oBAAS,CAAC,WAAW,CAAC,oBAAoB,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED,cAAc;AACd,SAAgB,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"]}
+{
+  "version": 3,
+  "sources": ["/mnt/vss/_work/1/s/sdk/cosmosdb/cosmos/src/queryExecutionContext/headerUtils.ts"],
+  "sourcesContent": ["\uFEFF// 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"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,oBAA0B;AAC1B,0BAA6B;AAQtB,SAAS,sBAAsB,SAAyC;AAC7E,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO;AAAA,EACT,WAAW,OAAO,YAAY,UAAU;AACtC,WAAO,WAAW,OAAO;AAAA,EAC3B;AAEA,MAAI,SAAS;AACX,UAAM,KAAK,QAAQ,wBAAU,YAAY,aAAa;AACtD,QAAI,IAAI;AACN,aAAO,WAAW,EAAY;AAAA,IAChC,OAAO;AACL,aAAO;AAAA,IACT;AAAA,EACF,OAAO;AACL,WAAO;AAAA,EACT;AACF;AAKO,SAAS,mBAAkC;AAChD,QAAM,UAAyB,CAAC;AAChC,UAAQ,wBAAU,YAAY,aAAa,IAAI;AAC/C,UAAQ,wBAAU,YAAY,YAAY,IAAI,CAAC;AAC/C,SAAO;AACT;AAMO,SAAS,aAAa,SAAwB,mBAAwC;AAC3F,MAAI,QAAQ,wBAAU,YAAY,aAAa,MAAM,QAAW;AAC9D,YAAQ,wBAAU,YAAY,aAAa,IAAI;AAAA,EACjD;AAEA,MAAI,QAAQ,wBAAU,YAAY,YAAY,MAAM,QAAW;AAC7D,YAAQ,wBAAU,YAAY,YAAY,IAAI,iCAAa;AAAA,EAC7D;AAEA,MAAI,CAAC,mBAAmB;AACtB;AAAA,EACF;AAEA,UAAQ,wBAAU,YAAY,aAAa,KAAK,sBAAsB,iBAAiB;AACvF,MAAI,kBAAkB,wBAAU,YAAY,iBAAiB,GAAG;AAC9D,YAAQ,wBAAU,YAAY,iBAAiB,IAC7C,kBAAkB,wBAAU,YAAY,iBAAiB;AAAA,EAC7D;AAEA,MAAI,wBAAU,YAAY,gBAAgB,mBAAmB;AAC3D,UAAM,qBAAqB,QAAQ,wBAAU,YAAY,YAAY;AACrE,UAAM,+BAA+B,kBAAkB,wBAAU,YAAY,YAAY;AAEzF,eAAW,eAAe,8BAA8B;AACtD,UAAI,mBAAmB,WAAW,GAAG;AACnC,cAAM,uBAAuB,mBAAmB,WAAW,EAAE,IAAI;AAAA,UAC/D,6BAA6B,WAAW;AAAA,QAC1C,CAAC;AACD,2BAAmB,WAAW,IAAI;AAAA,MACpC,OAAO;AACL,2BAAmB,WAAW,IAAI,6BAA6B,WAAW;AAAA,MAC5E;AAAA,IACF;AAAA,EACF;AAEA,MAAI,wBAAU,YAAY,oBAAoB,mBAAmB;AAC/D,YAAQ,wBAAU,YAAY,gBAAgB,IAC5C,kBAAkB,wBAAU,YAAY,gBAAgB;AAAA,EAC5D;AAEA,MAAI,wBAAU,YAAY,wBAAwB,mBAAmB;AACnE,YAAQ,wBAAU,YAAY,oBAAoB,IAChD,kBAAkB,wBAAU,YAAY,oBAAoB;AAAA,EAChE;AACF;AAGO,SAAS,yBAAyB,aAA6B;AACpE,MAAI;AACF,QAAI,CAAC,eAAe,gBAAgB,IAAI;AACtC,aAAO;AAAA,IACT;AACA,UAAM,gBAAgB,mBAAmB,WAAW;AACpD,UAAM,eAAe,KAAK,MAAM,aAAa;AAC7C,UAAM,eAAe,KAAK,UAAU,YAAY;AAChD,WAAO;AAAA,EACT,SAAS,GAAG;AACV,YAAQ,MAAM,4BAA4B,EAAE,OAAO;AAAA,EACrD;AACF;",
+  "names": []
+}