@azure/cosmos 4.5.0 → 4.6.0

package/dist/esm/CosmosDiagnostics.js.map

@@ -1 +1 @@
-{"version":3,"file":"CosmosDiagnostics.js","sourceRoot":"","sources":["../../src/CosmosDiagnostics.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAOlC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,OAAO,iBAAiB;IAI5B;;OAEG;IACH,YACE,2BAAwD,EACxD,cAA+B,EAC/B,YAAqC;QAErC,IAAI,CAAC,2BAA2B,GAAG,2BAA2B,CAAC;QAC/D,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;CACF;AAsID;;GAEG;AACH,MAAM,CAAN,IAAY,kBAMX;AAND,WAAY,kBAAkB;IAC5B,6EAAuD,CAAA;IACvD,wEAAkD,CAAA;IAClD,4DAAsC,CAAA;IACtC,yDAAmC,CAAA;IACnC,2DAAqC,CAAA;AACvC,CAAC,EANW,kBAAkB,KAAlB,kBAAkB,QAM7B;AA4CD,MAAM,UAAU,WAAW,CAAC,IAA4B;IACtD,IAAI,IAAI,CAAC,MAAM;QAAE,OAAO,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;;QAC5C,OAAO,IAAI,CAAC;AACnB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { OperationType, ResourceType } from \"./common/index.js\";\nimport type { CosmosDbDiagnosticLevel } from \"./diagnostics/CosmosDbDiagnosticLevel.js\";\nimport type { DiagnosticNodeInternal } from \"./diagnostics/DiagnosticNodeInternal.js\";\nimport type { ConsistencyLevel } from \"./documents/index.js\";\n\n/**\n *  * This is a Cosmos Diagnostic type that holds collected diagnostic information during a client operations. ie. Item.read(), Container.create().\n * It has three members -\n * 1. `clientSideRequestStatistics` member contains aggregate diagnostic information, including -\n *   - metadata lookups. Here all the server requests, apart from the final intended resource are considered as metadata calls.\n *    i.e. for item.read(id), if the client makes server call to discover endpoints it would be considered as metadata call.\n *   - retries\n *   - endpoints contacted.\n *   - request, response payload stats.\n *   - gatewayStatistics - Information corresponding to main operation. For example during Item.read(), the client might perform many operations\n *    i.e. metadata lookup etc, but gatewayStatistics represents the diagnostics information for actual read operation.\n *\n * 2. diagnosticNode - Is a tree like structure which captures detailed diagnostic information. By default it is disabled, and is intended to be\n * used only for debugging on non production environments. The kind of details captured in diagnosticNode is controlled by `CosmosDbDiagnosticLevel`.\n * - CosmosDbDiagnosticLevel.info - Is default value. In this level only clientSideRequestStatistics are captured. Is is meant for production environments.\n * - CosmosDbDiagnosticLevel.debug - Captures diagnosticNode and clientConfig. No request and response payloads are captured. Is not meant to be used\n * in production environment.\n * - CosmosDbDiagnosticLevel.debug-unsafe - In addition to data captured in CosmosDbDiagnosticLevel.debug, also captures request and response payloads.\n * Is not meant to be used in production environment.\n * 3. clientConfig - Captures information related to how client was configured during initialization.\n */\nexport class CosmosDiagnostics {\n  public readonly clientSideRequestStatistics: ClientSideRequestStatistics;\n  public readonly diagnosticNode: DiagnosticNode;\n  public readonly clientConfig?: ClientConfigDiagnostic;\n  /**\n   * @internal\n   */\n  constructor(\n    clientSideRequestStatistics: ClientSideRequestStatistics,\n    diagnosticNode?: DiagnosticNode,\n    clientConfig?: ClientConfigDiagnostic,\n  ) {\n    this.clientSideRequestStatistics = clientSideRequestStatistics;\n    this.diagnosticNode = diagnosticNode;\n    this.clientConfig = clientConfig;\n  }\n}\n\n/**\n * This type holds information related to initialization of `CosmosClient`\n */\nexport type ClientConfigDiagnostic = {\n  /**\n   * End point configured during client initialization.\n   */\n  endpoint: string;\n  /**\n   * True if `resourceTokens` was supplied during client initialization.\n   */\n  resourceTokensConfigured: boolean;\n  /**\n   * True if `tokenProvider` was supplied during client initialization.\n   */\n  tokenProviderConfigured: boolean;\n  /**\n   * True if `aadCredentials` was supplied during client initialization.\n   */\n  aadCredentialsConfigured: boolean;\n  /**\n   * True if `connectionPolicy` was supplied during client initialization.\n   */\n  connectionPolicyConfigured: boolean;\n  /**\n   * `consistencyLevel` supplied during client initialization.\n   */\n  consistencyLevel?: keyof typeof ConsistencyLevel;\n  /**\n   * `defaultHeaders` supplied during client initialization.\n   */\n  defaultHeaders?: { [key: string]: any };\n  /**\n   * True if `connectionPolicy` were supplied during client initialization.\n   */\n  agentConfigured: boolean;\n  /**\n   * `userAgentSuffix` supplied during client initialization.\n   */\n  userAgentSuffix: string;\n  /**\n   * `diagnosticLevel` supplied during client initialization.\n   */\n  diagnosticLevel?: CosmosDbDiagnosticLevel;\n  /**\n   * True if `plugins` were supplied during client initialization.\n   */\n  pluginsConfigured: boolean;\n  /**\n   * SDK version\n   */\n  sDKVersion: string;\n};\n\n/**\n * This type contains diagnostic information regarding all metadata request to server during an CosmosDB client operation.\n */\nexport type MetadataLookUpDiagnostics = {\n  metadataLookups: MetadataLookUpDiagnostic[];\n};\n\n/**\n * This type captures diagnostic information regarding retries attempt during an CosmosDB client operation.\n */\nexport type RetryDiagnostics = {\n  failedAttempts: FailedRequestAttemptDiagnostic[];\n};\n\nexport type GatewayStatistics = {\n  /**\n   * This is the activityId for request, made to server for fetching the requested resource. (As opposed to other potential meta data requests)\n   */\n  activityId?: string;\n  /*\n   * The correlated activity ID is added in the header of every request made to the backend for a query. This aids in troubleshooting by linking all\n   * requests associated with a particular query.\n   */\n  correlatedActivityId?: string;\n  startTimeUTCInMs: number;\n  durationInMs: number;\n  operationType?: OperationType;\n  resourceType?: ResourceType;\n  statusCode?: number;\n  subStatusCode?: number;\n  requestCharge?: number;\n  requestPayloadLengthInBytes: number;\n  responsePayloadLengthInBytes: number;\n};\n\n/**\n * This type contains diagnostic information regarding a single metadata request to server.\n */\nexport interface MetadataLookUpDiagnostic {\n  activityId: string;\n  startTimeUTCInMs: number;\n  durationInMs: number;\n  operationType?: OperationType;\n  resourceType?: ResourceType;\n  metaDataType: MetadataLookUpType;\n  requestPayloadLengthInBytes: number;\n  responsePayloadLengthInBytes: number;\n}\n\n/**\n * This type captures diagnostic information regarding a failed request to server api.\n */\nexport interface FailedRequestAttemptDiagnostic {\n  attemptNumber: number;\n  activityId: string;\n  startTimeUTCInMs: number;\n  durationInMs: number;\n  operationType?: OperationType;\n  resourceType?: ResourceType;\n  statusCode: number;\n  substatusCode?: number;\n  requestPayloadLengthInBytes: number;\n  responsePayloadLengthInBytes: number;\n}\n\n/**\n * @hidden\n * Represents the diagnostics information for encryption operations.\n */\nexport interface EncryptionDiagnostics {\n  /** shows start time, duration and properties count for encryption*/\n  encryptContent: Record<string, any>;\n  /** shows start time, duration and properties count for decryption*/\n  decryptContent: Record<string, any>;\n  /** represents total processing duration for encryption/decryption*/\n  processingDurationInMs: number;\n}\n\n/**\n * This is enum for Type of Metadata lookups possible.\n */\nexport enum MetadataLookUpType {\n  PartitionKeyRangeLookUp = \"PARTITION_KEY_RANGE_LOOK_UP\",\n  DatabaseAccountLookUp = \"DATABASE_ACCOUNT_LOOK_UP\",\n  QueryPlanLookUp = \"QUERY_PLAN_LOOK_UP\",\n  DatabaseLookUp = \"DATABASE_LOOK_UP\",\n  ContainerLookUp = \"CONTAINER_LOOK_UP\",\n}\n\n/**\n * This is a collection type for all client side diagnostic information.\n */\nexport type ClientSideRequestStatistics = {\n  /**\n   * This is the UTC timestamp for start of client operation.\n   */\n  requestStartTimeUTCInMs: number;\n  /**\n   * This is the duration in milli seconds taken by client operation.\n   */\n  requestDurationInMs: number;\n  /**\n   * This is the list of Location Endpoints contacted during the client operation.\n   */\n  locationEndpointsContacted: string[];\n  /**\n   * This field captures diagnostic information for retries happened during client operation.\n   */\n  retryDiagnostics: RetryDiagnostics;\n  /**\n   * This field captures diagnostic information for meta data lookups happened during client operation.\n   */\n  metadataDiagnostics: MetadataLookUpDiagnostics;\n  /**\n   * These are the statistics for main point look operation.\n   */\n  gatewayStatistics: GatewayStatistics[];\n  /**\n   * This is the cumulated Request Payload Length n bytes, this includes metadata calls along with the main operation.\n   */\n  totalRequestPayloadLengthInBytes: number;\n  /**\n   * This is the cumulated Response Payload Length n bytes, this includes metadata calls along with the main operation.\n   */\n  totalResponsePayloadLengthInBytes: number;\n  /**\n   * This field captures diagnostic information for encryption/decryption happened during CRUD operation if encryption is enabled.\n   */\n  encryptionDiagnostics?: EncryptionDiagnostics;\n};\n\nexport function getRootNode(node: DiagnosticNodeInternal): DiagnosticNodeInternal {\n  if (node.parent) return getRootNode(node.parent);\n  else return node;\n}\n\n/**\n * Represents a tree like structure, for capturing diagnostic information.\n */\nexport interface DiagnosticNode {\n  id: string;\n  nodeType: string;\n  children: DiagnosticNode[];\n  data: { [key: string]: any };\n  startTimeUTCInMs: number;\n  durationInMs: number;\n}\n"]}
+{"version":3,"file":"CosmosDiagnostics.js","sourceRoot":"","sources":["../../src/CosmosDiagnostics.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAOlC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,OAAO,iBAAiB;IACZ,2BAA2B,CAA8B;IACzD,cAAc,CAAiB;IAC/B,YAAY,CAA0B;IACtD;;OAEG;IACH,YACE,2BAAwD,EACxD,cAA+B,EAC/B,YAAqC;QAErC,IAAI,CAAC,2BAA2B,GAAG,2BAA2B,CAAC;QAC/D,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;CACF;AA0ID;;GAEG;AACH,MAAM,CAAN,IAAY,kBAMX;AAND,WAAY,kBAAkB;IAC5B,6EAAuD,CAAA;IACvD,wEAAkD,CAAA;IAClD,4DAAsC,CAAA;IACtC,yDAAmC,CAAA;IACnC,2DAAqC,CAAA;AACvC,CAAC,EANW,kBAAkB,KAAlB,kBAAkB,QAM7B;AA4CD,MAAM,UAAU,WAAW,CAAC,IAA4B;IACtD,IAAI,IAAI,CAAC,MAAM;QAAE,OAAO,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;;QAC5C,OAAO,IAAI,CAAC;AACnB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { OperationType, ResourceType } from \"./common/index.js\";\nimport type { CosmosDbDiagnosticLevel } from \"./diagnostics/CosmosDbDiagnosticLevel.js\";\nimport type { DiagnosticNodeInternal } from \"./diagnostics/DiagnosticNodeInternal.js\";\nimport type { ConsistencyLevel } from \"./documents/index.js\";\n\n/**\n *  * This is a Cosmos Diagnostic type that holds collected diagnostic information during a client operations. ie. Item.read(), Container.create().\n * It has three members -\n * 1. `clientSideRequestStatistics` member contains aggregate diagnostic information, including -\n *   - metadata lookups. Here all the server requests, apart from the final intended resource are considered as metadata calls.\n *    i.e. for item.read(id), if the client makes server call to discover endpoints it would be considered as metadata call.\n *   - retries\n *   - endpoints contacted.\n *   - request, response payload stats.\n *   - gatewayStatistics - Information corresponding to main operation. For example during Item.read(), the client might perform many operations\n *    i.e. metadata lookup etc, but gatewayStatistics represents the diagnostics information for actual read operation.\n *\n * 2. diagnosticNode - Is a tree like structure which captures detailed diagnostic information. By default it is disabled, and is intended to be\n * used only for debugging on non production environments. The kind of details captured in diagnosticNode is controlled by `CosmosDbDiagnosticLevel`.\n * - CosmosDbDiagnosticLevel.info - Is default value. In this level only clientSideRequestStatistics are captured. Is is meant for production environments.\n * - CosmosDbDiagnosticLevel.debug - Captures diagnosticNode and clientConfig. No request and response payloads are captured. Is not meant to be used\n * in production environment.\n * - CosmosDbDiagnosticLevel.debug-unsafe - In addition to data captured in CosmosDbDiagnosticLevel.debug, also captures request and response payloads.\n * Is not meant to be used in production environment.\n * 3. clientConfig - Captures information related to how client was configured during initialization.\n */\nexport class CosmosDiagnostics {\n  public readonly clientSideRequestStatistics: ClientSideRequestStatistics;\n  public readonly diagnosticNode: DiagnosticNode;\n  public readonly clientConfig?: ClientConfigDiagnostic;\n  /**\n   * @internal\n   */\n  constructor(\n    clientSideRequestStatistics: ClientSideRequestStatistics,\n    diagnosticNode?: DiagnosticNode,\n    clientConfig?: ClientConfigDiagnostic,\n  ) {\n    this.clientSideRequestStatistics = clientSideRequestStatistics;\n    this.diagnosticNode = diagnosticNode;\n    this.clientConfig = clientConfig;\n  }\n}\n\n/**\n * This type holds information related to initialization of `CosmosClient`\n */\nexport type ClientConfigDiagnostic = {\n  /**\n   * End point configured during client initialization.\n   */\n  endpoint: string;\n  /**\n   * True if `resourceTokens` was supplied during client initialization.\n   */\n  resourceTokensConfigured: boolean;\n  /**\n   * True if `tokenProvider` was supplied during client initialization.\n   */\n  tokenProviderConfigured: boolean;\n  /**\n   * True if `aadCredentials` was supplied during client initialization.\n   */\n  aadCredentialsConfigured: boolean;\n  /**\n   * True if `connectionPolicy` was supplied during client initialization.\n   */\n  connectionPolicyConfigured: boolean;\n  /**\n   * `consistencyLevel` supplied during client initialization.\n   */\n  consistencyLevel?: keyof typeof ConsistencyLevel;\n  /**\n   * `defaultHeaders` supplied during client initialization.\n   */\n  defaultHeaders?: { [key: string]: any };\n  /**\n   * True if `connectionPolicy` were supplied during client initialization.\n   */\n  agentConfigured: boolean;\n  /**\n   * `userAgentSuffix` supplied during client initialization.\n   */\n  userAgentSuffix: string;\n  /**\n   * `diagnosticLevel` supplied during client initialization.\n   */\n  diagnosticLevel?: CosmosDbDiagnosticLevel;\n  /**\n   * True if `plugins` were supplied during client initialization.\n   */\n  pluginsConfigured: boolean;\n  /**\n   * SDK version\n   */\n  sDKVersion: string;\n  /**\n   * True if `aadScope` were supplied during client initialization.\n   */\n  aadScopeOverride?: boolean;\n};\n\n/**\n * This type contains diagnostic information regarding all metadata request to server during an CosmosDB client operation.\n */\nexport type MetadataLookUpDiagnostics = {\n  metadataLookups: MetadataLookUpDiagnostic[];\n};\n\n/**\n * This type captures diagnostic information regarding retries attempt during an CosmosDB client operation.\n */\nexport type RetryDiagnostics = {\n  failedAttempts: FailedRequestAttemptDiagnostic[];\n};\n\nexport type GatewayStatistics = {\n  /**\n   * This is the activityId for request, made to server for fetching the requested resource. (As opposed to other potential meta data requests)\n   */\n  activityId?: string;\n  /*\n   * The correlated activity ID is added in the header of every request made to the backend for a query. This aids in troubleshooting by linking all\n   * requests associated with a particular query.\n   */\n  correlatedActivityId?: string;\n  startTimeUTCInMs: number;\n  durationInMs: number;\n  operationType?: OperationType;\n  resourceType?: ResourceType;\n  statusCode?: number;\n  subStatusCode?: number;\n  requestCharge?: number;\n  requestPayloadLengthInBytes: number;\n  responsePayloadLengthInBytes: number;\n};\n\n/**\n * This type contains diagnostic information regarding a single metadata request to server.\n */\nexport interface MetadataLookUpDiagnostic {\n  activityId: string;\n  startTimeUTCInMs: number;\n  durationInMs: number;\n  operationType?: OperationType;\n  resourceType?: ResourceType;\n  metaDataType: MetadataLookUpType;\n  requestPayloadLengthInBytes: number;\n  responsePayloadLengthInBytes: number;\n}\n\n/**\n * This type captures diagnostic information regarding a failed request to server api.\n */\nexport interface FailedRequestAttemptDiagnostic {\n  attemptNumber: number;\n  activityId: string;\n  startTimeUTCInMs: number;\n  durationInMs: number;\n  operationType?: OperationType;\n  resourceType?: ResourceType;\n  statusCode: number;\n  substatusCode?: number;\n  requestPayloadLengthInBytes: number;\n  responsePayloadLengthInBytes: number;\n}\n\n/**\n * @hidden\n * Represents the diagnostics information for encryption operations.\n */\nexport interface EncryptionDiagnostics {\n  /** shows start time, duration and properties count for encryption*/\n  encryptContent: Record<string, any>;\n  /** shows start time, duration and properties count for decryption*/\n  decryptContent: Record<string, any>;\n  /** represents total processing duration for encryption/decryption*/\n  processingDurationInMs: number;\n}\n\n/**\n * This is enum for Type of Metadata lookups possible.\n */\nexport enum MetadataLookUpType {\n  PartitionKeyRangeLookUp = \"PARTITION_KEY_RANGE_LOOK_UP\",\n  DatabaseAccountLookUp = \"DATABASE_ACCOUNT_LOOK_UP\",\n  QueryPlanLookUp = \"QUERY_PLAN_LOOK_UP\",\n  DatabaseLookUp = \"DATABASE_LOOK_UP\",\n  ContainerLookUp = \"CONTAINER_LOOK_UP\",\n}\n\n/**\n * This is a collection type for all client side diagnostic information.\n */\nexport type ClientSideRequestStatistics = {\n  /**\n   * This is the UTC timestamp for start of client operation.\n   */\n  requestStartTimeUTCInMs: number;\n  /**\n   * This is the duration in milli seconds taken by client operation.\n   */\n  requestDurationInMs: number;\n  /**\n   * This is the list of Location Endpoints contacted during the client operation.\n   */\n  locationEndpointsContacted: string[];\n  /**\n   * This field captures diagnostic information for retries happened during client operation.\n   */\n  retryDiagnostics: RetryDiagnostics;\n  /**\n   * This field captures diagnostic information for meta data lookups happened during client operation.\n   */\n  metadataDiagnostics: MetadataLookUpDiagnostics;\n  /**\n   * These are the statistics for main point look operation.\n   */\n  gatewayStatistics: GatewayStatistics[];\n  /**\n   * This is the cumulated Request Payload Length n bytes, this includes metadata calls along with the main operation.\n   */\n  totalRequestPayloadLengthInBytes: number;\n  /**\n   * This is the cumulated Response Payload Length n bytes, this includes metadata calls along with the main operation.\n   */\n  totalResponsePayloadLengthInBytes: number;\n  /**\n   * This field captures diagnostic information for encryption/decryption happened during CRUD operation if encryption is enabled.\n   */\n  encryptionDiagnostics?: EncryptionDiagnostics;\n};\n\nexport function getRootNode(node: DiagnosticNodeInternal): DiagnosticNodeInternal {\n  if (node.parent) return getRootNode(node.parent);\n  else return node;\n}\n\n/**\n * Represents a tree like structure, for capturing diagnostic information.\n */\nexport interface DiagnosticNode {\n  id: string;\n  nodeType: string;\n  children: DiagnosticNode[];\n  data: { [key: string]: any };\n  startTimeUTCInMs: number;\n  durationInMs: number;\n}\n"]}

package/dist/esm/GlobalEndpointManagerOptions.d.ts

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

package/dist/esm/GlobalEndpointManagerOptions.d.ts.map

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

package/dist/esm/GlobalEndpointManagerOptions.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=GlobalEndpointManagerOptions.js.map

package/dist/esm/GlobalEndpointManagerOptions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GlobalEndpointManagerOptions.js","sourceRoot":"","sources":["../../src/GlobalEndpointManagerOptions.ts"],"names":[],"mappings":"","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { ResourceType, OperationType } from \"./common/index.js\";\nimport type { DiagnosticNodeInternal } from \"./diagnostics/DiagnosticNodeInternal.js\";\n\n/**\n * @internal\n */\nexport interface ResolveServiceEndpointOptions {\n  diagnosticNode: DiagnosticNodeInternal;\n  resourceType: ResourceType;\n  operationType: OperationType;\n  /**\n   * This is to indicate the starting index for selecting servers.\n   */\n  startServiceEndpointIndex: number;\n  /**\n   * Excludes one or more Azure regions for the operation.\n   * <p>This option is only applied when enableEndPointDiscovery is set to true.</p>\n   */\n  excludedLocations?: string[];\n}\n"]}

package/dist/esm/PartitionKeyRangeFailoverInfo.js

@@ -8,15 +8,18 @@ import semaphore from "semaphore";
  * It tracks the current endpoint, failed endpoints, and the number of consecutive read/write request failures.
  */
 export class PartitionKeyRangeFailoverInfo {
+    failedEndPoints = [];
+    currentEndPoint;
+    consecutiveReadRequestFailureCount = 0;
+    consecutiveWriteRequestFailureCount = 0;
+    firstRequestFailureTime = Date.now();
+    lastRequestFailureTime = Date.now();
+    failureCountSemaphore;
+    tryMoveNextLocationSemaphore;
     /**
      * @internal
      */
     constructor(currentEndpoint) {
-        this.failedEndPoints = [];
-        this.consecutiveReadRequestFailureCount = 0;
-        this.consecutiveWriteRequestFailureCount = 0;
-        this.firstRequestFailureTime = Date.now();
-        this.lastRequestFailureTime = Date.now();
         this.currentEndPoint = currentEndpoint;
         this.failureCountSemaphore = semaphore(1);
         this.tryMoveNextLocationSemaphore = semaphore(1);

package/dist/esm/PartitionKeyRangeFailoverInfo.js.map

@@ -1 +1 @@
-{"version":3,"file":"PartitionKeyRangeFailoverInfo.js","sourceRoot":"","sources":["../../src/PartitionKeyRangeFailoverInfo.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,SAAS,EAA0B,MAAM,YAAY,CAAC;AAC/D,OAAO,SAAS,MAAM,WAAW,CAAC;AAElC;;;;GAIG;AACH,MAAM,OAAO,6BAA6B;IAYxC;;OAEG;IACH,YAAY,eAAuB;QAd3B,oBAAe,GAAa,EAAE,CAAC;QAG/B,uCAAkC,GAAW,CAAC,CAAC;QAC/C,wCAAmC,GAAW,CAAC,CAAC;QAChD,4BAAuB,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7C,2BAAsB,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;QASlD,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,qBAAqB,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC1C,IAAI,CAAC,4BAA4B,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;IACnD,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,yCAAyC,CACpD,iBAA0B;QAE1B,MAAM,EAAE,kCAAkC,EAAE,mCAAmC,EAAE,GAC/E,MAAM,IAAI,CAAC,sCAAsC,EAAE,CAAC;QAEtD,OAAO,iBAAiB;YACtB,CAAC,CAAC,kCAAkC,GAAG,SAAS,CAAC,gCAAgC;YACjF,CAAC,CAAC,mCAAmC,GAAG,SAAS,CAAC,iCAAiC,CAAC;IACxF,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,6BAA6B,CACxC,iBAA0B,EAC1B,yBAAiC;QAEjC,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;gBACzC,IAAI,CAAC;oBACH,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAEpF,IACE,yBAAyB,GAAG,sBAAsB;wBAClD,SAAS,CAAC,wCAAwC,EAClD,CAAC;wBACD,IAAI,CAAC,kCAAkC,GAAG,CAAC,CAAC;wBAC5C,IAAI,CAAC,mCAAmC,GAAG,CAAC,CAAC;oBAC/C,CAAC;oBAED,IAAI,iBAAiB,EAAE,CAAC;wBACtB,IAAI,CAAC,kCAAkC,EAAE,CAAC;oBAC5C,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAC7C,CAAC;oBACD,IAAI,CAAC,sBAAsB,GAAG,yBAAyB,CAAC;oBACxD,OAAO,OAAO,EAAE,CAAC;gBACnB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;wBAAS,CAAC;oBACT,6BAA6B;oBAC7B,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;gBACrC,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,mCAAmC;QAI9C,OAAO;YACL,uBAAuB,EAAE,IAAI,CAAC,uBAAuB;YACrD,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;SACpD,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,mBAAmB,CAC9B,SAA4B,EAC5B,cAAsB,EACtB,cAAsC,EACtC,mBAA2B;QAE3B,IAAI,cAAc,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC,GAAG,EAAE;gBAC1C,IAAI,CAAC;oBACH,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;wBACjC,IAAI,IAAI,CAAC,eAAe,KAAK,QAAQ,EAAE,CAAC;4BACtC,SAAS;wBACX,CAAC;wBAED,IAAI,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC5C,SAAS;wBACX,CAAC;wBAED,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;wBAC1C,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;wBAChC,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;oBACvB,CAAC;oBACD,cAAc,CAAC,OAAO,CAAC;wBACrB,6BAA6B,EAAE,wBAAwB,mBAAmB,sBAAsB,IAAI,CAAC,eAAe,kBAAkB,IAAI,CAAC,eAAe,EAAE;qBAC7J,CAAC,CAAC;oBACH,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC;gBACxB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;wBAAS,CAAC;oBACT,IAAI,CAAC,4BAA4B,CAAC,KAAK,EAAE,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iFAAiF;IAC1E,kBAAkB;QACvB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,sCAAsC;QAIlD,OAAO;YACL,kCAAkC,EAAE,IAAI,CAAC,kCAAkC;YAC3E,mCAAmC,EAAE,IAAI,CAAC,mCAAmC;SAC9E,CAAC;IACJ,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { Constants, DiagnosticNodeInternal } from \"./index.js\";\nimport semaphore from \"semaphore\";\n\n/**\n * @hidden\n * This class manages the failover information for partition key ranges in Cosmos DB.\n * It tracks the current endpoint, failed endpoints, and the number of consecutive read/write request failures.\n */\nexport class PartitionKeyRangeFailoverInfo {\n  private failedEndPoints: string[] = [];\n  private currentEndPoint: string;\n\n  private consecutiveReadRequestFailureCount: number = 0;\n  private consecutiveWriteRequestFailureCount: number = 0;\n  private firstRequestFailureTime: number = Date.now();\n  private lastRequestFailureTime: number = Date.now();\n\n  private failureCountSemaphore: semaphore.Semaphore;\n  private tryMoveNextLocationSemaphore: semaphore.Semaphore;\n\n  /**\n   * @internal\n   */\n  constructor(currentEndpoint: string) {\n    this.currentEndPoint = currentEndpoint;\n    this.failureCountSemaphore = semaphore(1);\n    this.tryMoveNextLocationSemaphore = semaphore(1);\n  }\n\n  /**\n   * Checks if the circuit breaker can trigger a partition failover based on the failure counts.\n   * Returns true if the number of consecutive failures exceeds the defined thresholds for read or write requests.\n   */\n  public async canCircuitBreakerTriggerPartitionFailOver(\n    isReadOnlyRequest: boolean,\n  ): Promise<boolean> {\n    const { consecutiveReadRequestFailureCount, consecutiveWriteRequestFailureCount } =\n      await this.snapshotConsecutiveRequestFailureCount();\n\n    return isReadOnlyRequest\n      ? consecutiveReadRequestFailureCount > Constants.ReadRequestFailureCountThreshold\n      : consecutiveWriteRequestFailureCount > Constants.WriteRequestFailureCountThreshold;\n  }\n\n  /**\n   * Increments the failure counts for read or write requests and updates the timestamps.\n   * If the time since the last failure exceeds the reset window, it resets the failure counts.\n   */\n  public async incrementRequestFailureCounts(\n    isReadOnlyRequest: boolean,\n    currentTimeInMilliseconds: number,\n  ): Promise<void> {\n    return new Promise<void>((resolve, reject) => {\n      this.failureCountSemaphore.take(async () => {\n        try {\n          const { lastRequestFailureTime } = await this.snapshotPartitionFailoverTimestamps();\n\n          if (\n            currentTimeInMilliseconds - lastRequestFailureTime >\n            Constants.ConsecutiveFailureCountResetIntervalInMS\n          ) {\n            this.consecutiveReadRequestFailureCount = 0;\n            this.consecutiveWriteRequestFailureCount = 0;\n          }\n\n          if (isReadOnlyRequest) {\n            this.consecutiveReadRequestFailureCount++;\n          } else {\n            this.consecutiveWriteRequestFailureCount++;\n          }\n          this.lastRequestFailureTime = currentTimeInMilliseconds;\n          return resolve();\n        } catch (error) {\n          reject(error);\n        } finally {\n          // Release the semaphore lock\n          this.failureCountSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /**\n   * Returns a snapshot of the first and last request failure timestamps.\n   * This method is used to retrieve the current state of failure timestamps without modifying them.\n   */\n  public async snapshotPartitionFailoverTimestamps(): Promise<{\n    firstRequestFailureTime: number;\n    lastRequestFailureTime: number;\n  }> {\n    return {\n      firstRequestFailureTime: this.firstRequestFailureTime,\n      lastRequestFailureTime: this.lastRequestFailureTime,\n    };\n  }\n\n  /**\n   * Attempts to move to the next available location for the partition key range.\n   * If the current endpoint is the same as the failed endpoint, it tries to find a new endpoint\n   * from the provided list of endpoints. If a new endpoint is found, it updates the current endpoint\n   * and returns true. If no new endpoint is found, it returns false.\n   */\n  public async tryMoveNextLocation(\n    endPoints: readonly string[],\n    failedEndPoint: string,\n    diagnosticNode: DiagnosticNodeInternal,\n    partitionKeyRangeId: string,\n  ): Promise<boolean> {\n    if (failedEndPoint !== this.currentEndPoint) {\n      return true;\n    }\n    return new Promise((resolve, reject) => {\n      this.tryMoveNextLocationSemaphore.take(() => {\n        try {\n          for (const endpoint of endPoints) {\n            if (this.currentEndPoint === endpoint) {\n              continue;\n            }\n\n            if (this.failedEndPoints.includes(endpoint)) {\n              continue;\n            }\n\n            this.failedEndPoints.push(failedEndPoint);\n            this.currentEndPoint = endpoint;\n            return resolve(true);\n          }\n          diagnosticNode.addData({\n            partitionKeyRangeFailoverInfo: `PartitionKeyRangeId: ${partitionKeyRangeId}, failedLocations: ${this.failedEndPoints}, newLocation: ${this.currentEndPoint}`,\n          });\n          return resolve(false);\n        } catch (err) {\n          reject(err);\n        } finally {\n          this.tryMoveNextLocationSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /** Returns the current endpoint being used for partition key range operations.*/\n  public getCurrentEndPoint(): string {\n    return this.currentEndPoint;\n  }\n\n  /**\n   * Returns a snapshot of the current consecutive request failure counts for read and write requests.\n   * This method is used to retrieve the current state of failure counts without modifying them.\n   */\n  private async snapshotConsecutiveRequestFailureCount(): Promise<{\n    consecutiveReadRequestFailureCount: number;\n    consecutiveWriteRequestFailureCount: number;\n  }> {\n    return {\n      consecutiveReadRequestFailureCount: this.consecutiveReadRequestFailureCount,\n      consecutiveWriteRequestFailureCount: this.consecutiveWriteRequestFailureCount,\n    };\n  }\n}\n"]}
+{"version":3,"file":"PartitionKeyRangeFailoverInfo.js","sourceRoot":"","sources":["../../src/PartitionKeyRangeFailoverInfo.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,SAAS,EAA0B,MAAM,YAAY,CAAC;AAC/D,OAAO,SAAS,MAAM,WAAW,CAAC;AAElC;;;;GAIG;AACH,MAAM,OAAO,6BAA6B;IAChC,eAAe,GAAa,EAAE,CAAC;IAC/B,eAAe,CAAS;IAExB,kCAAkC,GAAW,CAAC,CAAC;IAC/C,mCAAmC,GAAW,CAAC,CAAC;IAChD,uBAAuB,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7C,sBAAsB,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;IAE5C,qBAAqB,CAAsB;IAC3C,4BAA4B,CAAsB;IAE1D;;OAEG;IACH,YAAY,eAAuB;QACjC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,qBAAqB,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC1C,IAAI,CAAC,4BAA4B,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;IACnD,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,yCAAyC,CACpD,iBAA0B;QAE1B,MAAM,EAAE,kCAAkC,EAAE,mCAAmC,EAAE,GAC/E,MAAM,IAAI,CAAC,sCAAsC,EAAE,CAAC;QAEtD,OAAO,iBAAiB;YACtB,CAAC,CAAC,kCAAkC,GAAG,SAAS,CAAC,gCAAgC;YACjF,CAAC,CAAC,mCAAmC,GAAG,SAAS,CAAC,iCAAiC,CAAC;IACxF,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,6BAA6B,CACxC,iBAA0B,EAC1B,yBAAiC;QAEjC,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;gBACzC,IAAI,CAAC;oBACH,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAEpF,IACE,yBAAyB,GAAG,sBAAsB;wBAClD,SAAS,CAAC,wCAAwC,EAClD,CAAC;wBACD,IAAI,CAAC,kCAAkC,GAAG,CAAC,CAAC;wBAC5C,IAAI,CAAC,mCAAmC,GAAG,CAAC,CAAC;oBAC/C,CAAC;oBAED,IAAI,iBAAiB,EAAE,CAAC;wBACtB,IAAI,CAAC,kCAAkC,EAAE,CAAC;oBAC5C,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAC7C,CAAC;oBACD,IAAI,CAAC,sBAAsB,GAAG,yBAAyB,CAAC;oBACxD,OAAO,OAAO,EAAE,CAAC;gBACnB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;wBAAS,CAAC;oBACT,6BAA6B;oBAC7B,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;gBACrC,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,mCAAmC;QAI9C,OAAO;YACL,uBAAuB,EAAE,IAAI,CAAC,uBAAuB;YACrD,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;SACpD,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,mBAAmB,CAC9B,SAA4B,EAC5B,cAAsB,EACtB,cAAsC,EACtC,mBAA2B;QAE3B,IAAI,cAAc,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC,GAAG,EAAE;gBAC1C,IAAI,CAAC;oBACH,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;wBACjC,IAAI,IAAI,CAAC,eAAe,KAAK,QAAQ,EAAE,CAAC;4BACtC,SAAS;wBACX,CAAC;wBAED,IAAI,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC5C,SAAS;wBACX,CAAC;wBAED,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;wBAC1C,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;wBAChC,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;oBACvB,CAAC;oBACD,cAAc,CAAC,OAAO,CAAC;wBACrB,6BAA6B,EAAE,wBAAwB,mBAAmB,sBAAsB,IAAI,CAAC,eAAe,kBAAkB,IAAI,CAAC,eAAe,EAAE;qBAC7J,CAAC,CAAC;oBACH,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC;gBACxB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;wBAAS,CAAC;oBACT,IAAI,CAAC,4BAA4B,CAAC,KAAK,EAAE,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iFAAiF;IAC1E,kBAAkB;QACvB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,sCAAsC;QAIlD,OAAO;YACL,kCAAkC,EAAE,IAAI,CAAC,kCAAkC;YAC3E,mCAAmC,EAAE,IAAI,CAAC,mCAAmC;SAC9E,CAAC;IACJ,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { Constants, DiagnosticNodeInternal } from \"./index.js\";\nimport semaphore from \"semaphore\";\n\n/**\n * @hidden\n * This class manages the failover information for partition key ranges in Cosmos DB.\n * It tracks the current endpoint, failed endpoints, and the number of consecutive read/write request failures.\n */\nexport class PartitionKeyRangeFailoverInfo {\n  private failedEndPoints: string[] = [];\n  private currentEndPoint: string;\n\n  private consecutiveReadRequestFailureCount: number = 0;\n  private consecutiveWriteRequestFailureCount: number = 0;\n  private firstRequestFailureTime: number = Date.now();\n  private lastRequestFailureTime: number = Date.now();\n\n  private failureCountSemaphore: semaphore.Semaphore;\n  private tryMoveNextLocationSemaphore: semaphore.Semaphore;\n\n  /**\n   * @internal\n   */\n  constructor(currentEndpoint: string) {\n    this.currentEndPoint = currentEndpoint;\n    this.failureCountSemaphore = semaphore(1);\n    this.tryMoveNextLocationSemaphore = semaphore(1);\n  }\n\n  /**\n   * Checks if the circuit breaker can trigger a partition failover based on the failure counts.\n   * Returns true if the number of consecutive failures exceeds the defined thresholds for read or write requests.\n   */\n  public async canCircuitBreakerTriggerPartitionFailOver(\n    isReadOnlyRequest: boolean,\n  ): Promise<boolean> {\n    const { consecutiveReadRequestFailureCount, consecutiveWriteRequestFailureCount } =\n      await this.snapshotConsecutiveRequestFailureCount();\n\n    return isReadOnlyRequest\n      ? consecutiveReadRequestFailureCount > Constants.ReadRequestFailureCountThreshold\n      : consecutiveWriteRequestFailureCount > Constants.WriteRequestFailureCountThreshold;\n  }\n\n  /**\n   * Increments the failure counts for read or write requests and updates the timestamps.\n   * If the time since the last failure exceeds the reset window, it resets the failure counts.\n   */\n  public async incrementRequestFailureCounts(\n    isReadOnlyRequest: boolean,\n    currentTimeInMilliseconds: number,\n  ): Promise<void> {\n    return new Promise<void>((resolve, reject) => {\n      this.failureCountSemaphore.take(async () => {\n        try {\n          const { lastRequestFailureTime } = await this.snapshotPartitionFailoverTimestamps();\n\n          if (\n            currentTimeInMilliseconds - lastRequestFailureTime >\n            Constants.ConsecutiveFailureCountResetIntervalInMS\n          ) {\n            this.consecutiveReadRequestFailureCount = 0;\n            this.consecutiveWriteRequestFailureCount = 0;\n          }\n\n          if (isReadOnlyRequest) {\n            this.consecutiveReadRequestFailureCount++;\n          } else {\n            this.consecutiveWriteRequestFailureCount++;\n          }\n          this.lastRequestFailureTime = currentTimeInMilliseconds;\n          return resolve();\n        } catch (error) {\n          reject(error);\n        } finally {\n          // Release the semaphore lock\n          this.failureCountSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /**\n   * Returns a snapshot of the first and last request failure timestamps.\n   * This method is used to retrieve the current state of failure timestamps without modifying them.\n   */\n  public async snapshotPartitionFailoverTimestamps(): Promise<{\n    firstRequestFailureTime: number;\n    lastRequestFailureTime: number;\n  }> {\n    return {\n      firstRequestFailureTime: this.firstRequestFailureTime,\n      lastRequestFailureTime: this.lastRequestFailureTime,\n    };\n  }\n\n  /**\n   * Attempts to move to the next available location for the partition key range.\n   * If the current endpoint is the same as the failed endpoint, it tries to find a new endpoint\n   * from the provided list of endpoints. If a new endpoint is found, it updates the current endpoint\n   * and returns true. If no new endpoint is found, it returns false.\n   */\n  public async tryMoveNextLocation(\n    endPoints: readonly string[],\n    failedEndPoint: string,\n    diagnosticNode: DiagnosticNodeInternal,\n    partitionKeyRangeId: string,\n  ): Promise<boolean> {\n    if (failedEndPoint !== this.currentEndPoint) {\n      return true;\n    }\n    return new Promise((resolve, reject) => {\n      this.tryMoveNextLocationSemaphore.take(() => {\n        try {\n          for (const endpoint of endPoints) {\n            if (this.currentEndPoint === endpoint) {\n              continue;\n            }\n\n            if (this.failedEndPoints.includes(endpoint)) {\n              continue;\n            }\n\n            this.failedEndPoints.push(failedEndPoint);\n            this.currentEndPoint = endpoint;\n            return resolve(true);\n          }\n          diagnosticNode.addData({\n            partitionKeyRangeFailoverInfo: `PartitionKeyRangeId: ${partitionKeyRangeId}, failedLocations: ${this.failedEndPoints}, newLocation: ${this.currentEndPoint}`,\n          });\n          return resolve(false);\n        } catch (err) {\n          reject(err);\n        } finally {\n          this.tryMoveNextLocationSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /** Returns the current endpoint being used for partition key range operations.*/\n  public getCurrentEndPoint(): string {\n    return this.currentEndPoint;\n  }\n\n  /**\n   * Returns a snapshot of the current consecutive request failure counts for read and write requests.\n   * This method is used to retrieve the current state of failure counts without modifying them.\n   */\n  private async snapshotConsecutiveRequestFailureCount(): Promise<{\n    consecutiveReadRequestFailureCount: number;\n    consecutiveWriteRequestFailureCount: number;\n  }> {\n    return {\n      consecutiveReadRequestFailureCount: this.consecutiveReadRequestFailureCount,\n      consecutiveWriteRequestFailureCount: this.consecutiveWriteRequestFailureCount,\n    };\n  }\n}\n"]}

package/dist/esm/bulk/Batcher.js

@@ -12,6 +12,17 @@ import { getCurrentTimestampInMs } from "../utils/time.js";
  * @hidden
  */
 export class Batcher {
+    batchOperationsList;
+    currentSize;
+    toBeDispatched;
+    executor;
+    retrier;
+    diagnosticLevel;
+    encryptionEnabled;
+    encryptionProcessor;
+    clientConfigDiagnostics;
+    limiter;
+    processedOperationCountRef;
     constructor(limiter, executor, retrier, diagnosticLevel, encryptionEnabled, clientConfig, encryptionProcessor, processedOperationCountRef) {
         this.limiter = limiter;
         this.batchOperationsList = [];
@@ -59,7 +70,6 @@ export class Batcher {
      * Handles retries for failed operations and updates the ordered response.
      */
     async dispatch(partitionMetric) {
-        var _a, _b;
         this.toBeDispatched = true;
         const startTime = getCurrentTimestampInMs();
         const diagnosticNode = new DiagnosticNodeInternal(this.diagnosticLevel, DiagnosticNodeType.BATCH_REQUEST, null);
@@ -67,10 +77,10 @@ export class Batcher {
             const response = await this.executor(this.batchOperationsList, diagnosticNode);
             const hasThrottles = 1;
             const noThrottle = 0;
-            const numThrottle = ((_a = response === null || response === void 0 ? void 0 : response.results) === null || _a === void 0 ? void 0 : _a.some((result) => "code" in result && result.code === StatusCodes.TooManyRequests))
+            const numThrottle = response?.results?.some((result) => "code" in result && result.code === StatusCodes.TooManyRequests)
                 ? hasThrottles
                 : noThrottle;
-            const splitOrMerge = ((_b = response === null || response === void 0 ? void 0 : response.results) === null || _b === void 0 ? void 0 : _b.some((result) => "code" in result && result.code === StatusCodes.Gone))
+            const splitOrMerge = response?.results?.some((result) => "code" in result && result.code === StatusCodes.Gone)
                 ? true
                 : false;
             if (splitOrMerge) {

package/dist/esm/bulk/Batcher.js.map

@@ -1 +1 @@
-{"version":3,"file":"Batcher.js","sourceRoot":"","sources":["../../../src/bulk/Batcher.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAEvD,OAAO,EACL,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,0CAA0C,CAAC;AAGlD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,OAAO,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AAC/D,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAK3D;;;;GAIG;AAEH,MAAM,OAAO,OAAO;IAalB,YACE,OAAqB,EACrB,QAAyB,EACzB,OAAsB,EACtB,eAAwC,EACxC,iBAA0B,EAC1B,YAAoC,EACpC,mBAAwC,EACxC,0BAA6C;QAE7C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,uBAAuB,GAAG,YAAY,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,IAAI,CAAC,0BAA0B,GAAG,0BAA0B,CAAC;IAC/D,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,SAAwB;QACpC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,0BAA0B,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;YAChC,MAAM,IAAI,aAAa,CAAC,kCAAkC,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,KAAK,SAAS,CAAC,sBAAsB,EAAE,CAAC;YACzE,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,oBAAoB,GAAG,0BAA0B,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAClF,IACE,IAAI,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC;YACnC,IAAI,CAAC,WAAW,GAAG,oBAAoB,GAAG,SAAS,CAAC,oCAAoC,EACxF,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,oBAAoB,CAAC;QACzC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,OAAO;QACZ,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,QAAQ,CAAC,eAAgC;;QACpD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,cAAc,GAAG,IAAI,sBAAsB,CAC/C,IAAI,CAAC,eAAe,EACpB,kBAAkB,CAAC,aAAa,EAChC,IAAI,CACL,CAAC;QACF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC;YAE/E,MAAM,YAAY,GAAG,CAAC,CAAC;YACvB,MAAM,UAAU,GAAG,CAAC,CAAC;YACrB,MAAM,WAAW,GAAG,CAAA,MAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,OAAO,0CAAE,IAAI,CACzC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,eAAe,CAC5E;gBACC,CAAC,CAAC,YAAY;gBACd,CAAC,CAAC,UAAU,CAAC;YACf,MAAM,YAAY,GAAG,CAAA,MAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,OAAO,0CAAE,IAAI,CAC1C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,IAAI,CACjE;gBACC,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,KAAK,CAAC;YACV,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YACrE,CAAC;YACD,eAAe,CAAC,GAAG,CACjB,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAC/B,uBAAuB,EAAE,GAAG,SAAS,EACrC,WAAW,CACZ,CAAC;YACF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,mBAAmB,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAChD,IAAI,mBAAmB,YAAY,aAAa,EAAE,CAAC;oBACjD,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,gBAAgB,CAAC,WAAW,CAAC,WAAW,CAC1E,mBAAmB,EACnB,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAC1C,CAAC;oBACF,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;wBACzE,SAAS;oBACX,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC;oBACH,IAAI,IAAI,CAAC,iBAAiB,IAAI,mBAAmB,CAAC,YAAY,EAAE,CAAC;wBAC/D,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,0BAA0B,CAClE,SAAS,CAAC,UAAU,CAAC,2BAA2B,CACjD,CAAC;wBACF,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,wBAAwB,EAAE,GACrD,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;wBAC3E,mBAAmB,CAAC,YAAY,GAAG,aAAa,CAAC;wBACjD,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,wBAAwB,CAChE,SAAS,CAAC,UAAU,CAAC,2BAA2B,EAChD,wBAAwB,CACzB,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,sGAAsG;oBACtG,IAAI,SAAS,CAAC,cAAc,CAAC,aAAa,KAAK,MAAM,EAAE,CAAC;wBACtD,MAAM,eAAe,GAAG,IAAI,aAAa,CACvC,QAAQ,SAAS,CAAC,cAAc,CAAC,aAAa,+DAA+D,KAAK,CAAC,OAAO,EAAE,CAC7H,CAAC;wBACF,eAAe,CAAC,IAAI,GAAG,WAAW,CAAC,kBAAkB,CAAC;wBACtD,MAAM,eAAe,CAAC;oBACxB,CAAC;gBACH,CAAC;gBACD,SAAS,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;gBAC9D,mBAAmB,CAAC,WAAW,GAAG,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,YAAY,CACtF,IAAI,CAAC,uBAAuB,CAC7B,CAAC;gBACF,MAAM,gBAAgB,GAAwB;oBAC5C,cAAc,EAAE,SAAS,CAAC,yBAAyB;iBACpD,CAAC;gBACF,IAAI,mBAAmB,YAAY,aAAa,EAAE,CAAC;oBACjD,gBAAgB,CAAC,KAAK,GAAG,mBAAmB,CAAC;gBAC/C,CAAC;qBAAM,CAAC;oBACN,gBAAgB,CAAC,QAAQ,GAAG,mBAAmB,CAAC;gBAClD,CAAC;gBACD,SAAS,CAAC,gBAAgB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;gBACtD,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YAC1C,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,6CAA6C;YAC7C,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACjD,MAAM,QAAQ,GAAwB;oBACpC,cAAc,EAAE,SAAS,CAAC,cAAc;oBACxC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE;wBACrD,IAAI,EAAE,WAAW,CAAC,mBAAmB;wBACrC,WAAW,EAAE,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,YAAY,CACjE,IAAI,CAAC,uBAAuB,CAC7B;qBACF,CAAC;iBACH,CAAC;gBACF,SAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC1C,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YAC1C,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,uBAAuB;YACvB,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAC;QAChC,CAAC;IACH,CAAC;IAEM,aAAa;QAClB,OAAO,IAAI,CAAC,mBAAmB,CAAC;IAClC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Constants } from \"../common/constants.js\";\nimport { StatusCodes } from \"../common/statusCodes.js\";\nimport type { CosmosDbDiagnosticLevel } from \"../diagnostics/CosmosDbDiagnosticLevel.js\";\nimport {\n  DiagnosticNodeInternal,\n  DiagnosticNodeType,\n} from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport type { EncryptionProcessor } from \"../encryption/EncryptionProcessor.js\";\nimport type { ClientConfigDiagnostic } from \"../index.js\";\nimport { ErrorResponse } from \"../index.js\";\nimport type { ExecuteCallback, RetryCallback, BulkOperationResult } from \"../utils/batch.js\";\nimport { calculateObjectSizeInBytes } from \"../utils/batch.js\";\nimport { getCurrentTimestampInMs } from \"../utils/time.js\";\nimport type { PartitionMetric } from \"./PartitionMetric.js\";\nimport type { ItemOperation } from \"./index.js\";\nimport type { LimiterQueue } from \"./Limiter.js\";\n\n/**\n * Maintains a batch of operations and dispatches it as a unit of work.\n * Execution of the request is done by the @see {@link ExecuteCallback} and retry is done by the @see {@link RetryCallback}.\n * @hidden\n */\n\nexport class Batcher {\n  private batchOperationsList: ItemOperation[];\n  private currentSize: number;\n  private toBeDispatched: boolean;\n  private readonly executor: ExecuteCallback;\n  private readonly retrier: RetryCallback;\n  private readonly diagnosticLevel: CosmosDbDiagnosticLevel;\n  private readonly encryptionEnabled: boolean;\n  private readonly encryptionProcessor: EncryptionProcessor;\n  private readonly clientConfigDiagnostics: ClientConfigDiagnostic;\n  private readonly limiter: LimiterQueue;\n  private processedOperationCountRef: { count: number };\n\n  constructor(\n    limiter: LimiterQueue,\n    executor: ExecuteCallback,\n    retrier: RetryCallback,\n    diagnosticLevel: CosmosDbDiagnosticLevel,\n    encryptionEnabled: boolean,\n    clientConfig: ClientConfigDiagnostic,\n    encryptionProcessor: EncryptionProcessor,\n    processedOperationCountRef: { count: number },\n  ) {\n    this.limiter = limiter;\n    this.batchOperationsList = [];\n    this.executor = executor;\n    this.retrier = retrier;\n    this.diagnosticLevel = diagnosticLevel;\n    this.encryptionEnabled = encryptionEnabled;\n    this.encryptionProcessor = encryptionProcessor;\n    this.clientConfigDiagnostics = clientConfig;\n    this.currentSize = 0;\n    this.toBeDispatched = false;\n    this.processedOperationCountRef = processedOperationCountRef;\n  }\n\n  /**\n   * Attempts to add an operation to the current batch.\n   * Returns false if the batch is full or already dispatched.\n   */\n  public tryAdd(operation: ItemOperation): boolean {\n    if (this.toBeDispatched) {\n      return false;\n    }\n    if (!operation) {\n      throw new ErrorResponse(\"Operation is not defined\");\n    }\n    if (!operation.operationContext) {\n      throw new ErrorResponse(\"Operation context is not defined\");\n    }\n    if (this.batchOperationsList.length === Constants.MaxBulkOperationsCount) {\n      return false;\n    }\n    const currentOperationSize = calculateObjectSizeInBytes(operation.operationInput);\n    if (\n      this.batchOperationsList.length > 0 &&\n      this.currentSize + currentOperationSize > Constants.DefaultMaxBulkRequestBodySizeInBytes\n    ) {\n      return false;\n    }\n\n    this.currentSize += currentOperationSize;\n    this.batchOperationsList.push(operation);\n    return true;\n  }\n\n  public isEmpty(): boolean {\n    return this.batchOperationsList.length === 0;\n  }\n\n  /**\n   * Dispatches the current batch of operations.\n   * Handles retries for failed operations and updates the ordered response.\n   */\n  public async dispatch(partitionMetric: PartitionMetric): Promise<void> {\n    this.toBeDispatched = true;\n    const startTime = getCurrentTimestampInMs();\n    const diagnosticNode = new DiagnosticNodeInternal(\n      this.diagnosticLevel,\n      DiagnosticNodeType.BATCH_REQUEST,\n      null,\n    );\n    try {\n      const response = await this.executor(this.batchOperationsList, diagnosticNode);\n\n      const hasThrottles = 1;\n      const noThrottle = 0;\n      const numThrottle = response?.results?.some(\n        (result) => \"code\" in result && result.code === StatusCodes.TooManyRequests,\n      )\n        ? hasThrottles\n        : noThrottle;\n      const splitOrMerge = response?.results?.some(\n        (result) => \"code\" in result && result.code === StatusCodes.Gone,\n      )\n        ? true\n        : false;\n      if (splitOrMerge) {\n        await this.limiter.pauseAndClear(StatusCodes.Gone, diagnosticNode);\n      }\n      partitionMetric.add(\n        this.batchOperationsList.length,\n        getCurrentTimestampInMs() - startTime,\n        numThrottle,\n      );\n      for (let i = 0; i < response.operations.length; i++) {\n        const operation = response.operations[i];\n        const bulkOperationResult = response.results[i];\n        if (bulkOperationResult instanceof ErrorResponse) {\n          const shouldRetry = await operation.operationContext.retryPolicy.shouldRetry(\n            bulkOperationResult,\n            operation.operationContext.diagnosticNode,\n          );\n          if (shouldRetry) {\n            await this.retrier(operation, operation.operationContext.diagnosticNode);\n            continue;\n          }\n        }\n        try {\n          if (this.encryptionEnabled && bulkOperationResult.resourceBody) {\n            operation.operationContext.diagnosticNode.beginEncryptionDiagnostics(\n              Constants.Encryption.DiagnosticsDecryptOperation,\n            );\n            const { body: decryptedBody, propertiesDecryptedCount } =\n              await this.encryptionProcessor.decrypt(bulkOperationResult.resourceBody);\n            bulkOperationResult.resourceBody = decryptedBody;\n            operation.operationContext.diagnosticNode.endEncryptionDiagnostics(\n              Constants.Encryption.DiagnosticsDecryptOperation,\n              propertiesDecryptedCount,\n            );\n          }\n        } catch (error) {\n          // if decryption fails after successful write operation, fail the operation with internal server error\n          if (operation.operationInput.operationType !== \"Read\") {\n            const decryptionError = new ErrorResponse(\n              `Item ${operation.operationInput.operationType} operation was successful but response decryption failed: + ${error.message}`,\n            );\n            decryptionError.code = StatusCodes.ServiceUnavailable;\n            throw decryptionError;\n          }\n        }\n        operation.operationContext.addDiagnosticChild(diagnosticNode);\n        bulkOperationResult.diagnostics = operation.operationContext.diagnosticNode.toDiagnostic(\n          this.clientConfigDiagnostics,\n        );\n        const bulkItemResponse: BulkOperationResult = {\n          operationInput: operation.unencryptedOperationInput,\n        };\n        if (bulkOperationResult instanceof ErrorResponse) {\n          bulkItemResponse.error = bulkOperationResult;\n        } else {\n          bulkItemResponse.response = bulkOperationResult;\n        }\n        operation.operationContext.complete(bulkItemResponse);\n        this.processedOperationCountRef.count++;\n      }\n    } catch (error) {\n      // Mark all operations in the batch as failed\n      for (const operation of this.batchOperationsList) {\n        const response: BulkOperationResult = {\n          operationInput: operation.operationInput,\n          error: Object.assign(new ErrorResponse(error.message), {\n            code: StatusCodes.InternalServerError,\n            diagnostics: operation.operationContext.diagnosticNode.toDiagnostic(\n              this.clientConfigDiagnostics,\n            ),\n          }),\n        };\n        operation.operationContext.fail(response);\n        this.processedOperationCountRef.count++;\n      }\n    } finally {\n      // Clean up batch state\n      this.batchOperationsList = [];\n    }\n  }\n\n  public getOperations(): ItemOperation[] {\n    return this.batchOperationsList;\n  }\n}\n"]}
+{"version":3,"file":"Batcher.js","sourceRoot":"","sources":["../../../src/bulk/Batcher.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAEvD,OAAO,EACL,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,0CAA0C,CAAC;AAGlD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C,OAAO,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AAC/D,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAK3D;;;;GAIG;AAEH,MAAM,OAAO,OAAO;IACV,mBAAmB,CAAkB;IACrC,WAAW,CAAS;IACpB,cAAc,CAAU;IACf,QAAQ,CAAkB;IAC1B,OAAO,CAAgB;IACvB,eAAe,CAA0B;IACzC,iBAAiB,CAAU;IAC3B,mBAAmB,CAAsB;IACzC,uBAAuB,CAAyB;IAChD,OAAO,CAAe;IAC/B,0BAA0B,CAAoB;IAEtD,YACE,OAAqB,EACrB,QAAyB,EACzB,OAAsB,EACtB,eAAwC,EACxC,iBAA0B,EAC1B,YAAoC,EACpC,mBAAwC,EACxC,0BAA6C;QAE7C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,uBAAuB,GAAG,YAAY,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,IAAI,CAAC,0BAA0B,GAAG,0BAA0B,CAAC;IAC/D,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,SAAwB;QACpC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,aAAa,CAAC,0BAA0B,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;YAChC,MAAM,IAAI,aAAa,CAAC,kCAAkC,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,KAAK,SAAS,CAAC,sBAAsB,EAAE,CAAC;YACzE,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,oBAAoB,GAAG,0BAA0B,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QAClF,IACE,IAAI,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC;YACnC,IAAI,CAAC,WAAW,GAAG,oBAAoB,GAAG,SAAS,CAAC,oCAAoC,EACxF,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,oBAAoB,CAAC;QACzC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,OAAO;QACZ,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,QAAQ,CAAC,eAAgC;QACpD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;QAC5C,MAAM,cAAc,GAAG,IAAI,sBAAsB,CAC/C,IAAI,CAAC,eAAe,EACpB,kBAAkB,CAAC,aAAa,EAChC,IAAI,CACL,CAAC;QACF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,mBAAmB,EAAE,cAAc,CAAC,CAAC;YAE/E,MAAM,YAAY,GAAG,CAAC,CAAC;YACvB,MAAM,UAAU,GAAG,CAAC,CAAC;YACrB,MAAM,WAAW,GAAG,QAAQ,EAAE,OAAO,EAAE,IAAI,CACzC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,eAAe,CAC5E;gBACC,CAAC,CAAC,YAAY;gBACd,CAAC,CAAC,UAAU,CAAC;YACf,MAAM,YAAY,GAAG,QAAQ,EAAE,OAAO,EAAE,IAAI,CAC1C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,IAAI,CACjE;gBACC,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,KAAK,CAAC;YACV,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YACrE,CAAC;YACD,eAAe,CAAC,GAAG,CACjB,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAC/B,uBAAuB,EAAE,GAAG,SAAS,EACrC,WAAW,CACZ,CAAC;YACF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpD,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,mBAAmB,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAChD,IAAI,mBAAmB,YAAY,aAAa,EAAE,CAAC;oBACjD,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,gBAAgB,CAAC,WAAW,CAAC,WAAW,CAC1E,mBAAmB,EACnB,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAC1C,CAAC;oBACF,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;wBACzE,SAAS;oBACX,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC;oBACH,IAAI,IAAI,CAAC,iBAAiB,IAAI,mBAAmB,CAAC,YAAY,EAAE,CAAC;wBAC/D,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,0BAA0B,CAClE,SAAS,CAAC,UAAU,CAAC,2BAA2B,CACjD,CAAC;wBACF,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,wBAAwB,EAAE,GACrD,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;wBAC3E,mBAAmB,CAAC,YAAY,GAAG,aAAa,CAAC;wBACjD,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,wBAAwB,CAChE,SAAS,CAAC,UAAU,CAAC,2BAA2B,EAChD,wBAAwB,CACzB,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,sGAAsG;oBACtG,IAAI,SAAS,CAAC,cAAc,CAAC,aAAa,KAAK,MAAM,EAAE,CAAC;wBACtD,MAAM,eAAe,GAAG,IAAI,aAAa,CACvC,QAAQ,SAAS,CAAC,cAAc,CAAC,aAAa,+DAA+D,KAAK,CAAC,OAAO,EAAE,CAC7H,CAAC;wBACF,eAAe,CAAC,IAAI,GAAG,WAAW,CAAC,kBAAkB,CAAC;wBACtD,MAAM,eAAe,CAAC;oBACxB,CAAC;gBACH,CAAC;gBACD,SAAS,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;gBAC9D,mBAAmB,CAAC,WAAW,GAAG,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,YAAY,CACtF,IAAI,CAAC,uBAAuB,CAC7B,CAAC;gBACF,MAAM,gBAAgB,GAAwB;oBAC5C,cAAc,EAAE,SAAS,CAAC,yBAAyB;iBACpD,CAAC;gBACF,IAAI,mBAAmB,YAAY,aAAa,EAAE,CAAC;oBACjD,gBAAgB,CAAC,KAAK,GAAG,mBAAmB,CAAC;gBAC/C,CAAC;qBAAM,CAAC;oBACN,gBAAgB,CAAC,QAAQ,GAAG,mBAAmB,CAAC;gBAClD,CAAC;gBACD,SAAS,CAAC,gBAAgB,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;gBACtD,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YAC1C,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,6CAA6C;YAC7C,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBACjD,MAAM,QAAQ,GAAwB;oBACpC,cAAc,EAAE,SAAS,CAAC,cAAc;oBACxC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE;wBACrD,IAAI,EAAE,WAAW,CAAC,mBAAmB;wBACrC,WAAW,EAAE,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,YAAY,CACjE,IAAI,CAAC,uBAAuB,CAC7B;qBACF,CAAC;iBACH,CAAC;gBACF,SAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC1C,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YAC1C,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,uBAAuB;YACvB,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAC;QAChC,CAAC;IACH,CAAC;IAEM,aAAa;QAClB,OAAO,IAAI,CAAC,mBAAmB,CAAC;IAClC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Constants } from \"../common/constants.js\";\nimport { StatusCodes } from \"../common/statusCodes.js\";\nimport type { CosmosDbDiagnosticLevel } from \"../diagnostics/CosmosDbDiagnosticLevel.js\";\nimport {\n  DiagnosticNodeInternal,\n  DiagnosticNodeType,\n} from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport type { EncryptionProcessor } from \"../encryption/EncryptionProcessor.js\";\nimport type { ClientConfigDiagnostic } from \"../index.js\";\nimport { ErrorResponse } from \"../index.js\";\nimport type { ExecuteCallback, RetryCallback, BulkOperationResult } from \"../utils/batch.js\";\nimport { calculateObjectSizeInBytes } from \"../utils/batch.js\";\nimport { getCurrentTimestampInMs } from \"../utils/time.js\";\nimport type { PartitionMetric } from \"./PartitionMetric.js\";\nimport type { ItemOperation } from \"./index.js\";\nimport type { LimiterQueue } from \"./Limiter.js\";\n\n/**\n * Maintains a batch of operations and dispatches it as a unit of work.\n * Execution of the request is done by the @see {@link ExecuteCallback} and retry is done by the @see {@link RetryCallback}.\n * @hidden\n */\n\nexport class Batcher {\n  private batchOperationsList: ItemOperation[];\n  private currentSize: number;\n  private toBeDispatched: boolean;\n  private readonly executor: ExecuteCallback;\n  private readonly retrier: RetryCallback;\n  private readonly diagnosticLevel: CosmosDbDiagnosticLevel;\n  private readonly encryptionEnabled: boolean;\n  private readonly encryptionProcessor: EncryptionProcessor;\n  private readonly clientConfigDiagnostics: ClientConfigDiagnostic;\n  private readonly limiter: LimiterQueue;\n  private processedOperationCountRef: { count: number };\n\n  constructor(\n    limiter: LimiterQueue,\n    executor: ExecuteCallback,\n    retrier: RetryCallback,\n    diagnosticLevel: CosmosDbDiagnosticLevel,\n    encryptionEnabled: boolean,\n    clientConfig: ClientConfigDiagnostic,\n    encryptionProcessor: EncryptionProcessor,\n    processedOperationCountRef: { count: number },\n  ) {\n    this.limiter = limiter;\n    this.batchOperationsList = [];\n    this.executor = executor;\n    this.retrier = retrier;\n    this.diagnosticLevel = diagnosticLevel;\n    this.encryptionEnabled = encryptionEnabled;\n    this.encryptionProcessor = encryptionProcessor;\n    this.clientConfigDiagnostics = clientConfig;\n    this.currentSize = 0;\n    this.toBeDispatched = false;\n    this.processedOperationCountRef = processedOperationCountRef;\n  }\n\n  /**\n   * Attempts to add an operation to the current batch.\n   * Returns false if the batch is full or already dispatched.\n   */\n  public tryAdd(operation: ItemOperation): boolean {\n    if (this.toBeDispatched) {\n      return false;\n    }\n    if (!operation) {\n      throw new ErrorResponse(\"Operation is not defined\");\n    }\n    if (!operation.operationContext) {\n      throw new ErrorResponse(\"Operation context is not defined\");\n    }\n    if (this.batchOperationsList.length === Constants.MaxBulkOperationsCount) {\n      return false;\n    }\n    const currentOperationSize = calculateObjectSizeInBytes(operation.operationInput);\n    if (\n      this.batchOperationsList.length > 0 &&\n      this.currentSize + currentOperationSize > Constants.DefaultMaxBulkRequestBodySizeInBytes\n    ) {\n      return false;\n    }\n\n    this.currentSize += currentOperationSize;\n    this.batchOperationsList.push(operation);\n    return true;\n  }\n\n  public isEmpty(): boolean {\n    return this.batchOperationsList.length === 0;\n  }\n\n  /**\n   * Dispatches the current batch of operations.\n   * Handles retries for failed operations and updates the ordered response.\n   */\n  public async dispatch(partitionMetric: PartitionMetric): Promise<void> {\n    this.toBeDispatched = true;\n    const startTime = getCurrentTimestampInMs();\n    const diagnosticNode = new DiagnosticNodeInternal(\n      this.diagnosticLevel,\n      DiagnosticNodeType.BATCH_REQUEST,\n      null,\n    );\n    try {\n      const response = await this.executor(this.batchOperationsList, diagnosticNode);\n\n      const hasThrottles = 1;\n      const noThrottle = 0;\n      const numThrottle = response?.results?.some(\n        (result) => \"code\" in result && result.code === StatusCodes.TooManyRequests,\n      )\n        ? hasThrottles\n        : noThrottle;\n      const splitOrMerge = response?.results?.some(\n        (result) => \"code\" in result && result.code === StatusCodes.Gone,\n      )\n        ? true\n        : false;\n      if (splitOrMerge) {\n        await this.limiter.pauseAndClear(StatusCodes.Gone, diagnosticNode);\n      }\n      partitionMetric.add(\n        this.batchOperationsList.length,\n        getCurrentTimestampInMs() - startTime,\n        numThrottle,\n      );\n      for (let i = 0; i < response.operations.length; i++) {\n        const operation = response.operations[i];\n        const bulkOperationResult = response.results[i];\n        if (bulkOperationResult instanceof ErrorResponse) {\n          const shouldRetry = await operation.operationContext.retryPolicy.shouldRetry(\n            bulkOperationResult,\n            operation.operationContext.diagnosticNode,\n          );\n          if (shouldRetry) {\n            await this.retrier(operation, operation.operationContext.diagnosticNode);\n            continue;\n          }\n        }\n        try {\n          if (this.encryptionEnabled && bulkOperationResult.resourceBody) {\n            operation.operationContext.diagnosticNode.beginEncryptionDiagnostics(\n              Constants.Encryption.DiagnosticsDecryptOperation,\n            );\n            const { body: decryptedBody, propertiesDecryptedCount } =\n              await this.encryptionProcessor.decrypt(bulkOperationResult.resourceBody);\n            bulkOperationResult.resourceBody = decryptedBody;\n            operation.operationContext.diagnosticNode.endEncryptionDiagnostics(\n              Constants.Encryption.DiagnosticsDecryptOperation,\n              propertiesDecryptedCount,\n            );\n          }\n        } catch (error) {\n          // if decryption fails after successful write operation, fail the operation with internal server error\n          if (operation.operationInput.operationType !== \"Read\") {\n            const decryptionError = new ErrorResponse(\n              `Item ${operation.operationInput.operationType} operation was successful but response decryption failed: + ${error.message}`,\n            );\n            decryptionError.code = StatusCodes.ServiceUnavailable;\n            throw decryptionError;\n          }\n        }\n        operation.operationContext.addDiagnosticChild(diagnosticNode);\n        bulkOperationResult.diagnostics = operation.operationContext.diagnosticNode.toDiagnostic(\n          this.clientConfigDiagnostics,\n        );\n        const bulkItemResponse: BulkOperationResult = {\n          operationInput: operation.unencryptedOperationInput,\n        };\n        if (bulkOperationResult instanceof ErrorResponse) {\n          bulkItemResponse.error = bulkOperationResult;\n        } else {\n          bulkItemResponse.response = bulkOperationResult;\n        }\n        operation.operationContext.complete(bulkItemResponse);\n        this.processedOperationCountRef.count++;\n      }\n    } catch (error) {\n      // Mark all operations in the batch as failed\n      for (const operation of this.batchOperationsList) {\n        const response: BulkOperationResult = {\n          operationInput: operation.operationInput,\n          error: Object.assign(new ErrorResponse(error.message), {\n            code: StatusCodes.InternalServerError,\n            diagnostics: operation.operationContext.diagnosticNode.toDiagnostic(\n              this.clientConfigDiagnostics,\n            ),\n          }),\n        };\n        operation.operationContext.fail(response);\n        this.processedOperationCountRef.count++;\n      }\n    } finally {\n      // Clean up batch state\n      this.batchOperationsList = [];\n    }\n  }\n\n  public getOperations(): ItemOperation[] {\n    return this.batchOperationsList;\n  }\n}\n"]}

package/dist/esm/bulk/BulkHelper.js

@@ -1,6 +1,5 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-import { __rest } from "tslib";
 import { readPartitionKeyDefinition } from "../client/ClientUtils.js";
 import { Constants, ResourceType } from "../common/constants.js";
 import { sleep, copyObject, getPathFromLink } from "../common/helper.js";
@@ -24,15 +23,25 @@ import { ItemOperationContext, BulkResponse } from "./index.js";
  * @hidden
  */
 export class BulkHelper {
+    container;
+    clientContext;
+    partitionKeyRangeCache;
+    helpersByPartitionKeyRangeId;
+    options;
+    partitionKeyDefinition;
+    partitionKeyDefinitionPromise;
+    isCancelled;
+    processedOperationCountRef = { count: 0 };
+    operationPromisesList = [];
+    congestionControlTimer;
+    congestionControlDelayInMs = 1000;
+    staleRidError;
+    operationsPerSleep = 100; // Number of operations to add per sleep
+    intervalForPartialBatchInMs = 1000; // Sleep interval before adding partial batch to dispatch queue
     /**
      * @internal
      */
     constructor(container, clientContext, partitionKeyRangeCache, options) {
-        this.processedOperationCountRef = { count: 0 };
-        this.operationPromisesList = [];
-        this.congestionControlDelayInMs = 1000;
-        this.operationsPerSleep = 100; // Number of operations to add per sleep
-        this.intervalForPartialBatchInMs = 1000; // Sleep interval before adding partial batch to dispatch queue
         this.container = container;
         this.clientContext = clientContext;
         this.partitionKeyRangeCache = partitionKeyRangeCache;
@@ -87,9 +96,12 @@ export class BulkHelper {
         // Formatting result: if an error is present, removing the stack trace details.
         const formattedResults = bulkOperationResults.map((result) => {
             if (result && result.error) {
-                const _a = result.error, { stack } = _a, otherProps = __rest(_a, ["stack"]);
-                const trimmedError = Object.assign({ message: result.error.message }, otherProps);
-                return Object.assign(Object.assign({}, result), { error: trimmedError });
+                const { stack, ...otherProps } = result.error;
+                const trimmedError = { message: result.error.message, ...otherProps };
+                return {
+                    ...result,
+                    error: trimmedError,
+                };
             }
             return result;
         });
@@ -183,7 +195,7 @@ export class BulkHelper {
                 operationInput: unencryptedOperation,
                 error: Object.assign(new ErrorResponse(operationError.message), {
                     code: StatusCodes.InternalServerError,
-                    diagnostics: diagnosticNode === null || diagnosticNode === void 0 ? void 0 : diagnosticNode.toDiagnostic(this.clientContext.getClientConfig()),
+                    diagnostics: diagnosticNode?.toDiagnostic(this.clientContext.getClientConfig()),
                 }),
             };
             context.fail(response);
@@ -251,7 +263,10 @@ export class BulkHelper {
     }
     prepareOperation(operationInput) {
         operationInput.partitionKey = convertToInternalPartitionKey(operationInput.partitionKey);
-        return Object.assign(Object.assign({}, operationInput), { partitionKey: JSON.stringify(operationInput.partitionKey) });
+        return {
+            ...operationInput,
+            partitionKey: JSON.stringify(operationInput.partitionKey),
+        };
     }
     async reBatchOperation(operation, diagnosticNode) {
         const partitionKeyRangeId = await this.resolvePartitionKeyRangeId(operation.operationInput, diagnosticNode);

package/dist/esm/bulk/BulkHelper.js.map

@@ -1 +1 @@
-{"version":3,"file":"BulkHelper.js","sourceRoot":"","sources":["../../../src/bulk/BulkHelper.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAGlC,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAGtE,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EACL,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,0CAA0C,CAAC;AAElD,OAAO,EAAE,6BAA6B,EAAE,MAAM,sCAAsC,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,wBAAwB,EAAE,MAAM,sCAAsC,CAAC;AAChF,OAAO,EAAE,2BAA2B,EAAE,MAAM,yCAAyC,CAAC;AAItF,OAAO,EAAE,qBAAqB,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACxE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAEhE;;;;;;;GAOG;AAEH,MAAM,OAAO,UAAU;IAiBrB;;OAEG;IACH,YACE,SAAoB,EACpB,aAA4B,EAC5B,sBAA8C,EAC9C,OAAuB;QAfjB,+BAA0B,GAAsB,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QAC7D,0BAAqB,GAAmC,EAAE,CAAC;QAElD,+BAA0B,GAAW,IAAI,CAAC;QAE1C,uBAAkB,GAAW,GAAG,CAAC,CAAC,wCAAwC;QAC1E,gCAA2B,GAAW,IAAI,CAAC,CAAC,+DAA+D;QAW1H,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,4BAA4B,GAAG,IAAI,GAAG,EAAE,CAAC;QAC9C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,6BAA6B,GAAG,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnF,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,yBAAyB,EAAE,CAAC;IACnC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO,CAAC,cAAgC;QAC5C,MAAM,oBAAoB,GAAoB,EAAE,CAAC;QACjD,MAAM,YAAY,GAAG,CAAC,CAAC,CAAC,gEAAgE;QACxF,IAAI,CAAC;YACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,mGAAmG;gBACnG,2GAA2G;gBAC3G,2DAA2D;gBAC3D,IAAI,CAAC,GAAG,IAAI,CAAC,kBAAkB,KAAK,CAAC,EAAE,CAAC;oBACtC,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC5B,CAAC;gBACD,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACrE,CAAC;YACD,MAAM,OAAO,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;YAE/C,gIAAgI;YAChI,qHAAqH;YACrH,kIAAkI;YAClI,kFAAkF;YAClF,OAAO,IAAI,CAAC,0BAA0B,CAAC,KAAK,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC;gBACrE,IAAI,CAAC,4BAA4B,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;oBACnD,MAAM,CAAC,sBAAsB,EAAE,CAAC;gBAClC,CAAC,CAAC,CAAC;gBACH,0HAA0H;gBAC1H,MAAM,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBAChC,aAAa,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QACD,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC5E,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC3C,MAAM,IAAI,CAAC,aAAa,CAAC;QAC3B,CAAC;QAED,MAAM,oBAAoB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACzD,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAC7D,CAAC;QAEF,+EAA+E;QAC/E,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAC3D,IAAI,MAAM,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAA2B,MAAM,CAAC,KAAK,EAAvC,EAAE,KAAK,OAAgC,EAA3B,UAAU,cAAtB,SAAwB,CAAe,CAAC;gBAC9C,MAAM,YAAY,mBAAK,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,IAAK,UAAU,CAAE,CAAC;gBACtE,uCACK,MAAM,KACT,KAAK,EAAE,YAAY,IACnB;YACJ,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,SAAyB,EAAE,GAAW;QAC/D,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;gBAChD,cAAc,EAAE,SAAS;gBACzB,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,wCAAwC,CAAC,EAAE;oBAChF,IAAI,EAAE,WAAW,CAAC,mBAAmB;iBACtC,CAAC;aACH,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,gDAAgD;QAChD,IACE,SAAS,CAAC,aAAa,KAAK,QAAQ;YACpC,SAAS,CAAC,aAAa,KAAK,QAAQ;YACpC,SAAS,CAAC,aAAa,KAAK,SAAS,EACrC,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,EAAE,CAAC;gBAC/B,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;oBAChD,cAAc,EAAE,SAAS;oBACzB,KAAK,EAAE,MAAM,CAAC,MAAM,CAClB,IAAI,aAAa,CACf,iDAAiD,SAAS,CAAC,aAAa,cAAc,CACvF,EACD,EAAE,IAAI,EAAE,WAAW,CAAC,mBAAmB,EAAE,CAC1C;iBACF,CAAC,CAAC;gBACH,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;gBACxC,OAAO;YACT,CAAC;QACH,CAAC;QACD,IAAI,SAAS,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YACzC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;gBAChD,cAAc,EAAE,SAAS;gBACzB,KAAK,EAAE,MAAM,CAAC,MAAM,CAClB,IAAI,aAAa,CAAC,gCAAgC,SAAS,CAAC,aAAa,cAAc,CAAC,EACxF,EAAE,IAAI,EAAE,WAAW,CAAC,mBAAmB,EAAE,CAC1C;aACF,CAAC,CAAC;YACH,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QAED,IAAI,cAAiC,CAAC;QACtC,IAAI,cAAsC,CAAC;QAC3C,IAAI,oBAAoC,CAAC;QACzC,IAAI,mBAA2B,CAAC;QAChC,IAAI,CAAC;YACH,cAAc,GAAG,IAAI,sBAAsB,CACzC,IAAI,CAAC,aAAa,CAAC,eAAe,EAClC,kBAAkB,CAAC,mBAAmB,EACtC,IAAI,CACL,CAAC;YACF,gDAAgD;YAChD,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBACjC,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,CAAC;oBACxC,IAAI,CAAC,6BAA6B,GAAG,CAAC,KAAK,IAAI,EAAE;wBAC/C,IAAI,CAAC;4BACH,MAAM,sBAAsB,GAAG,MAAM,0BAA0B,CAC7D,cAAc,EACd,IAAI,CAAC,SAAS,CACf,CAAC;4BACF,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;4BACrD,OAAO,sBAAsB,CAAC;wBAChC,CAAC;gCAAS,CAAC;4BACT,IAAI,CAAC,6BAA6B,GAAG,IAAI,CAAC;wBAC5C,CAAC;oBACH,CAAC,CAAC,EAAE,CAAC;gBACP,CAAC;gBACD,MAAM,IAAI,CAAC,6BAA6B,CAAC;YAC3C,CAAC;YACD,oBAAoB,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;YAC7C,yDAAyD;YACzD,IAAI,IAAI,CAAC,aAAa,CAAC,gBAAgB,EAAE,CAAC;gBACxC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;gBAClC,MAAM,IAAI,CAAC,SAAS,CAAC,4BAA4B,EAAE,CAAC;gBACpD,cAAc,CAAC,0BAA0B,CAAC,SAAS,CAAC,UAAU,CAAC,2BAA2B,CAAC,CAAC;gBAC5F,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,6BAA6B,EAAE,GAC7D,MAAM,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;gBAChF,SAAS,GAAG,WAAW,CAAC;gBACxB,cAAc,CAAC,wBAAwB,CACrC,SAAS,CAAC,UAAU,CAAC,2BAA2B,EAChD,6BAA6B,CAC9B,CAAC;YACJ,CAAC;YACD,sCAAsC;YACtC,mBAAmB,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;QACzF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,cAAc,GAAG,KAAK,CAAC;QACzB,CAAC;QAED,wBAAwB;QACxB,MAAM,kBAAkB,GAAG,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,CAAC;QACzE,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,mBAAmB,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;QAC3F,MAAM,aAAa,GAAkB;YACnC,yBAAyB,EAAE,oBAAoB;YAC/C,cAAc,EAAE,SAAS;YACzB,gBAAgB,EAAE,OAAO;SAC1B,CAAC;QAEF,6DAA6D;QAC7D,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;QAE3D,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,QAAQ,GAAwB;gBACpC,cAAc,EAAE,oBAAoB;gBACpC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE;oBAC9D,IAAI,EAAE,WAAW,CAAC,mBAAmB;oBACrC,WAAW,EAAE,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC;iBAChF,CAAC;aACH,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvB,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QACD,mCAAmC;QACnC,OAAO,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,SAAyB,EACzB,cAAsC;QAEtC,MAAM,kBAAkB,GAAG,CACzB,MAAM,IAAI,CAAC,sBAAsB,CAAC,sBAAsB,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,cAAc,CAAC,CAC7F,CAAC,2BAA2B,EAAE,CAAC;QAEhC,MAAM,YAAY,GAAG,6BAA6B,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAE3E,MAAM,SAAS,GAAG,gBAAgB,CAAC,YAAY,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAE9E,MAAM,aAAa,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CACtD,YAAY,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,SAAS,CAAC,CAChE,CAAC;QAEF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC9E,CAAC;QACD,OAAO,aAAa,CAAC,EAAE,CAAC;IAC1B,CAAC;IAEO,cAAc;QACpB,MAAM,eAAe,GAAG,IAAI,2BAA2B,CAAC,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC,CAAC;QAC9F,OAAO,IAAI,wBAAwB,CAAC,eAAe,CAAC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,UAA2B,EAC3B,cAAsC;QAEtC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,aAAa,CAAC,mDAAmD,CAAC,CAAC;QAC/E,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,MAAM;YAAE,OAAO;QAC/B,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC3D,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;QACpE,MAAM,WAAW,GAAgB,EAAE,CAAC;QACpC,KAAK,MAAM,aAAa,IAAI,UAAU,EAAE,CAAC;YACvC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;QAClD,CAAC;QACD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CACvC,KAAK,EAAE,SAAiC,EAAE,EAAE,CAC1C,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,WAAW;gBACjB,mBAAmB,EAAE,SAAS;gBAC9B,IAAI,EAAE,IAAI;gBACV,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG;gBAC9B,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,cAAc,EAAE,SAAS;aAC1B,CAAC,EACJ,cAAc,EACd,kBAAkB,CAAC,aAAa,CACjC,CAAC;YACF,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,aAAa,CAAC,gCAAgC,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,YAAY,CAAC,mBAAmB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAChE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,aAAa,CAAC,gBAAgB,EAAE,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,SAAS,CAAC,0CAA0C,CAAC,KAAK,CAAC,CAAC;gBACzE,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;oBAChC,OAAO,YAAY,CAAC,mBAAmB,CAAC,UAAU,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC;YACD,OAAO,YAAY,CAAC,mBAAmB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,cAA8B;QACrD,cAAc,CAAC,YAAY,GAAG,6BAA6B,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACzF,OAAO,gCACF,cAAc,KACjB,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,YAAY,CAAC,GAC7C,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,SAAwB,EACxB,cAAsC;QAEtC,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAC/D,SAAS,CAAC,cAAc,EACxB,cAAc,CACf,CAAC;QACF,SAAS,CAAC,gBAAgB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,CAAC;QAC7D,MAAM,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,KAAoB;QAChD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,4BAA4B,CAAC,MAAM,EAAE,EAAE,CAAC;YAChE,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,4BAA4B,CAAC,KAAK,EAAE,CAAC;IAC5C,CAAC;IAEO,mBAAmB,CAAC,SAAiB;QAC3C,IAAI,IAAI,CAAC,4BAA4B,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACrD,OAAO,IAAI,CAAC,4BAA4B,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,kBAAkB,CACtC,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,6BAA6B,EAClC,IAAI,CAAC,aAAa,CAAC,eAAe,EAClC,IAAI,CAAC,aAAa,CAAC,gBAAgB,EACnC,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,EACpC,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAClC,IAAI,CAAC,0BAA0B,CAChC,CAAC;QACF,IAAI,CAAC,4BAA4B,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC5D,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,yBAAyB;QAC/B,IAAI,CAAC,sBAAsB,GAAG,WAAW,CAAC,GAAG,EAAE;YAC7C,IAAI,CAAC,4BAA4B,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACnD,MAAM,CAAC,sBAAsB,EAAE,CAAC;YAClC,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,0BAA0B,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,6BAA6B,CACzC,cAAsC;QAEtC,MAAM,IAAI,CAAC,sBAAsB,CAAC,sBAAsB,CACtD,IAAI,CAAC,SAAS,CAAC,GAAG,EAClB,cAAc,EACd,IAAI,CACL,CAAC;IACJ,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { RequestOptions } from \"../request/RequestOptions.js\";\nimport { readPartitionKeyDefinition } from \"../client/ClientUtils.js\";\nimport type { Container } from \"../client/index.js\";\nimport type { ClientContext } from \"../ClientContext.js\";\nimport { Constants, ResourceType } from \"../common/constants.js\";\nimport { sleep, copyObject, getPathFromLink } from \"../common/helper.js\";\nimport { StatusCodes } from \"../common/statusCodes.js\";\nimport {\n  DiagnosticNodeInternal,\n  DiagnosticNodeType,\n} from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport type { PartitionKeyDefinition } from \"../documents/PartitionKeyDefinition.js\";\nimport { convertToInternalPartitionKey } from \"../documents/PartitionKeyInternal.js\";\nimport { ErrorResponse } from \"../index.js\";\nimport { BulkExecutionRetryPolicy } from \"../retry/bulkExecutionRetryPolicy.js\";\nimport { ResourceThrottleRetryPolicy } from \"../retry/resourceThrottleRetryPolicy.js\";\nimport type { RetryPolicy } from \"../retry/RetryPolicy.js\";\nimport type { PartitionKeyRangeCache } from \"../routing/partitionKeyRangeCache.js\";\nimport type { BulkOperationResult, OperationInput, Operation } from \"../utils/batch.js\";\nimport { encryptOperationInput, isKeyInRange } from \"../utils/batch.js\";\nimport { addDiagnosticChild } from \"../utils/diagnostics.js\";\nimport { hashPartitionKey } from \"../utils/hashing/hash.js\";\nimport { HelperPerPartition } from \"./HelperPerPartition.js\";\nimport type { ItemOperation } from \"./index.js\";\nimport { ItemOperationContext, BulkResponse } from \"./index.js\";\n\n/**\n * BulkHelper for bulk operations in a container.\n * It maintains one @see {@link HelperPerPartition} for each Partition Key Range, which allows independent execution of requests. Queue based limiters @see {@link LimiterQueue}\n * rate limit requestsbat the helper / Partition Key Range level, this means that we can send parallel and independent requests to different Partition Key Ranges, but for the same Range, requests\n * will be limited. Two callback implementations define how a particular request should be executed, and how operations should be retried. When the helper dispatches a batch\n * the batch will create a request and call the execute callback (executeRequest), if conditions are met, it might call the retry callback (reBatchOperation).\n * @hidden\n */\n\nexport class BulkHelper {\n  private readonly container: Container;\n  private readonly clientContext: ClientContext;\n  private readonly partitionKeyRangeCache: PartitionKeyRangeCache;\n  private readonly helpersByPartitionKeyRangeId: Map<string, HelperPerPartition>;\n  private options: RequestOptions;\n  private partitionKeyDefinition: PartitionKeyDefinition;\n  private partitionKeyDefinitionPromise: Promise<PartitionKeyDefinition>;\n  private isCancelled: boolean;\n  private processedOperationCountRef: { count: number } = { count: 0 };\n  private operationPromisesList: Promise<BulkOperationResult>[] = [];\n  private congestionControlTimer: NodeJS.Timeout;\n  private readonly congestionControlDelayInMs: number = 1000;\n  private staleRidError: ErrorResponse | undefined;\n  private readonly operationsPerSleep: number = 100; // Number of operations to add per sleep\n  private readonly intervalForPartialBatchInMs: number = 1000; // Sleep interval before adding partial batch to dispatch queue\n\n  /**\n   * @internal\n   */\n  constructor(\n    container: Container,\n    clientContext: ClientContext,\n    partitionKeyRangeCache: PartitionKeyRangeCache,\n    options: RequestOptions,\n  ) {\n    this.container = container;\n    this.clientContext = clientContext;\n    this.partitionKeyRangeCache = partitionKeyRangeCache;\n    this.helpersByPartitionKeyRangeId = new Map();\n    this.options = options;\n    this.executeRequest = this.executeRequest.bind(this);\n    this.reBatchOperation = this.reBatchOperation.bind(this);\n    this.refreshPartitionKeyRangeCache = this.refreshPartitionKeyRangeCache.bind(this);\n    this.isCancelled = false;\n    this.runCongestionControlTimer();\n  }\n\n  /**\n   * adds operation(s) to the helper\n   * @param operationInput - bulk operation or list of bulk operations\n   */\n  async execute(operationInput: OperationInput[]): Promise<BulkOperationResult[]> {\n    const addOperationPromises: Promise<void>[] = [];\n    const minimalPause = 0; // minimal pause (0 ms) inserted periodically during processing.\n    try {\n      for (let i = 0; i < operationInput.length; i++) {\n        // After every 100 operations,sleep of 0 ms is added to allow the event loop to process any pending\n        // callbacks/tasks such as fetching partition key definition and dispatching batches from queue. This helps\n        // to prevent blocking and improves overall responsiveness.\n        if (i % this.operationsPerSleep === 0) {\n          await sleep(minimalPause);\n        }\n        addOperationPromises.push(this.addOperation(operationInput[i], i));\n      }\n      await Promise.allSettled(addOperationPromises);\n\n      // After processing all operations via addOperation, it's possible that the current batch in each helper is not completely full.\n      // In such cases, addPartialBatchToQueue is called to ensure that all the operations are added to the dispatch queue.\n      // while loop below waits until the count of processed operations equals the number of input operations. This is necessary because\n      // some operations might fail and then again get added to current batch for retry.\n      while (this.processedOperationCountRef.count < operationInput.length) {\n        this.helpersByPartitionKeyRangeId.forEach((helper) => {\n          helper.addPartialBatchToQueue();\n        });\n        // Pause for 1000 ms to give pending operations chance to accumulate into a batch to avoid sending multiple small batches.\n        await sleep(this.intervalForPartialBatchInMs);\n      }\n    } finally {\n      if (this.congestionControlTimer) {\n        clearInterval(this.congestionControlTimer);\n      }\n    }\n    const settledResults = await Promise.allSettled(this.operationPromisesList);\n    if (this.isCancelled && this.staleRidError) {\n      throw this.staleRidError;\n    }\n\n    const bulkOperationResults = settledResults.map((result) =>\n      result.status === \"fulfilled\" ? result.value : result.reason,\n    );\n\n    // Formatting result: if an error is present, removing the stack trace details.\n    const formattedResults = bulkOperationResults.map((result) => {\n      if (result && result.error) {\n        const { stack, ...otherProps } = result.error;\n        const trimmedError = { message: result.error.message, ...otherProps };\n        return {\n          ...result,\n          error: trimmedError,\n        };\n      }\n      return result;\n    });\n    return formattedResults;\n  }\n\n  private async addOperation(operation: OperationInput, idx: number): Promise<void> {\n    if (this.isCancelled) {\n      return;\n    }\n    if (!operation) {\n      this.operationPromisesList[idx] = Promise.resolve({\n        operationInput: operation,\n        error: Object.assign(new ErrorResponse(\"Operation cannot be null or undefined.\"), {\n          code: StatusCodes.InternalServerError,\n        }),\n      });\n      return;\n    }\n\n    // Checks for id and partition key in input body\n    if (\n      operation.operationType === \"Create\" ||\n      operation.operationType === \"Upsert\" ||\n      operation.operationType === \"Replace\"\n    ) {\n      if (!operation.resourceBody.id) {\n        this.operationPromisesList[idx] = Promise.resolve({\n          operationInput: operation,\n          error: Object.assign(\n            new ErrorResponse(\n              `Operation resource body must have an 'id' for ${operation.operationType} operations.`,\n            ),\n            { code: StatusCodes.InternalServerError },\n          ),\n        });\n        this.processedOperationCountRef.count++;\n        return;\n      }\n    }\n    if (operation.partitionKey === undefined) {\n      this.operationPromisesList[idx] = Promise.resolve({\n        operationInput: operation,\n        error: Object.assign(\n          new ErrorResponse(`PartitionKey is required for ${operation.operationType} operations.`),\n          { code: StatusCodes.InternalServerError },\n        ),\n      });\n      this.processedOperationCountRef.count++;\n      return;\n    }\n\n    let operationError: Error | undefined;\n    let diagnosticNode: DiagnosticNodeInternal;\n    let unencryptedOperation: OperationInput;\n    let partitionKeyRangeId: string;\n    try {\n      diagnosticNode = new DiagnosticNodeInternal(\n        this.clientContext.diagnosticLevel,\n        DiagnosticNodeType.CLIENT_REQUEST_NODE,\n        null,\n      );\n      // Ensure partition key definition is available.\n      if (!this.partitionKeyDefinition) {\n        if (!this.partitionKeyDefinitionPromise) {\n          this.partitionKeyDefinitionPromise = (async () => {\n            try {\n              const partitionKeyDefinition = await readPartitionKeyDefinition(\n                diagnosticNode,\n                this.container,\n              );\n              this.partitionKeyDefinition = partitionKeyDefinition;\n              return partitionKeyDefinition;\n            } finally {\n              this.partitionKeyDefinitionPromise = null;\n            }\n          })();\n        }\n        await this.partitionKeyDefinitionPromise;\n      }\n      unencryptedOperation = copyObject(operation);\n      // If encryption is enabled, encrypt the operation input.\n      if (this.clientContext.enableEncryption) {\n        operation = copyObject(operation);\n        await this.container.checkAndInitializeEncryption();\n        diagnosticNode.beginEncryptionDiagnostics(Constants.Encryption.DiagnosticsEncryptOperation);\n        const { operation: encryptedOp, totalPropertiesEncryptedCount } =\n          await encryptOperationInput(this.container.encryptionProcessor, operation, 0);\n        operation = encryptedOp;\n        diagnosticNode.endEncryptionDiagnostics(\n          Constants.Encryption.DiagnosticsEncryptOperation,\n          totalPropertiesEncryptedCount,\n        );\n      }\n      // Resolve the partition key range id.\n      partitionKeyRangeId = await this.resolvePartitionKeyRangeId(operation, diagnosticNode);\n    } catch (error) {\n      operationError = error;\n    }\n\n    // Get helper & context.\n    const helperForPartition = this.getHelperForPKRange(partitionKeyRangeId);\n    const retryPolicy = this.getRetryPolicy();\n    const context = new ItemOperationContext(partitionKeyRangeId, retryPolicy, diagnosticNode);\n    const itemOperation: ItemOperation = {\n      unencryptedOperationInput: unencryptedOperation,\n      operationInput: operation,\n      operationContext: context,\n    };\n\n    // Assign the promise (ensuring position matches input order)\n    this.operationPromisesList[idx] = context.operationPromise;\n\n    if (operationError) {\n      const response: BulkOperationResult = {\n        operationInput: unencryptedOperation,\n        error: Object.assign(new ErrorResponse(operationError.message), {\n          code: StatusCodes.InternalServerError,\n          diagnostics: diagnosticNode?.toDiagnostic(this.clientContext.getClientConfig()),\n        }),\n      };\n      context.fail(response);\n      this.processedOperationCountRef.count++;\n      return;\n    }\n    // Add the operation to the helper.\n    return helperForPartition.add(itemOperation);\n  }\n\n  private async resolvePartitionKeyRangeId(\n    operation: OperationInput,\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<string> {\n    const partitionKeyRanges = (\n      await this.partitionKeyRangeCache.onCollectionRoutingMap(this.container.url, diagnosticNode)\n    ).getOrderedParitionKeyRanges();\n\n    const partitionKey = convertToInternalPartitionKey(operation.partitionKey);\n\n    const hashedKey = hashPartitionKey(partitionKey, this.partitionKeyDefinition);\n\n    const matchingRange = partitionKeyRanges.find((range) =>\n      isKeyInRange(range.minInclusive, range.maxExclusive, hashedKey),\n    );\n\n    if (!matchingRange) {\n      throw new Error(\"No matching partition key range found for the operation.\");\n    }\n    return matchingRange.id;\n  }\n\n  private getRetryPolicy(): RetryPolicy {\n    const nextRetryPolicy = new ResourceThrottleRetryPolicy(this.clientContext.getRetryOptions());\n    return new BulkExecutionRetryPolicy(nextRetryPolicy);\n  }\n\n  private async executeRequest(\n    operations: ItemOperation[],\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<BulkResponse> {\n    if (this.isCancelled) {\n      throw new ErrorResponse(\"Bulk execution cancelled due to a previous error.\");\n    }\n    if (!operations.length) return;\n    const pkRangeId = operations[0].operationContext.pkRangeId;\n    const path = getPathFromLink(this.container.url, ResourceType.item);\n    const requestBody: Operation[] = [];\n    for (const itemOperation of operations) {\n      requestBody.push(this.prepareOperation(itemOperation.operationInput));\n    }\n    if (!this.options.containerRid) {\n      this.options.containerRid = this.container._rid;\n    }\n    try {\n      const response = await addDiagnosticChild(\n        async (childNode: DiagnosticNodeInternal) =>\n          this.clientContext.bulk({\n            body: requestBody,\n            partitionKeyRangeId: pkRangeId,\n            path: path,\n            resourceId: this.container.url,\n            options: this.options,\n            diagnosticNode: childNode,\n          }),\n        diagnosticNode,\n        DiagnosticNodeType.BATCH_REQUEST,\n      );\n      if (!response) {\n        throw new ErrorResponse(\"Failed to fetch bulk response.\");\n      }\n      return BulkResponse.fromResponseMessage(response, operations);\n    } catch (error) {\n      if (this.clientContext.enableEncryption) {\n        try {\n          await this.container.throwIfRequestNeedsARetryPostPolicyRefresh(error);\n        } catch (err) {\n          await this.cancelExecution(err);\n          return BulkResponse.createEmptyResponse(operations, 0, 0, {});\n        }\n      }\n      return BulkResponse.fromResponseMessage(error, operations);\n    }\n  }\n\n  private prepareOperation(operationInput: OperationInput): Operation {\n    operationInput.partitionKey = convertToInternalPartitionKey(operationInput.partitionKey);\n    return {\n      ...operationInput,\n      partitionKey: JSON.stringify(operationInput.partitionKey),\n    } as Operation;\n  }\n\n  private async reBatchOperation(\n    operation: ItemOperation,\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<void> {\n    const partitionKeyRangeId = await this.resolvePartitionKeyRangeId(\n      operation.operationInput,\n      diagnosticNode,\n    );\n    operation.operationContext.updatePKRangeId(partitionKeyRangeId);\n    const helper = this.getHelperForPKRange(partitionKeyRangeId);\n    await helper.add(operation);\n  }\n\n  private async cancelExecution(error: ErrorResponse): Promise<void> {\n    this.isCancelled = true;\n    this.staleRidError = error;\n    for (const helper of this.helpersByPartitionKeyRangeId.values()) {\n      await helper.dispose();\n    }\n    this.helpersByPartitionKeyRangeId.clear();\n  }\n\n  private getHelperForPKRange(pkRangeId: string): HelperPerPartition {\n    if (this.helpersByPartitionKeyRangeId.has(pkRangeId)) {\n      return this.helpersByPartitionKeyRangeId.get(pkRangeId);\n    }\n\n    const newHelper = new HelperPerPartition(\n      this.executeRequest,\n      this.reBatchOperation,\n      this.refreshPartitionKeyRangeCache,\n      this.clientContext.diagnosticLevel,\n      this.clientContext.enableEncryption,\n      this.clientContext.getClientConfig(),\n      this.container.encryptionProcessor,\n      this.processedOperationCountRef,\n    );\n    this.helpersByPartitionKeyRangeId.set(pkRangeId, newHelper);\n    return newHelper;\n  }\n\n  private runCongestionControlTimer(): void {\n    this.congestionControlTimer = setInterval(() => {\n      this.helpersByPartitionKeyRangeId.forEach((helper) => {\n        helper.runCongestionAlgorithm();\n      });\n    }, this.congestionControlDelayInMs);\n  }\n\n  private async refreshPartitionKeyRangeCache(\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<void> {\n    await this.partitionKeyRangeCache.onCollectionRoutingMap(\n      this.container.url,\n      diagnosticNode,\n      true,\n    );\n  }\n}\n"]}
+{"version":3,"file":"BulkHelper.js","sourceRoot":"","sources":["../../../src/bulk/BulkHelper.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAGtE,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACzE,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EACL,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,0CAA0C,CAAC;AAElD,OAAO,EAAE,6BAA6B,EAAE,MAAM,sCAAsC,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,wBAAwB,EAAE,MAAM,sCAAsC,CAAC;AAChF,OAAO,EAAE,2BAA2B,EAAE,MAAM,yCAAyC,CAAC;AAItF,OAAO,EAAE,qBAAqB,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACxE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAEhE;;;;;;;GAOG;AAEH,MAAM,OAAO,UAAU;IACJ,SAAS,CAAY;IACrB,aAAa,CAAgB;IAC7B,sBAAsB,CAAyB;IAC/C,4BAA4B,CAAkC;IACvE,OAAO,CAAiB;IACxB,sBAAsB,CAAyB;IAC/C,6BAA6B,CAAkC;IAC/D,WAAW,CAAU;IACrB,0BAA0B,GAAsB,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;IAC7D,qBAAqB,GAAmC,EAAE,CAAC;IAC3D,sBAAsB,CAAiB;IAC9B,0BAA0B,GAAW,IAAI,CAAC;IACnD,aAAa,CAA4B;IAChC,kBAAkB,GAAW,GAAG,CAAC,CAAC,wCAAwC;IAC1E,2BAA2B,GAAW,IAAI,CAAC,CAAC,+DAA+D;IAE5H;;OAEG;IACH,YACE,SAAoB,EACpB,aAA4B,EAC5B,sBAA8C,EAC9C,OAAuB;QAEvB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,4BAA4B,GAAG,IAAI,GAAG,EAAE,CAAC;QAC9C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,6BAA6B,GAAG,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnF,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,yBAAyB,EAAE,CAAC;IACnC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO,CAAC,cAAgC;QAC5C,MAAM,oBAAoB,GAAoB,EAAE,CAAC;QACjD,MAAM,YAAY,GAAG,CAAC,CAAC,CAAC,gEAAgE;QACxF,IAAI,CAAC;YACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,mGAAmG;gBACnG,2GAA2G;gBAC3G,2DAA2D;gBAC3D,IAAI,CAAC,GAAG,IAAI,CAAC,kBAAkB,KAAK,CAAC,EAAE,CAAC;oBACtC,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC5B,CAAC;gBACD,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACrE,CAAC;YACD,MAAM,OAAO,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;YAE/C,gIAAgI;YAChI,qHAAqH;YACrH,kIAAkI;YAClI,kFAAkF;YAClF,OAAO,IAAI,CAAC,0BAA0B,CAAC,KAAK,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC;gBACrE,IAAI,CAAC,4BAA4B,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;oBACnD,MAAM,CAAC,sBAAsB,EAAE,CAAC;gBAClC,CAAC,CAAC,CAAC;gBACH,0HAA0H;gBAC1H,MAAM,KAAK,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBAChC,aAAa,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QACD,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC5E,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC3C,MAAM,IAAI,CAAC,aAAa,CAAC;QAC3B,CAAC;QAED,MAAM,oBAAoB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACzD,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAC7D,CAAC;QAEF,+EAA+E;QAC/E,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAC3D,IAAI,MAAM,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAC3B,MAAM,EAAE,KAAK,EAAE,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC9C,MAAM,YAAY,GAAG,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,UAAU,EAAE,CAAC;gBACtE,OAAO;oBACL,GAAG,MAAM;oBACT,KAAK,EAAE,YAAY;iBACpB,CAAC;YACJ,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,SAAyB,EAAE,GAAW;QAC/D,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;gBAChD,cAAc,EAAE,SAAS;gBACzB,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,wCAAwC,CAAC,EAAE;oBAChF,IAAI,EAAE,WAAW,CAAC,mBAAmB;iBACtC,CAAC;aACH,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,gDAAgD;QAChD,IACE,SAAS,CAAC,aAAa,KAAK,QAAQ;YACpC,SAAS,CAAC,aAAa,KAAK,QAAQ;YACpC,SAAS,CAAC,aAAa,KAAK,SAAS,EACrC,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,EAAE,CAAC;gBAC/B,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;oBAChD,cAAc,EAAE,SAAS;oBACzB,KAAK,EAAE,MAAM,CAAC,MAAM,CAClB,IAAI,aAAa,CACf,iDAAiD,SAAS,CAAC,aAAa,cAAc,CACvF,EACD,EAAE,IAAI,EAAE,WAAW,CAAC,mBAAmB,EAAE,CAC1C;iBACF,CAAC,CAAC;gBACH,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;gBACxC,OAAO;YACT,CAAC;QACH,CAAC;QACD,IAAI,SAAS,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;YACzC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;gBAChD,cAAc,EAAE,SAAS;gBACzB,KAAK,EAAE,MAAM,CAAC,MAAM,CAClB,IAAI,aAAa,CAAC,gCAAgC,SAAS,CAAC,aAAa,cAAc,CAAC,EACxF,EAAE,IAAI,EAAE,WAAW,CAAC,mBAAmB,EAAE,CAC1C;aACF,CAAC,CAAC;YACH,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QAED,IAAI,cAAiC,CAAC;QACtC,IAAI,cAAsC,CAAC;QAC3C,IAAI,oBAAoC,CAAC;QACzC,IAAI,mBAA2B,CAAC;QAChC,IAAI,CAAC;YACH,cAAc,GAAG,IAAI,sBAAsB,CACzC,IAAI,CAAC,aAAa,CAAC,eAAe,EAClC,kBAAkB,CAAC,mBAAmB,EACtC,IAAI,CACL,CAAC;YACF,gDAAgD;YAChD,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;gBACjC,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,CAAC;oBACxC,IAAI,CAAC,6BAA6B,GAAG,CAAC,KAAK,IAAI,EAAE;wBAC/C,IAAI,CAAC;4BACH,MAAM,sBAAsB,GAAG,MAAM,0BAA0B,CAC7D,cAAc,EACd,IAAI,CAAC,SAAS,CACf,CAAC;4BACF,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;4BACrD,OAAO,sBAAsB,CAAC;wBAChC,CAAC;gCAAS,CAAC;4BACT,IAAI,CAAC,6BAA6B,GAAG,IAAI,CAAC;wBAC5C,CAAC;oBACH,CAAC,CAAC,EAAE,CAAC;gBACP,CAAC;gBACD,MAAM,IAAI,CAAC,6BAA6B,CAAC;YAC3C,CAAC;YACD,oBAAoB,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;YAC7C,yDAAyD;YACzD,IAAI,IAAI,CAAC,aAAa,CAAC,gBAAgB,EAAE,CAAC;gBACxC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;gBAClC,MAAM,IAAI,CAAC,SAAS,CAAC,4BAA4B,EAAE,CAAC;gBACpD,cAAc,CAAC,0BAA0B,CAAC,SAAS,CAAC,UAAU,CAAC,2BAA2B,CAAC,CAAC;gBAC5F,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,6BAA6B,EAAE,GAC7D,MAAM,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;gBAChF,SAAS,GAAG,WAAW,CAAC;gBACxB,cAAc,CAAC,wBAAwB,CACrC,SAAS,CAAC,UAAU,CAAC,2BAA2B,EAChD,6BAA6B,CAC9B,CAAC;YACJ,CAAC;YACD,sCAAsC;YACtC,mBAAmB,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;QACzF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,cAAc,GAAG,KAAK,CAAC;QACzB,CAAC;QAED,wBAAwB;QACxB,MAAM,kBAAkB,GAAG,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,CAAC;QACzE,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,oBAAoB,CAAC,mBAAmB,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;QAC3F,MAAM,aAAa,GAAkB;YACnC,yBAAyB,EAAE,oBAAoB;YAC/C,cAAc,EAAE,SAAS;YACzB,gBAAgB,EAAE,OAAO;SAC1B,CAAC;QAEF,6DAA6D;QAC7D,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;QAE3D,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,QAAQ,GAAwB;gBACpC,cAAc,EAAE,oBAAoB;gBACpC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE;oBAC9D,IAAI,EAAE,WAAW,CAAC,mBAAmB;oBACrC,WAAW,EAAE,cAAc,EAAE,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC;iBAChF,CAAC;aACH,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvB,IAAI,CAAC,0BAA0B,CAAC,KAAK,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QACD,mCAAmC;QACnC,OAAO,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,SAAyB,EACzB,cAAsC;QAEtC,MAAM,kBAAkB,GAAG,CACzB,MAAM,IAAI,CAAC,sBAAsB,CAAC,sBAAsB,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,cAAc,CAAC,CAC7F,CAAC,2BAA2B,EAAE,CAAC;QAEhC,MAAM,YAAY,GAAG,6BAA6B,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAE3E,MAAM,SAAS,GAAG,gBAAgB,CAAC,YAAY,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAE9E,MAAM,aAAa,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CACtD,YAAY,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,SAAS,CAAC,CAChE,CAAC;QAEF,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC9E,CAAC;QACD,OAAO,aAAa,CAAC,EAAE,CAAC;IAC1B,CAAC;IAEO,cAAc;QACpB,MAAM,eAAe,GAAG,IAAI,2BAA2B,CAAC,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC,CAAC;QAC9F,OAAO,IAAI,wBAAwB,CAAC,eAAe,CAAC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,UAA2B,EAC3B,cAAsC;QAEtC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,aAAa,CAAC,mDAAmD,CAAC,CAAC;QAC/E,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,MAAM;YAAE,OAAO;QAC/B,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC3D,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;QACpE,MAAM,WAAW,GAAgB,EAAE,CAAC;QACpC,KAAK,MAAM,aAAa,IAAI,UAAU,EAAE,CAAC;YACvC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;QAClD,CAAC;QACD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CACvC,KAAK,EAAE,SAAiC,EAAE,EAAE,CAC1C,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,WAAW;gBACjB,mBAAmB,EAAE,SAAS;gBAC9B,IAAI,EAAE,IAAI;gBACV,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG;gBAC9B,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,cAAc,EAAE,SAAS;aAC1B,CAAC,EACJ,cAAc,EACd,kBAAkB,CAAC,aAAa,CACjC,CAAC;YACF,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,aAAa,CAAC,gCAAgC,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,YAAY,CAAC,mBAAmB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAChE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,aAAa,CAAC,gBAAgB,EAAE,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,SAAS,CAAC,0CAA0C,CAAC,KAAK,CAAC,CAAC;gBACzE,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;oBAChC,OAAO,YAAY,CAAC,mBAAmB,CAAC,UAAU,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC;YACD,OAAO,YAAY,CAAC,mBAAmB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,cAA8B;QACrD,cAAc,CAAC,YAAY,GAAG,6BAA6B,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACzF,OAAO;YACL,GAAG,cAAc;YACjB,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,YAAY,CAAC;SAC7C,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,SAAwB,EACxB,cAAsC;QAEtC,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAC/D,SAAS,CAAC,cAAc,EACxB,cAAc,CACf,CAAC;QACF,SAAS,CAAC,gBAAgB,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC,CAAC;QAC7D,MAAM,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,KAAoB;QAChD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,4BAA4B,CAAC,MAAM,EAAE,EAAE,CAAC;YAChE,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACzB,CAAC;QACD,IAAI,CAAC,4BAA4B,CAAC,KAAK,EAAE,CAAC;IAC5C,CAAC;IAEO,mBAAmB,CAAC,SAAiB;QAC3C,IAAI,IAAI,CAAC,4BAA4B,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YACrD,OAAO,IAAI,CAAC,4BAA4B,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,kBAAkB,CACtC,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,6BAA6B,EAClC,IAAI,CAAC,aAAa,CAAC,eAAe,EAClC,IAAI,CAAC,aAAa,CAAC,gBAAgB,EACnC,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,EACpC,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAClC,IAAI,CAAC,0BAA0B,CAChC,CAAC;QACF,IAAI,CAAC,4BAA4B,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC5D,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,yBAAyB;QAC/B,IAAI,CAAC,sBAAsB,GAAG,WAAW,CAAC,GAAG,EAAE;YAC7C,IAAI,CAAC,4BAA4B,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACnD,MAAM,CAAC,sBAAsB,EAAE,CAAC;YAClC,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,0BAA0B,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,6BAA6B,CACzC,cAAsC;QAEtC,MAAM,IAAI,CAAC,sBAAsB,CAAC,sBAAsB,CACtD,IAAI,CAAC,SAAS,CAAC,GAAG,EAClB,cAAc,EACd,IAAI,CACL,CAAC;IACJ,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { RequestOptions } from \"../request/RequestOptions.js\";\nimport { readPartitionKeyDefinition } from \"../client/ClientUtils.js\";\nimport type { Container } from \"../client/index.js\";\nimport type { ClientContext } from \"../ClientContext.js\";\nimport { Constants, ResourceType } from \"../common/constants.js\";\nimport { sleep, copyObject, getPathFromLink } from \"../common/helper.js\";\nimport { StatusCodes } from \"../common/statusCodes.js\";\nimport {\n  DiagnosticNodeInternal,\n  DiagnosticNodeType,\n} from \"../diagnostics/DiagnosticNodeInternal.js\";\nimport type { PartitionKeyDefinition } from \"../documents/PartitionKeyDefinition.js\";\nimport { convertToInternalPartitionKey } from \"../documents/PartitionKeyInternal.js\";\nimport { ErrorResponse } from \"../index.js\";\nimport { BulkExecutionRetryPolicy } from \"../retry/bulkExecutionRetryPolicy.js\";\nimport { ResourceThrottleRetryPolicy } from \"../retry/resourceThrottleRetryPolicy.js\";\nimport type { RetryPolicy } from \"../retry/RetryPolicy.js\";\nimport type { PartitionKeyRangeCache } from \"../routing/partitionKeyRangeCache.js\";\nimport type { BulkOperationResult, OperationInput, Operation } from \"../utils/batch.js\";\nimport { encryptOperationInput, isKeyInRange } from \"../utils/batch.js\";\nimport { addDiagnosticChild } from \"../utils/diagnostics.js\";\nimport { hashPartitionKey } from \"../utils/hashing/hash.js\";\nimport { HelperPerPartition } from \"./HelperPerPartition.js\";\nimport type { ItemOperation } from \"./index.js\";\nimport { ItemOperationContext, BulkResponse } from \"./index.js\";\n\n/**\n * BulkHelper for bulk operations in a container.\n * It maintains one @see {@link HelperPerPartition} for each Partition Key Range, which allows independent execution of requests. Queue based limiters @see {@link LimiterQueue}\n * rate limit requestsbat the helper / Partition Key Range level, this means that we can send parallel and independent requests to different Partition Key Ranges, but for the same Range, requests\n * will be limited. Two callback implementations define how a particular request should be executed, and how operations should be retried. When the helper dispatches a batch\n * the batch will create a request and call the execute callback (executeRequest), if conditions are met, it might call the retry callback (reBatchOperation).\n * @hidden\n */\n\nexport class BulkHelper {\n  private readonly container: Container;\n  private readonly clientContext: ClientContext;\n  private readonly partitionKeyRangeCache: PartitionKeyRangeCache;\n  private readonly helpersByPartitionKeyRangeId: Map<string, HelperPerPartition>;\n  private options: RequestOptions;\n  private partitionKeyDefinition: PartitionKeyDefinition;\n  private partitionKeyDefinitionPromise: Promise<PartitionKeyDefinition>;\n  private isCancelled: boolean;\n  private processedOperationCountRef: { count: number } = { count: 0 };\n  private operationPromisesList: Promise<BulkOperationResult>[] = [];\n  private congestionControlTimer: NodeJS.Timeout;\n  private readonly congestionControlDelayInMs: number = 1000;\n  private staleRidError: ErrorResponse | undefined;\n  private readonly operationsPerSleep: number = 100; // Number of operations to add per sleep\n  private readonly intervalForPartialBatchInMs: number = 1000; // Sleep interval before adding partial batch to dispatch queue\n\n  /**\n   * @internal\n   */\n  constructor(\n    container: Container,\n    clientContext: ClientContext,\n    partitionKeyRangeCache: PartitionKeyRangeCache,\n    options: RequestOptions,\n  ) {\n    this.container = container;\n    this.clientContext = clientContext;\n    this.partitionKeyRangeCache = partitionKeyRangeCache;\n    this.helpersByPartitionKeyRangeId = new Map();\n    this.options = options;\n    this.executeRequest = this.executeRequest.bind(this);\n    this.reBatchOperation = this.reBatchOperation.bind(this);\n    this.refreshPartitionKeyRangeCache = this.refreshPartitionKeyRangeCache.bind(this);\n    this.isCancelled = false;\n    this.runCongestionControlTimer();\n  }\n\n  /**\n   * adds operation(s) to the helper\n   * @param operationInput - bulk operation or list of bulk operations\n   */\n  async execute(operationInput: OperationInput[]): Promise<BulkOperationResult[]> {\n    const addOperationPromises: Promise<void>[] = [];\n    const minimalPause = 0; // minimal pause (0 ms) inserted periodically during processing.\n    try {\n      for (let i = 0; i < operationInput.length; i++) {\n        // After every 100 operations,sleep of 0 ms is added to allow the event loop to process any pending\n        // callbacks/tasks such as fetching partition key definition and dispatching batches from queue. This helps\n        // to prevent blocking and improves overall responsiveness.\n        if (i % this.operationsPerSleep === 0) {\n          await sleep(minimalPause);\n        }\n        addOperationPromises.push(this.addOperation(operationInput[i], i));\n      }\n      await Promise.allSettled(addOperationPromises);\n\n      // After processing all operations via addOperation, it's possible that the current batch in each helper is not completely full.\n      // In such cases, addPartialBatchToQueue is called to ensure that all the operations are added to the dispatch queue.\n      // while loop below waits until the count of processed operations equals the number of input operations. This is necessary because\n      // some operations might fail and then again get added to current batch for retry.\n      while (this.processedOperationCountRef.count < operationInput.length) {\n        this.helpersByPartitionKeyRangeId.forEach((helper) => {\n          helper.addPartialBatchToQueue();\n        });\n        // Pause for 1000 ms to give pending operations chance to accumulate into a batch to avoid sending multiple small batches.\n        await sleep(this.intervalForPartialBatchInMs);\n      }\n    } finally {\n      if (this.congestionControlTimer) {\n        clearInterval(this.congestionControlTimer);\n      }\n    }\n    const settledResults = await Promise.allSettled(this.operationPromisesList);\n    if (this.isCancelled && this.staleRidError) {\n      throw this.staleRidError;\n    }\n\n    const bulkOperationResults = settledResults.map((result) =>\n      result.status === \"fulfilled\" ? result.value : result.reason,\n    );\n\n    // Formatting result: if an error is present, removing the stack trace details.\n    const formattedResults = bulkOperationResults.map((result) => {\n      if (result && result.error) {\n        const { stack, ...otherProps } = result.error;\n        const trimmedError = { message: result.error.message, ...otherProps };\n        return {\n          ...result,\n          error: trimmedError,\n        };\n      }\n      return result;\n    });\n    return formattedResults;\n  }\n\n  private async addOperation(operation: OperationInput, idx: number): Promise<void> {\n    if (this.isCancelled) {\n      return;\n    }\n    if (!operation) {\n      this.operationPromisesList[idx] = Promise.resolve({\n        operationInput: operation,\n        error: Object.assign(new ErrorResponse(\"Operation cannot be null or undefined.\"), {\n          code: StatusCodes.InternalServerError,\n        }),\n      });\n      return;\n    }\n\n    // Checks for id and partition key in input body\n    if (\n      operation.operationType === \"Create\" ||\n      operation.operationType === \"Upsert\" ||\n      operation.operationType === \"Replace\"\n    ) {\n      if (!operation.resourceBody.id) {\n        this.operationPromisesList[idx] = Promise.resolve({\n          operationInput: operation,\n          error: Object.assign(\n            new ErrorResponse(\n              `Operation resource body must have an 'id' for ${operation.operationType} operations.`,\n            ),\n            { code: StatusCodes.InternalServerError },\n          ),\n        });\n        this.processedOperationCountRef.count++;\n        return;\n      }\n    }\n    if (operation.partitionKey === undefined) {\n      this.operationPromisesList[idx] = Promise.resolve({\n        operationInput: operation,\n        error: Object.assign(\n          new ErrorResponse(`PartitionKey is required for ${operation.operationType} operations.`),\n          { code: StatusCodes.InternalServerError },\n        ),\n      });\n      this.processedOperationCountRef.count++;\n      return;\n    }\n\n    let operationError: Error | undefined;\n    let diagnosticNode: DiagnosticNodeInternal;\n    let unencryptedOperation: OperationInput;\n    let partitionKeyRangeId: string;\n    try {\n      diagnosticNode = new DiagnosticNodeInternal(\n        this.clientContext.diagnosticLevel,\n        DiagnosticNodeType.CLIENT_REQUEST_NODE,\n        null,\n      );\n      // Ensure partition key definition is available.\n      if (!this.partitionKeyDefinition) {\n        if (!this.partitionKeyDefinitionPromise) {\n          this.partitionKeyDefinitionPromise = (async () => {\n            try {\n              const partitionKeyDefinition = await readPartitionKeyDefinition(\n                diagnosticNode,\n                this.container,\n              );\n              this.partitionKeyDefinition = partitionKeyDefinition;\n              return partitionKeyDefinition;\n            } finally {\n              this.partitionKeyDefinitionPromise = null;\n            }\n          })();\n        }\n        await this.partitionKeyDefinitionPromise;\n      }\n      unencryptedOperation = copyObject(operation);\n      // If encryption is enabled, encrypt the operation input.\n      if (this.clientContext.enableEncryption) {\n        operation = copyObject(operation);\n        await this.container.checkAndInitializeEncryption();\n        diagnosticNode.beginEncryptionDiagnostics(Constants.Encryption.DiagnosticsEncryptOperation);\n        const { operation: encryptedOp, totalPropertiesEncryptedCount } =\n          await encryptOperationInput(this.container.encryptionProcessor, operation, 0);\n        operation = encryptedOp;\n        diagnosticNode.endEncryptionDiagnostics(\n          Constants.Encryption.DiagnosticsEncryptOperation,\n          totalPropertiesEncryptedCount,\n        );\n      }\n      // Resolve the partition key range id.\n      partitionKeyRangeId = await this.resolvePartitionKeyRangeId(operation, diagnosticNode);\n    } catch (error) {\n      operationError = error;\n    }\n\n    // Get helper & context.\n    const helperForPartition = this.getHelperForPKRange(partitionKeyRangeId);\n    const retryPolicy = this.getRetryPolicy();\n    const context = new ItemOperationContext(partitionKeyRangeId, retryPolicy, diagnosticNode);\n    const itemOperation: ItemOperation = {\n      unencryptedOperationInput: unencryptedOperation,\n      operationInput: operation,\n      operationContext: context,\n    };\n\n    // Assign the promise (ensuring position matches input order)\n    this.operationPromisesList[idx] = context.operationPromise;\n\n    if (operationError) {\n      const response: BulkOperationResult = {\n        operationInput: unencryptedOperation,\n        error: Object.assign(new ErrorResponse(operationError.message), {\n          code: StatusCodes.InternalServerError,\n          diagnostics: diagnosticNode?.toDiagnostic(this.clientContext.getClientConfig()),\n        }),\n      };\n      context.fail(response);\n      this.processedOperationCountRef.count++;\n      return;\n    }\n    // Add the operation to the helper.\n    return helperForPartition.add(itemOperation);\n  }\n\n  private async resolvePartitionKeyRangeId(\n    operation: OperationInput,\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<string> {\n    const partitionKeyRanges = (\n      await this.partitionKeyRangeCache.onCollectionRoutingMap(this.container.url, diagnosticNode)\n    ).getOrderedParitionKeyRanges();\n\n    const partitionKey = convertToInternalPartitionKey(operation.partitionKey);\n\n    const hashedKey = hashPartitionKey(partitionKey, this.partitionKeyDefinition);\n\n    const matchingRange = partitionKeyRanges.find((range) =>\n      isKeyInRange(range.minInclusive, range.maxExclusive, hashedKey),\n    );\n\n    if (!matchingRange) {\n      throw new Error(\"No matching partition key range found for the operation.\");\n    }\n    return matchingRange.id;\n  }\n\n  private getRetryPolicy(): RetryPolicy {\n    const nextRetryPolicy = new ResourceThrottleRetryPolicy(this.clientContext.getRetryOptions());\n    return new BulkExecutionRetryPolicy(nextRetryPolicy);\n  }\n\n  private async executeRequest(\n    operations: ItemOperation[],\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<BulkResponse> {\n    if (this.isCancelled) {\n      throw new ErrorResponse(\"Bulk execution cancelled due to a previous error.\");\n    }\n    if (!operations.length) return;\n    const pkRangeId = operations[0].operationContext.pkRangeId;\n    const path = getPathFromLink(this.container.url, ResourceType.item);\n    const requestBody: Operation[] = [];\n    for (const itemOperation of operations) {\n      requestBody.push(this.prepareOperation(itemOperation.operationInput));\n    }\n    if (!this.options.containerRid) {\n      this.options.containerRid = this.container._rid;\n    }\n    try {\n      const response = await addDiagnosticChild(\n        async (childNode: DiagnosticNodeInternal) =>\n          this.clientContext.bulk({\n            body: requestBody,\n            partitionKeyRangeId: pkRangeId,\n            path: path,\n            resourceId: this.container.url,\n            options: this.options,\n            diagnosticNode: childNode,\n          }),\n        diagnosticNode,\n        DiagnosticNodeType.BATCH_REQUEST,\n      );\n      if (!response) {\n        throw new ErrorResponse(\"Failed to fetch bulk response.\");\n      }\n      return BulkResponse.fromResponseMessage(response, operations);\n    } catch (error) {\n      if (this.clientContext.enableEncryption) {\n        try {\n          await this.container.throwIfRequestNeedsARetryPostPolicyRefresh(error);\n        } catch (err) {\n          await this.cancelExecution(err);\n          return BulkResponse.createEmptyResponse(operations, 0, 0, {});\n        }\n      }\n      return BulkResponse.fromResponseMessage(error, operations);\n    }\n  }\n\n  private prepareOperation(operationInput: OperationInput): Operation {\n    operationInput.partitionKey = convertToInternalPartitionKey(operationInput.partitionKey);\n    return {\n      ...operationInput,\n      partitionKey: JSON.stringify(operationInput.partitionKey),\n    } as Operation;\n  }\n\n  private async reBatchOperation(\n    operation: ItemOperation,\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<void> {\n    const partitionKeyRangeId = await this.resolvePartitionKeyRangeId(\n      operation.operationInput,\n      diagnosticNode,\n    );\n    operation.operationContext.updatePKRangeId(partitionKeyRangeId);\n    const helper = this.getHelperForPKRange(partitionKeyRangeId);\n    await helper.add(operation);\n  }\n\n  private async cancelExecution(error: ErrorResponse): Promise<void> {\n    this.isCancelled = true;\n    this.staleRidError = error;\n    for (const helper of this.helpersByPartitionKeyRangeId.values()) {\n      await helper.dispose();\n    }\n    this.helpersByPartitionKeyRangeId.clear();\n  }\n\n  private getHelperForPKRange(pkRangeId: string): HelperPerPartition {\n    if (this.helpersByPartitionKeyRangeId.has(pkRangeId)) {\n      return this.helpersByPartitionKeyRangeId.get(pkRangeId);\n    }\n\n    const newHelper = new HelperPerPartition(\n      this.executeRequest,\n      this.reBatchOperation,\n      this.refreshPartitionKeyRangeCache,\n      this.clientContext.diagnosticLevel,\n      this.clientContext.enableEncryption,\n      this.clientContext.getClientConfig(),\n      this.container.encryptionProcessor,\n      this.processedOperationCountRef,\n    );\n    this.helpersByPartitionKeyRangeId.set(pkRangeId, newHelper);\n    return newHelper;\n  }\n\n  private runCongestionControlTimer(): void {\n    this.congestionControlTimer = setInterval(() => {\n      this.helpersByPartitionKeyRangeId.forEach((helper) => {\n        helper.runCongestionAlgorithm();\n      });\n    }, this.congestionControlDelayInMs);\n  }\n\n  private async refreshPartitionKeyRangeCache(\n    diagnosticNode: DiagnosticNodeInternal,\n  ): Promise<void> {\n    await this.partitionKeyRangeCache.onCollectionRoutingMap(\n      this.container.url,\n      diagnosticNode,\n      true,\n    );\n  }\n}\n"]}

package/dist/esm/bulk/BulkResponse.js

@@ -9,8 +9,13 @@ import { isSuccessStatusCode } from "../utils/batch.js";
  * @hidden
  */
 export class BulkResponse {
+    statusCode;
+    subStatusCode;
+    headers;
+    operations;
+    results = [];
+    diagnostics;
     constructor(statusCode, subStatusCode, headers, operations) {
-        this.results = [];
         this.statusCode = statusCode;
         this.subStatusCode = subStatusCode;
         this.headers = headers;
@@ -28,7 +33,6 @@ export class BulkResponse {
      * static method to create BulkResponse from Response object
      */
     static fromResponseMessage(responseMessage, operations) {
-        var _a;
         // Create and populate the response object
         let bulkResponse = this.populateFromResponse(responseMessage, operations);
         if (!bulkResponse.results || bulkResponse.results.length !== operations.length) {
@@ -40,7 +44,7 @@ export class BulkResponse {
             // When the overall response status code is TooManyRequests, propagate the RetryAfter into the individual operations.
             let retryAfterMilliseconds = 0;
             if (responseMessage.code === StatusCodes.TooManyRequests) {
-                const retryAfter = (_a = responseMessage.headers) === null || _a === void 0 ? void 0 : _a[Constants.HttpHeaders.RetryAfterInMs];
+                const retryAfter = responseMessage.headers?.[Constants.HttpHeaders.RetryAfterInMs];
                 retryAfterMilliseconds = !retryAfter || isNaN(Number(retryAfter)) ? 0 : Number(retryAfter);
             }
             bulkResponse.createAndPopulateResults(operations, retryAfterMilliseconds, responseMessage);
@@ -48,19 +52,18 @@ export class BulkResponse {
         return bulkResponse;
     }
     static populateFromResponse(responseMessage, operations) {
-        var _a, _b, _c;
         const results = [];
         if (responseMessage.result) {
             for (let i = 0; i < operations.length; i++) {
                 const itemResponse = responseMessage.result[i];
-                if (isSuccessStatusCode(itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.statusCode)) {
+                if (isSuccessStatusCode(itemResponse?.statusCode)) {
                     const result = {
-                        statusCode: itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.statusCode,
-                        eTag: itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.eTag,
-                        activityId: (_a = responseMessage.headers) === null || _a === void 0 ? void 0 : _a[Constants.HttpHeaders.ActivityId],
-                        sessionToken: (_b = responseMessage.headers) === null || _b === void 0 ? void 0 : _b[Constants.HttpHeaders.SessionToken],
-                        requestCharge: itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.requestCharge,
-                        resourceBody: itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.resourceBody,
+                        statusCode: itemResponse?.statusCode,
+                        eTag: itemResponse?.eTag,
+                        activityId: responseMessage.headers?.[Constants.HttpHeaders.ActivityId],
+                        sessionToken: responseMessage.headers?.[Constants.HttpHeaders.SessionToken],
+                        requestCharge: itemResponse?.requestCharge,
+                        resourceBody: itemResponse?.resourceBody,
                         // diagnostics will be filled in Batcher dispatch to capture the complete diagnostics(e.g. decryption)
                         diagnostics: null,
                         headers: responseMessage.headers,
@@ -69,14 +72,14 @@ export class BulkResponse {
                 }
                 else {
                     const error = new ErrorResponse();
-                    error.code = itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.statusCode;
-                    error.substatus = itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.subStatusCode;
-                    error.message = itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.message;
-                    error.requestCharge = itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.requestCharge;
-                    error.body = itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.resourceBody;
+                    error.code = itemResponse?.statusCode;
+                    error.substatus = itemResponse?.subStatusCode;
+                    error.message = itemResponse?.message;
+                    error.requestCharge = itemResponse?.requestCharge;
+                    error.body = itemResponse?.resourceBody;
                     error.headers = responseMessage.headers;
-                    error.activityId = (_c = responseMessage.headers) === null || _c === void 0 ? void 0 : _c[Constants.HttpHeaders.ActivityId];
-                    error.retryAfterInMs = itemResponse === null || itemResponse === void 0 ? void 0 : itemResponse.retryAfterMilliseconds;
+                    error.activityId = responseMessage.headers?.[Constants.HttpHeaders.ActivityId];
+                    error.retryAfterInMs = itemResponse?.retryAfterMilliseconds;
                     error.diagnostics = responseMessage.diagnostics;
                     results.push(error);
                 }
@@ -101,13 +104,12 @@ export class BulkResponse {
     }
     createAndPopulateResults(operations, retryAfterInMs, error) {
         this.results = operations.map(() => {
-            var _a;
             const errorResponse = new ErrorResponse();
             errorResponse.message = error.message;
             errorResponse.code = this.statusCode;
             errorResponse.substatus = this.subStatusCode;
             errorResponse.retryAfterInMs = retryAfterInMs;
-            errorResponse.activityId = (_a = this.headers) === null || _a === void 0 ? void 0 : _a[Constants.HttpHeaders.ActivityId];
+            errorResponse.activityId = this.headers?.[Constants.HttpHeaders.ActivityId];
             errorResponse.body = error.body;
             errorResponse.diagnostics = error.diagnostics;
             errorResponse.headers = this.headers;

package/dist/esm/bulk/BulkResponse.js.map

@@ -1 +1 @@
-{"version":3,"file":"BulkResponse.js","sourceRoot":"","sources":["../../../src/bulk/BulkResponse.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAEvE,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAI5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAGxD;;;GAGG;AAEH,MAAM,OAAO,YAAY;IAQvB,YACE,UAAsB,EACtB,aAA4B,EAC5B,OAAsB,EACtB,UAA2B;QAP7B,YAAO,GAAkD,EAAE,CAAC;QAS1D,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,mBAAmB,CACxB,UAA2B,EAC3B,UAAsB,EACtB,aAA4B,EAC5B,OAAsB;QAEtB,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,UAAU,EAAE,aAAa,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACtF,YAAY,CAAC,wBAAwB,CAAC,UAAU,EAAE,CAAC,EAAE,IAAI,aAAa,EAAE,CAAC,CAAC;QAC1E,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,mBAAmB,CACxB,eAA8B,EAC9B,UAA2B;;QAE3B,0CAA0C;QAC1C,IAAI,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QAE1E,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,MAAM,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;YAC/E,2EAA2E;YAC3E,wFAAwF;YACxF,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9C,YAAY,GAAG,IAAI,YAAY,CAC7B,WAAW,CAAC,mBAAmB,EAC/B,cAAc,CAAC,OAAO,EACtB,eAAe,CAAC,OAAO,EACvB,UAAU,CACX,CAAC;YACJ,CAAC;YAED,qHAAqH;YACrH,IAAI,sBAAsB,GAAG,CAAC,CAAC;YAE/B,IAAI,eAAe,CAAC,IAAI,KAAK,WAAW,CAAC,eAAe,EAAE,CAAC;gBACzD,MAAM,UAAU,GAAG,MAAA,eAAe,CAAC,OAAO,0CAAG,SAAS,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;gBACnF,sBAAsB,GAAG,CAAC,UAAU,IAAI,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC7F,CAAC;YAED,YAAY,CAAC,wBAAwB,CACnC,UAAU,EACV,sBAAsB,EACtB,eAAgC,CACjC,CAAC;QACJ,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,oBAAoB,CACjC,eAA8B,EAC9B,UAA2B;;QAE3B,MAAM,OAAO,GAAkD,EAAE,CAAC;QAElE,IAAI,eAAe,CAAC,MAAM,EAAE,CAAC;YAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3C,MAAM,YAAY,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAE/C,IAAI,mBAAmB,CAAC,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,UAAU,CAAC,EAAE,CAAC;oBAClD,MAAM,MAAM,GAA8B;wBACxC,UAAU,EAAE,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,UAAU;wBACpC,IAAI,EAAE,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,IAAI;wBACxB,UAAU,EAAE,MAAA,eAAe,CAAC,OAAO,0CAAG,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC;wBACvE,YAAY,EAAE,MAAA,eAAe,CAAC,OAAO,0CAAG,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC;wBAC3E,aAAa,EAAE,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,aAAa;wBAC1C,YAAY,EAAE,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,YAAY;wBACxC,sGAAsG;wBACtG,WAAW,EAAE,IAAI;wBACjB,OAAO,EAAE,eAAe,CAAC,OAAO;qBACjC,CAAC;oBACF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,GAAkB,IAAI,aAAa,EAAE,CAAC;oBACjD,KAAK,CAAC,IAAI,GAAG,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,UAAU,CAAC;oBACtC,KAAK,CAAC,SAAS,GAAG,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,aAAa,CAAC;oBAC9C,KAAK,CAAC,OAAO,GAAG,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,OAAO,CAAC;oBACtC,KAAK,CAAC,aAAa,GAAG,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,aAAa,CAAC;oBAClD,KAAK,CAAC,IAAI,GAAG,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,YAAY,CAAC;oBACxC,KAAK,CAAC,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC;oBACxC,KAAK,CAAC,UAAU,GAAG,MAAA,eAAe,CAAC,OAAO,0CAAG,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;oBAC/E,KAAK,CAAC,cAAc,GAAG,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,sBAAsB,CAAC;oBAC5D,KAAK,CAAC,WAAW,GAAG,eAAe,CAAC,WAAW,CAAC;oBAChD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC;QACtC,IAAI,aAAa,GAAG,eAAe,CAAC,SAAS,CAAC;QAE9C,IAAI,eAAe,CAAC,IAAI,KAAK,WAAW,CAAC,WAAW,EAAE,CAAC;YACrD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,IACE,MAAM,YAAY,aAAa;oBAC/B,MAAM,CAAC,UAAU,KAAK,WAAW,CAAC,gBAAgB;oBAClD,MAAM,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,EAC3C,CAAC;oBACD,UAAU,GAAG,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBACjF,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC;oBACjC,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,YAAY,CACnC,UAAU,EACV,aAAa,EACb,eAAe,CAAC,OAAO,EACvB,UAAU,CACX,CAAC;QACF,YAAY,CAAC,OAAO,GAAG,OAAO,CAAC;QAC/B,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,wBAAwB,CAC9B,UAA2B,EAC3B,cAAsB,EACtB,KAAoB;QAEpB,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE;;YACjC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;YAC1C,aAAa,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YACtC,aAAa,CAAC,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC;YACrC,aAAa,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;YAC7C,aAAa,CAAC,cAAc,GAAG,cAAc,CAAC;YAC9C,aAAa,CAAC,UAAU,GAAG,MAAA,IAAI,CAAC,OAAO,0CAAG,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YAC5E,aAAa,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;YAChC,aAAa,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;YAC9C,aAAa,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YACrC,aAAa,CAAC,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;YAClD,OAAO,aAAa,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Constants } from \"../common/constants.js\";\nimport { StatusCodes, SubStatusCodes } from \"../common/statusCodes.js\";\nimport type { CosmosDiagnostics, Response } from \"../index.js\";\nimport { ErrorResponse } from \"../index.js\";\nimport type { CosmosHeaders } from \"../queryExecutionContext/headerUtils.js\";\nimport type { StatusCode, SubStatusCode } from \"../request/StatusCodes.js\";\nimport type { ExtendedOperationResponse } from \"../utils/batch.js\";\nimport { isSuccessStatusCode } from \"../utils/batch.js\";\nimport type { ItemOperation } from \"./index.js\";\n\n/**\n * Represents a batch response for bulk request.\n * @hidden\n */\n\nexport class BulkResponse {\n  statusCode: StatusCode;\n  subStatusCode: SubStatusCode;\n  headers: CosmosHeaders;\n  operations: ItemOperation[];\n  results: (ExtendedOperationResponse | ErrorResponse)[] = [];\n  diagnostics: CosmosDiagnostics;\n\n  constructor(\n    statusCode: StatusCode,\n    subStatusCode: SubStatusCode,\n    headers: CosmosHeaders,\n    operations: ItemOperation[],\n  ) {\n    this.statusCode = statusCode;\n    this.subStatusCode = subStatusCode;\n    this.headers = headers;\n    this.operations = operations;\n  }\n\n  /**\n   * Generate empty response object\n   */\n  static createEmptyResponse(\n    operations: ItemOperation[],\n    statusCode: StatusCode,\n    subStatusCode: SubStatusCode,\n    headers: CosmosHeaders,\n  ): BulkResponse {\n    const bulkResponse = new BulkResponse(statusCode, subStatusCode, headers, operations);\n    bulkResponse.createAndPopulateResults(operations, 0, new ErrorResponse());\n    return bulkResponse;\n  }\n\n  /**\n   * static method to create BulkResponse from Response object\n   */\n  static fromResponseMessage(\n    responseMessage: Response<any>,\n    operations: ItemOperation[],\n  ): BulkResponse {\n    // Create and populate the response object\n    let bulkResponse = this.populateFromResponse(responseMessage, operations);\n\n    if (!bulkResponse.results || bulkResponse.results.length !== operations.length) {\n      // Server should be guaranteeing number of results equal to operations when\n      // batch request is successful - so fail as InternalServerError if this is not the case.\n      if (isSuccessStatusCode(responseMessage.code)) {\n        bulkResponse = new BulkResponse(\n          StatusCodes.InternalServerError,\n          SubStatusCodes.Unknown,\n          responseMessage.headers,\n          operations,\n        );\n      }\n\n      // When the overall response status code is TooManyRequests, propagate the RetryAfter into the individual operations.\n      let retryAfterMilliseconds = 0;\n\n      if (responseMessage.code === StatusCodes.TooManyRequests) {\n        const retryAfter = responseMessage.headers?.[Constants.HttpHeaders.RetryAfterInMs];\n        retryAfterMilliseconds = !retryAfter || isNaN(Number(retryAfter)) ? 0 : Number(retryAfter);\n      }\n\n      bulkResponse.createAndPopulateResults(\n        operations,\n        retryAfterMilliseconds,\n        responseMessage as ErrorResponse,\n      );\n    }\n    return bulkResponse;\n  }\n\n  private static populateFromResponse(\n    responseMessage: Response<any>,\n    operations: ItemOperation[],\n  ): BulkResponse {\n    const results: (ExtendedOperationResponse | ErrorResponse)[] = [];\n\n    if (responseMessage.result) {\n      for (let i = 0; i < operations.length; i++) {\n        const itemResponse = responseMessage.result[i];\n\n        if (isSuccessStatusCode(itemResponse?.statusCode)) {\n          const result: ExtendedOperationResponse = {\n            statusCode: itemResponse?.statusCode,\n            eTag: itemResponse?.eTag,\n            activityId: responseMessage.headers?.[Constants.HttpHeaders.ActivityId],\n            sessionToken: responseMessage.headers?.[Constants.HttpHeaders.SessionToken],\n            requestCharge: itemResponse?.requestCharge,\n            resourceBody: itemResponse?.resourceBody,\n            // diagnostics will be filled in Batcher dispatch to capture the complete diagnostics(e.g. decryption)\n            diagnostics: null,\n            headers: responseMessage.headers,\n          };\n          results.push(result);\n        } else {\n          const error: ErrorResponse = new ErrorResponse();\n          error.code = itemResponse?.statusCode;\n          error.substatus = itemResponse?.subStatusCode;\n          error.message = itemResponse?.message;\n          error.requestCharge = itemResponse?.requestCharge;\n          error.body = itemResponse?.resourceBody;\n          error.headers = responseMessage.headers;\n          error.activityId = responseMessage.headers?.[Constants.HttpHeaders.ActivityId];\n          error.retryAfterInMs = itemResponse?.retryAfterMilliseconds;\n          error.diagnostics = responseMessage.diagnostics;\n          results.push(error);\n        }\n      }\n    }\n    let statusCode = responseMessage.code;\n    let subStatusCode = responseMessage.substatus;\n\n    if (responseMessage.code === StatusCodes.MultiStatus) {\n      for (const result of results) {\n        if (\n          result instanceof ErrorResponse &&\n          result.statusCode !== StatusCodes.FailedDependency &&\n          result.statusCode >= StatusCodes.BadRequest\n        ) {\n          statusCode = typeof result.code === \"number\" ? result.code : Number(result.code);\n          subStatusCode = result.substatus;\n          break;\n        }\n      }\n    }\n\n    const bulkResponse = new BulkResponse(\n      statusCode,\n      subStatusCode,\n      responseMessage.headers,\n      operations,\n    );\n    bulkResponse.results = results;\n    return bulkResponse;\n  }\n\n  private createAndPopulateResults(\n    operations: ItemOperation[],\n    retryAfterInMs: number,\n    error: ErrorResponse,\n  ): void {\n    this.results = operations.map(() => {\n      const errorResponse = new ErrorResponse();\n      errorResponse.message = error.message;\n      errorResponse.code = this.statusCode;\n      errorResponse.substatus = this.subStatusCode;\n      errorResponse.retryAfterInMs = retryAfterInMs;\n      errorResponse.activityId = this.headers?.[Constants.HttpHeaders.ActivityId];\n      errorResponse.body = error.body;\n      errorResponse.diagnostics = error.diagnostics;\n      errorResponse.headers = this.headers;\n      errorResponse.requestCharge = error.requestCharge;\n      return errorResponse;\n    });\n  }\n}\n"]}
+{"version":3,"file":"BulkResponse.js","sourceRoot":"","sources":["../../../src/bulk/BulkResponse.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAEvE,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAI5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAGxD;;;GAGG;AAEH,MAAM,OAAO,YAAY;IACvB,UAAU,CAAa;IACvB,aAAa,CAAgB;IAC7B,OAAO,CAAgB;IACvB,UAAU,CAAkB;IAC5B,OAAO,GAAkD,EAAE,CAAC;IAC5D,WAAW,CAAoB;IAE/B,YACE,UAAsB,EACtB,aAA4B,EAC5B,OAAsB,EACtB,UAA2B;QAE3B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,mBAAmB,CACxB,UAA2B,EAC3B,UAAsB,EACtB,aAA4B,EAC5B,OAAsB;QAEtB,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,UAAU,EAAE,aAAa,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACtF,YAAY,CAAC,wBAAwB,CAAC,UAAU,EAAE,CAAC,EAAE,IAAI,aAAa,EAAE,CAAC,CAAC;QAC1E,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,mBAAmB,CACxB,eAA8B,EAC9B,UAA2B;QAE3B,0CAA0C;QAC1C,IAAI,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;QAE1E,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,MAAM,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;YAC/E,2EAA2E;YAC3E,wFAAwF;YACxF,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9C,YAAY,GAAG,IAAI,YAAY,CAC7B,WAAW,CAAC,mBAAmB,EAC/B,cAAc,CAAC,OAAO,EACtB,eAAe,CAAC,OAAO,EACvB,UAAU,CACX,CAAC;YACJ,CAAC;YAED,qHAAqH;YACrH,IAAI,sBAAsB,GAAG,CAAC,CAAC;YAE/B,IAAI,eAAe,CAAC,IAAI,KAAK,WAAW,CAAC,eAAe,EAAE,CAAC;gBACzD,MAAM,UAAU,GAAG,eAAe,CAAC,OAAO,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;gBACnF,sBAAsB,GAAG,CAAC,UAAU,IAAI,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC7F,CAAC;YAED,YAAY,CAAC,wBAAwB,CACnC,UAAU,EACV,sBAAsB,EACtB,eAAgC,CACjC,CAAC;QACJ,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,oBAAoB,CACjC,eAA8B,EAC9B,UAA2B;QAE3B,MAAM,OAAO,GAAkD,EAAE,CAAC;QAElE,IAAI,eAAe,CAAC,MAAM,EAAE,CAAC;YAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3C,MAAM,YAAY,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAE/C,IAAI,mBAAmB,CAAC,YAAY,EAAE,UAAU,CAAC,EAAE,CAAC;oBAClD,MAAM,MAAM,GAA8B;wBACxC,UAAU,EAAE,YAAY,EAAE,UAAU;wBACpC,IAAI,EAAE,YAAY,EAAE,IAAI;wBACxB,UAAU,EAAE,eAAe,CAAC,OAAO,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC;wBACvE,YAAY,EAAE,eAAe,CAAC,OAAO,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC;wBAC3E,aAAa,EAAE,YAAY,EAAE,aAAa;wBAC1C,YAAY,EAAE,YAAY,EAAE,YAAY;wBACxC,sGAAsG;wBACtG,WAAW,EAAE,IAAI;wBACjB,OAAO,EAAE,eAAe,CAAC,OAAO;qBACjC,CAAC;oBACF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,GAAkB,IAAI,aAAa,EAAE,CAAC;oBACjD,KAAK,CAAC,IAAI,GAAG,YAAY,EAAE,UAAU,CAAC;oBACtC,KAAK,CAAC,SAAS,GAAG,YAAY,EAAE,aAAa,CAAC;oBAC9C,KAAK,CAAC,OAAO,GAAG,YAAY,EAAE,OAAO,CAAC;oBACtC,KAAK,CAAC,aAAa,GAAG,YAAY,EAAE,aAAa,CAAC;oBAClD,KAAK,CAAC,IAAI,GAAG,YAAY,EAAE,YAAY,CAAC;oBACxC,KAAK,CAAC,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC;oBACxC,KAAK,CAAC,UAAU,GAAG,eAAe,CAAC,OAAO,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;oBAC/E,KAAK,CAAC,cAAc,GAAG,YAAY,EAAE,sBAAsB,CAAC;oBAC5D,KAAK,CAAC,WAAW,GAAG,eAAe,CAAC,WAAW,CAAC;oBAChD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC;QACtC,IAAI,aAAa,GAAG,eAAe,CAAC,SAAS,CAAC;QAE9C,IAAI,eAAe,CAAC,IAAI,KAAK,WAAW,CAAC,WAAW,EAAE,CAAC;YACrD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,IACE,MAAM,YAAY,aAAa;oBAC/B,MAAM,CAAC,UAAU,KAAK,WAAW,CAAC,gBAAgB;oBAClD,MAAM,CAAC,UAAU,IAAI,WAAW,CAAC,UAAU,EAC3C,CAAC;oBACD,UAAU,GAAG,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBACjF,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC;oBACjC,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,YAAY,CACnC,UAAU,EACV,aAAa,EACb,eAAe,CAAC,OAAO,EACvB,UAAU,CACX,CAAC;QACF,YAAY,CAAC,OAAO,GAAG,OAAO,CAAC;QAC/B,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,wBAAwB,CAC9B,UAA2B,EAC3B,cAAsB,EACtB,KAAoB;QAEpB,IAAI,CAAC,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE;YACjC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;YAC1C,aAAa,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;YACtC,aAAa,CAAC,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC;YACrC,aAAa,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;YAC7C,aAAa,CAAC,cAAc,GAAG,cAAc,CAAC;YAC9C,aAAa,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YAC5E,aAAa,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;YAChC,aAAa,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;YAC9C,aAAa,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YACrC,aAAa,CAAC,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;YAClD,OAAO,aAAa,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Constants } from \"../common/constants.js\";\nimport { StatusCodes, SubStatusCodes } from \"../common/statusCodes.js\";\nimport type { CosmosDiagnostics, Response } from \"../index.js\";\nimport { ErrorResponse } from \"../index.js\";\nimport type { CosmosHeaders } from \"../queryExecutionContext/headerUtils.js\";\nimport type { StatusCode, SubStatusCode } from \"../request/StatusCodes.js\";\nimport type { ExtendedOperationResponse } from \"../utils/batch.js\";\nimport { isSuccessStatusCode } from \"../utils/batch.js\";\nimport type { ItemOperation } from \"./index.js\";\n\n/**\n * Represents a batch response for bulk request.\n * @hidden\n */\n\nexport class BulkResponse {\n  statusCode: StatusCode;\n  subStatusCode: SubStatusCode;\n  headers: CosmosHeaders;\n  operations: ItemOperation[];\n  results: (ExtendedOperationResponse | ErrorResponse)[] = [];\n  diagnostics: CosmosDiagnostics;\n\n  constructor(\n    statusCode: StatusCode,\n    subStatusCode: SubStatusCode,\n    headers: CosmosHeaders,\n    operations: ItemOperation[],\n  ) {\n    this.statusCode = statusCode;\n    this.subStatusCode = subStatusCode;\n    this.headers = headers;\n    this.operations = operations;\n  }\n\n  /**\n   * Generate empty response object\n   */\n  static createEmptyResponse(\n    operations: ItemOperation[],\n    statusCode: StatusCode,\n    subStatusCode: SubStatusCode,\n    headers: CosmosHeaders,\n  ): BulkResponse {\n    const bulkResponse = new BulkResponse(statusCode, subStatusCode, headers, operations);\n    bulkResponse.createAndPopulateResults(operations, 0, new ErrorResponse());\n    return bulkResponse;\n  }\n\n  /**\n   * static method to create BulkResponse from Response object\n   */\n  static fromResponseMessage(\n    responseMessage: Response<any>,\n    operations: ItemOperation[],\n  ): BulkResponse {\n    // Create and populate the response object\n    let bulkResponse = this.populateFromResponse(responseMessage, operations);\n\n    if (!bulkResponse.results || bulkResponse.results.length !== operations.length) {\n      // Server should be guaranteeing number of results equal to operations when\n      // batch request is successful - so fail as InternalServerError if this is not the case.\n      if (isSuccessStatusCode(responseMessage.code)) {\n        bulkResponse = new BulkResponse(\n          StatusCodes.InternalServerError,\n          SubStatusCodes.Unknown,\n          responseMessage.headers,\n          operations,\n        );\n      }\n\n      // When the overall response status code is TooManyRequests, propagate the RetryAfter into the individual operations.\n      let retryAfterMilliseconds = 0;\n\n      if (responseMessage.code === StatusCodes.TooManyRequests) {\n        const retryAfter = responseMessage.headers?.[Constants.HttpHeaders.RetryAfterInMs];\n        retryAfterMilliseconds = !retryAfter || isNaN(Number(retryAfter)) ? 0 : Number(retryAfter);\n      }\n\n      bulkResponse.createAndPopulateResults(\n        operations,\n        retryAfterMilliseconds,\n        responseMessage as ErrorResponse,\n      );\n    }\n    return bulkResponse;\n  }\n\n  private static populateFromResponse(\n    responseMessage: Response<any>,\n    operations: ItemOperation[],\n  ): BulkResponse {\n    const results: (ExtendedOperationResponse | ErrorResponse)[] = [];\n\n    if (responseMessage.result) {\n      for (let i = 0; i < operations.length; i++) {\n        const itemResponse = responseMessage.result[i];\n\n        if (isSuccessStatusCode(itemResponse?.statusCode)) {\n          const result: ExtendedOperationResponse = {\n            statusCode: itemResponse?.statusCode,\n            eTag: itemResponse?.eTag,\n            activityId: responseMessage.headers?.[Constants.HttpHeaders.ActivityId],\n            sessionToken: responseMessage.headers?.[Constants.HttpHeaders.SessionToken],\n            requestCharge: itemResponse?.requestCharge,\n            resourceBody: itemResponse?.resourceBody,\n            // diagnostics will be filled in Batcher dispatch to capture the complete diagnostics(e.g. decryption)\n            diagnostics: null,\n            headers: responseMessage.headers,\n          };\n          results.push(result);\n        } else {\n          const error: ErrorResponse = new ErrorResponse();\n          error.code = itemResponse?.statusCode;\n          error.substatus = itemResponse?.subStatusCode;\n          error.message = itemResponse?.message;\n          error.requestCharge = itemResponse?.requestCharge;\n          error.body = itemResponse?.resourceBody;\n          error.headers = responseMessage.headers;\n          error.activityId = responseMessage.headers?.[Constants.HttpHeaders.ActivityId];\n          error.retryAfterInMs = itemResponse?.retryAfterMilliseconds;\n          error.diagnostics = responseMessage.diagnostics;\n          results.push(error);\n        }\n      }\n    }\n    let statusCode = responseMessage.code;\n    let subStatusCode = responseMessage.substatus;\n\n    if (responseMessage.code === StatusCodes.MultiStatus) {\n      for (const result of results) {\n        if (\n          result instanceof ErrorResponse &&\n          result.statusCode !== StatusCodes.FailedDependency &&\n          result.statusCode >= StatusCodes.BadRequest\n        ) {\n          statusCode = typeof result.code === \"number\" ? result.code : Number(result.code);\n          subStatusCode = result.substatus;\n          break;\n        }\n      }\n    }\n\n    const bulkResponse = new BulkResponse(\n      statusCode,\n      subStatusCode,\n      responseMessage.headers,\n      operations,\n    );\n    bulkResponse.results = results;\n    return bulkResponse;\n  }\n\n  private createAndPopulateResults(\n    operations: ItemOperation[],\n    retryAfterInMs: number,\n    error: ErrorResponse,\n  ): void {\n    this.results = operations.map(() => {\n      const errorResponse = new ErrorResponse();\n      errorResponse.message = error.message;\n      errorResponse.code = this.statusCode;\n      errorResponse.substatus = this.subStatusCode;\n      errorResponse.retryAfterInMs = retryAfterInMs;\n      errorResponse.activityId = this.headers?.[Constants.HttpHeaders.ActivityId];\n      errorResponse.body = error.body;\n      errorResponse.diagnostics = error.diagnostics;\n      errorResponse.headers = this.headers;\n      errorResponse.requestCharge = error.requestCharge;\n      return errorResponse;\n    });\n  }\n}\n"]}