@azure/cosmos 4.9.1 → 4.9.2

package/dist/esm/index.js

@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 export { DEFAULT_PARTITION_KEY_PATH } from "./common/partitionKeys.js";
-export { StatusCodes } from "./common/index.js";
+export { StatusCodes, } from "./common/index.js";
 export { setAuthorizationTokenHeaderUsingMasterKey } from "./auth.js";
 export { BulkOperationType, } from "./utils/batch.js";
 export { PatchOperationType, } from "./utils/patch.js";

package/dist/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,WAAW,EAAqD,MAAM,mBAAmB,CAAC;AACnG,OAAO,EAAE,yCAAyC,EAAE,MAAM,WAAW,CAAC;AACtE,OAAO,EAYL,iBAAiB,GAWlB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAEL,kBAAkB,GAInB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,cAAc,EACd,gBAAgB,EAEhB,eAAe,EACf,QAAQ,EAGR,YAAY,EAGZ,WAAW,EACX,cAAc,EACd,SAAS,EAMT,gBAAgB,EAEhB,6BAA6B,EAC7B,mBAAmB,EACnB,cAAc,EACd,aAAa,EACb,gBAAgB,EAChB,WAAW,EACX,uBAAuB,EAMvB,uBAAuB,EACvB,+BAA+B,EAC/B,eAAe,GAIhB,MAAM,sBAAsB,CAAC;AAI9B,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEvF,cAAc,oBAAoB,CAAC;AAEnC,OAAO,EACL,sBAAsB,EAEtB,kBAAkB,GACnB,MAAM,yCAAyC,CAAC;AAUjD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,cAAc,yBAAyB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,cAAc,mBAAmB,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EAA8B,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAG3E,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,EACL,iBAAiB,EAGjB,kBAAkB,GAQnB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAGL,0BAA0B,EAC1B,mBAAmB,EACnB,SAAS,EACT,cAAc,EACd,gBAAgB,EAChB,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,uBAAuB,EAAE,MAAM,0CAA0C,CAAC;AAEnF,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACrD,cAAc,6BAA6B,CAAC;AAC5C,cAAc,2CAA2C,CAAC;AAC1D,cAAc,6CAA6C,CAAC;AAC5D,OAAO,EAKL,sBAAsB,EAEtB,2BAA2B,EAE3B,kCAAkC,EAClC,cAAc,EACd,mBAAmB,EACnB,yBAAyB,EAGzB,yBAAyB,GAC1B,MAAM,uBAAuB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nexport { DEFAULT_PARTITION_KEY_PATH } from \"./common/partitionKeys.js\";\nexport { StatusCodes, StatusCodesType, PartitionKeyRangePropertiesNames } from \"./common/index.js\";\nexport { setAuthorizationTokenHeaderUsingMasterKey } from \"./auth.js\";\nexport {\n  Operation,\n  OperationResponse,\n  BulkOptions,\n  CreateOperation,\n  UpsertOperation,\n  ReplaceOperation,\n  DeleteOperation,\n  ReadOperation,\n  OperationBase,\n  OperationWithItem,\n  OperationInput,\n  BulkOperationType,\n  ExtendedOperationResponse,\n  BulkOperationResponse,\n  CreateOperationInput,\n  UpsertOperationInput,\n  ReplaceOperationInput,\n  ReadOperationInput,\n  DeleteOperationInput,\n  PatchOperationInput,\n  BulkPatchOperation,\n  BulkOperationResult,\n} from \"./utils/batch.js\";\nexport {\n  PatchOperation,\n  PatchOperationType,\n  ExistingKeyOperation,\n  RemoveOperation,\n  PatchRequestBody,\n} from \"./utils/patch.js\";\nexport {\n  ConnectionMode,\n  ConsistencyLevel,\n  ConnectionPolicy,\n  DatabaseAccount,\n  DataType,\n  Index,\n  IndexedPath,\n  IndexingMode,\n  IndexingPolicy,\n  SpatialIndex,\n  SpatialType,\n  GeospatialType,\n  IndexKind,\n  Location,\n  PartitionKey,\n  PrimitivePartitionKeyValue,\n  NullPartitionKeyType,\n  NonePartitionKeyType,\n  PartitionKeyKind,\n  PartitionKeyDefinition,\n  PartitionKeyDefinitionVersion,\n  PartitionKeyBuilder,\n  PermissionMode,\n  PriorityLevel,\n  TriggerOperation,\n  TriggerType,\n  UserDefinedFunctionType,\n  CompositePath,\n  ComputedProperty,\n  VectorEmbeddingPolicy,\n  VectorIndex,\n  VectorEmbedding,\n  VectorEmbeddingDataType,\n  VectorEmbeddingDistanceFunction,\n  VectorIndexType,\n  FullTextIndex,\n  FullTextPolicy,\n  FullTextPath,\n} from \"./documents/index.js\";\n\nexport { UniqueKeyPolicy, UniqueKey } from \"./client/Container/UniqueKeyPolicy.js\";\nexport { ContainerRequest } from \"./client/Container/ContainerRequest.js\";\nexport { Constants, OperationType, ResourceType, HTTPMethod } from \"./common/index.js\";\nexport { RetryOptions } from \"./retry/index.js\";\nexport * from \"./request/index.js\";\n\nexport {\n  DiagnosticNodeInternal,\n  DiagnosticDataValue,\n  DiagnosticNodeType,\n} from \"./diagnostics/DiagnosticNodeInternal.js\";\n\nexport {\n  CosmosHeaders,\n  SqlParameter,\n  SqlQuerySpec,\n  JSONValue,\n  JSONArray,\n  JSONObject,\n} from \"./queryExecutionContext/index.js\";\nexport { QueryIterator } from \"./queryIterator.js\";\nexport * from \"./queryMetrics/index.js\";\nexport { CosmosClient } from \"./CosmosClient.js\";\nexport { CosmosClientOptions, Agent } from \"./CosmosClientOptions.js\";\nexport * from \"./client/index.js\";\nexport { Scripts } from \"./client/Script/Scripts.js\";\nexport { Next, Plugin, PluginConfig, PluginOn } from \"./plugins/Plugin.js\";\nexport { TokenProvider, RequestInfo } from \"./auth.js\";\n\nexport { ChangeFeedIterator } from \"./ChangeFeedIterator.js\";\nexport { ChangeFeedOptions } from \"./ChangeFeedOptions.js\";\nexport { ChangeFeedResponse } from \"./ChangeFeedResponse.js\";\nexport { ClientContext } from \"./ClientContext.js\";\n\nexport {\n  CosmosDiagnostics,\n  MetadataLookUpDiagnostic,\n  MetadataLookUpDiagnostics,\n  MetadataLookUpType,\n  RetryDiagnostics,\n  FailedRequestAttemptDiagnostic,\n  GatewayStatistics,\n  ClientSideRequestStatistics,\n  ClientConfigDiagnostic,\n  DiagnosticNode,\n  EncryptionDiagnostics,\n} from \"./CosmosDiagnostics.js\";\n\nexport {\n  ChangeFeedPullModelIterator,\n  ChangeFeedIteratorOptions,\n  ChangeFeedIteratorResponse,\n  ChangeFeedStartFrom,\n  FeedRange,\n  ChangeFeedMode,\n  ChangeFeedPolicy,\n  ChangeFeedRetentionTimeSpan,\n} from \"./client/ChangeFeed/index.js\";\nexport { CosmosDbDiagnosticLevel } from \"./diagnostics/CosmosDbDiagnosticLevel.js\";\n\nexport { GlobalEndpointManager } from \"./globalEndpointManager.js\";\nexport { SasTokenPermissionKind } from \"./common/constants.js\";\nexport { createAuthorizationSasToken } from \"./utils/SasToken.js\";\nexport { RestError } from \"@azure/core-rest-pipeline\";\nexport { AbortError } from \"@azure/abort-controller\";\nexport * from \"./encryption/enums/index.js\";\nexport * from \"./encryption/ClientEncryptionKey/index.js\";\nexport * from \"./encryption/EncryptionKeyResolver/index.js\";\nexport {\n  ClientEncryptionIncludedPath,\n  ClientEncryptionPolicy,\n  ClientEncryptionKeyProperties,\n  EncryptionKeyWrapMetadata,\n  EncryptionQueryBuilder,\n  ClientEncryptionKeyRequest,\n  ClientEncryptionKeyResponse,\n  EncryptionKeyResolver,\n  AzureKeyVaultEncryptionKeyResolver,\n  EncryptionType,\n  EncryptionAlgorithm,\n  EncryptionKeyResolverName,\n  ClientEncryptionOptions,\n  CosmosEncryptedNumber,\n  CosmosEncryptedNumberType,\n} from \"./encryption/index.js\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,0BAA0B,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EACL,WAAW,GAGZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,yCAAyC,EAAE,MAAM,WAAW,CAAC;AACtE,OAAO,EAYL,iBAAiB,GAWlB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAEL,kBAAkB,GAInB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,cAAc,EACd,gBAAgB,EAEhB,eAAe,EACf,QAAQ,EAGR,YAAY,EAGZ,WAAW,EACX,cAAc,EACd,SAAS,EAMT,gBAAgB,EAEhB,6BAA6B,EAC7B,mBAAmB,EACnB,cAAc,EACd,aAAa,EACb,gBAAgB,EAChB,WAAW,EACX,uBAAuB,EAMvB,uBAAuB,EACvB,+BAA+B,EAC/B,eAAe,GAIhB,MAAM,sBAAsB,CAAC;AAI9B,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEvF,cAAc,oBAAoB,CAAC;AAEnC,OAAO,EACL,sBAAsB,EAEtB,kBAAkB,GACnB,MAAM,yCAAyC,CAAC;AAUjD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,cAAc,yBAAyB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,cAAc,mBAAmB,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EAA6C,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAG1F,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,EACL,iBAAiB,EAGjB,kBAAkB,GAQnB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAGL,0BAA0B,EAC1B,mBAAmB,EACnB,SAAS,EACT,cAAc,EACd,gBAAgB,EAChB,2BAA2B,GAC5B,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,uBAAuB,EAAE,MAAM,0CAA0C,CAAC;AAEnF,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACrD,cAAc,6BAA6B,CAAC;AAC5C,cAAc,2CAA2C,CAAC;AAC1D,cAAc,6CAA6C,CAAC;AAC5D,OAAO,EAKL,sBAAsB,EAEtB,2BAA2B,EAE3B,kCAAkC,EAClC,cAAc,EACd,mBAAmB,EACnB,yBAAyB,EAGzB,yBAAyB,GAC1B,MAAM,uBAAuB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nexport { DEFAULT_PARTITION_KEY_PATH } from \"./common/partitionKeys.js\";\nexport {\n  StatusCodes,\n  type StatusCodesType,\n  type PartitionKeyRangePropertiesNames,\n} from \"./common/index.js\";\nexport { setAuthorizationTokenHeaderUsingMasterKey } from \"./auth.js\";\nexport {\n  type Operation,\n  type OperationResponse,\n  type BulkOptions,\n  type CreateOperation,\n  type UpsertOperation,\n  type ReplaceOperation,\n  type DeleteOperation,\n  type ReadOperation,\n  type OperationBase,\n  type OperationWithItem,\n  type OperationInput,\n  BulkOperationType,\n  type ExtendedOperationResponse,\n  type BulkOperationResponse,\n  type CreateOperationInput,\n  type UpsertOperationInput,\n  type ReplaceOperationInput,\n  type ReadOperationInput,\n  type DeleteOperationInput,\n  type PatchOperationInput,\n  type BulkPatchOperation,\n  type BulkOperationResult,\n} from \"./utils/batch.js\";\nexport {\n  type PatchOperation,\n  PatchOperationType,\n  type ExistingKeyOperation,\n  type RemoveOperation,\n  type PatchRequestBody,\n} from \"./utils/patch.js\";\nexport {\n  ConnectionMode,\n  ConsistencyLevel,\n  type ConnectionPolicy,\n  DatabaseAccount,\n  DataType,\n  type Index,\n  type IndexedPath,\n  IndexingMode,\n  type IndexingPolicy,\n  type SpatialIndex,\n  SpatialType,\n  GeospatialType,\n  IndexKind,\n  type Location,\n  type PartitionKey,\n  type PrimitivePartitionKeyValue,\n  type NullPartitionKeyType,\n  type NonePartitionKeyType,\n  PartitionKeyKind,\n  type PartitionKeyDefinition,\n  PartitionKeyDefinitionVersion,\n  PartitionKeyBuilder,\n  PermissionMode,\n  PriorityLevel,\n  TriggerOperation,\n  TriggerType,\n  UserDefinedFunctionType,\n  type CompositePath,\n  type ComputedProperty,\n  type VectorEmbeddingPolicy,\n  type VectorIndex,\n  type VectorEmbedding,\n  VectorEmbeddingDataType,\n  VectorEmbeddingDistanceFunction,\n  VectorIndexType,\n  type FullTextIndex,\n  type FullTextPolicy,\n  type FullTextPath,\n} from \"./documents/index.js\";\n\nexport type { UniqueKeyPolicy, UniqueKey } from \"./client/Container/UniqueKeyPolicy.js\";\nexport type { ContainerRequest } from \"./client/Container/ContainerRequest.js\";\nexport { Constants, OperationType, ResourceType, HTTPMethod } from \"./common/index.js\";\nexport type { RetryOptions } from \"./retry/index.js\";\nexport * from \"./request/index.js\";\n\nexport {\n  DiagnosticNodeInternal,\n  type DiagnosticDataValue,\n  DiagnosticNodeType,\n} from \"./diagnostics/DiagnosticNodeInternal.js\";\n\nexport type {\n  CosmosHeaders,\n  SqlParameter,\n  SqlQuerySpec,\n  JSONValue,\n  JSONArray,\n  JSONObject,\n} from \"./queryExecutionContext/index.js\";\nexport { QueryIterator } from \"./queryIterator.js\";\nexport * from \"./queryMetrics/index.js\";\nexport { CosmosClient } from \"./CosmosClient.js\";\nexport type { CosmosClientOptions, Agent } from \"./CosmosClientOptions.js\";\nexport * from \"./client/index.js\";\nexport { Scripts } from \"./client/Script/Scripts.js\";\nexport { type Next, type Plugin, type PluginConfig, PluginOn } from \"./plugins/Plugin.js\";\nexport type { TokenProvider, RequestInfo } from \"./auth.js\";\n\nexport { ChangeFeedIterator } from \"./ChangeFeedIterator.js\";\nexport type { ChangeFeedOptions } from \"./ChangeFeedOptions.js\";\nexport { ChangeFeedResponse } from \"./ChangeFeedResponse.js\";\nexport { ClientContext } from \"./ClientContext.js\";\n\nexport {\n  CosmosDiagnostics,\n  type MetadataLookUpDiagnostic,\n  type MetadataLookUpDiagnostics,\n  MetadataLookUpType,\n  type RetryDiagnostics,\n  type FailedRequestAttemptDiagnostic,\n  type GatewayStatistics,\n  type ClientSideRequestStatistics,\n  type ClientConfigDiagnostic,\n  type DiagnosticNode,\n  type EncryptionDiagnostics,\n} from \"./CosmosDiagnostics.js\";\n\nexport {\n  type ChangeFeedPullModelIterator,\n  type ChangeFeedIteratorOptions,\n  ChangeFeedIteratorResponse,\n  ChangeFeedStartFrom,\n  FeedRange,\n  ChangeFeedMode,\n  ChangeFeedPolicy,\n  ChangeFeedRetentionTimeSpan,\n} from \"./client/ChangeFeed/index.js\";\nexport { CosmosDbDiagnosticLevel } from \"./diagnostics/CosmosDbDiagnosticLevel.js\";\n\nexport { GlobalEndpointManager } from \"./globalEndpointManager.js\";\nexport { SasTokenPermissionKind } from \"./common/constants.js\";\nexport { createAuthorizationSasToken } from \"./utils/SasToken.js\";\nexport { RestError } from \"@azure/core-rest-pipeline\";\nexport { AbortError } from \"@azure/abort-controller\";\nexport * from \"./encryption/enums/index.js\";\nexport * from \"./encryption/ClientEncryptionKey/index.js\";\nexport * from \"./encryption/EncryptionKeyResolver/index.js\";\nexport {\n  type ClientEncryptionIncludedPath,\n  type ClientEncryptionPolicy,\n  type ClientEncryptionKeyProperties,\n  type EncryptionKeyWrapMetadata,\n  EncryptionQueryBuilder,\n  type ClientEncryptionKeyRequest,\n  ClientEncryptionKeyResponse,\n  type EncryptionKeyResolver,\n  AzureKeyVaultEncryptionKeyResolver,\n  EncryptionType,\n  EncryptionAlgorithm,\n  EncryptionKeyResolverName,\n  type ClientEncryptionOptions,\n  type CosmosEncryptedNumber,\n  CosmosEncryptedNumberType,\n} from \"./encryption/index.js\";\n"]}

package/dist/esm/queryExecutionContext/Aggregators/index.d.ts

@@ -7,5 +7,5 @@ import { StaticValueAggregator } from "./StaticValueAggregator.js";
 import type { AggregateType } from "../../request/ErrorResponse.js";
 export declare function createAggregator(aggregateType: AggregateType): AverageAggregator | CountAggregator | MaxAggregator | MinAggregator | SumAggregator | StaticValueAggregator;
 export { AverageAggregator, CountAggregator, MaxAggregator, MinAggregator, SumAggregator };
-export { Aggregator } from "./Aggregator.js";
+export type { Aggregator } from "./Aggregator.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/esm/queryExecutionContext/Aggregators/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/Aggregators/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAIpE,wBAAgB,gBAAgB,CAC9B,aAAa,EAAE,aAAa,GAE1B,iBAAiB,GACjB,eAAe,GACf,aAAa,GACb,aAAa,GACb,aAAa,GACb,qBAAqB,CAoBxB;AAED,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC;AAC3F,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/queryExecutionContext/Aggregators/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAIpE,wBAAgB,gBAAgB,CAC9B,aAAa,EAAE,aAAa,GAE1B,iBAAiB,GACjB,eAAe,GACf,aAAa,GACb,aAAa,GACb,aAAa,GACb,qBAAqB,CAoBxB;AAED,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC;AAC3F,YAAY,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/esm/queryExecutionContext/Aggregators/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/Aggregators/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE3D,MAAM,UAAU,gBAAgB,CAC9B,aAA4B;IAQ5B,QAAQ,aAAa,EAAE,CAAC;QACtB,KAAK,SAAS;YACZ,OAAO,IAAI,iBAAiB,EAAE,CAAC;QACjC,KAAK,OAAO,CAAC;QACb,KAAK,SAAS;YACZ,OAAO,IAAI,eAAe,EAAE,CAAC;QAC/B,KAAK,KAAK;YACR,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,KAAK,KAAK;YACR,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,KAAK,KAAK;YACR,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,KAAK,UAAU;YACb,OAAO,IAAI,kBAAkB,EAAE,CAAC;QAClC,KAAK,SAAS;YACZ,OAAO,IAAI,iBAAiB,EAAE,CAAC;QACjC;YACE,OAAO,IAAI,qBAAqB,EAAE,CAAC;IACvC,CAAC;AACH,CAAC;AAED,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { AverageAggregator } from \"./AverageAggregator.js\";\nimport { CountAggregator } from \"./CountAggregator.js\";\nimport { MaxAggregator } from \"./MaxAggregator.js\";\nimport { MinAggregator } from \"./MinAggregator.js\";\nimport { SumAggregator } from \"./SumAggregator.js\";\nimport { StaticValueAggregator } from \"./StaticValueAggregator.js\";\nimport type { AggregateType } from \"../../request/ErrorResponse.js\";\nimport { MakeListAggregator } from \"./MakeListAggregator.js\";\nimport { MakeSetAggregator } from \"./MakeSetAggregator.js\";\n\nexport function createAggregator(\n  aggregateType: AggregateType,\n):\n  | AverageAggregator\n  | CountAggregator\n  | MaxAggregator\n  | MinAggregator\n  | SumAggregator\n  | StaticValueAggregator {\n  switch (aggregateType) {\n    case \"Average\":\n      return new AverageAggregator();\n    case \"Count\":\n    case \"CountIf\":\n      return new CountAggregator();\n    case \"Max\":\n      return new MaxAggregator();\n    case \"Min\":\n      return new MinAggregator();\n    case \"Sum\":\n      return new SumAggregator();\n    case \"MakeList\":\n      return new MakeListAggregator();\n    case \"MakeSet\":\n      return new MakeSetAggregator();\n    default:\n      return new StaticValueAggregator();\n  }\n}\n\nexport { AverageAggregator, CountAggregator, MaxAggregator, MinAggregator, SumAggregator };\nexport { Aggregator } from \"./Aggregator.js\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/Aggregators/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE3D,MAAM,UAAU,gBAAgB,CAC9B,aAA4B;IAQ5B,QAAQ,aAAa,EAAE,CAAC;QACtB,KAAK,SAAS;YACZ,OAAO,IAAI,iBAAiB,EAAE,CAAC;QACjC,KAAK,OAAO,CAAC;QACb,KAAK,SAAS;YACZ,OAAO,IAAI,eAAe,EAAE,CAAC;QAC/B,KAAK,KAAK;YACR,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,KAAK,KAAK;YACR,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,KAAK,KAAK;YACR,OAAO,IAAI,aAAa,EAAE,CAAC;QAC7B,KAAK,UAAU;YACb,OAAO,IAAI,kBAAkB,EAAE,CAAC;QAClC,KAAK,SAAS;YACZ,OAAO,IAAI,iBAAiB,EAAE,CAAC;QACjC;YACE,OAAO,IAAI,qBAAqB,EAAE,CAAC;IACvC,CAAC;AACH,CAAC;AAED,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,aAAa,EAAE,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { AverageAggregator } from \"./AverageAggregator.js\";\nimport { CountAggregator } from \"./CountAggregator.js\";\nimport { MaxAggregator } from \"./MaxAggregator.js\";\nimport { MinAggregator } from \"./MinAggregator.js\";\nimport { SumAggregator } from \"./SumAggregator.js\";\nimport { StaticValueAggregator } from \"./StaticValueAggregator.js\";\nimport type { AggregateType } from \"../../request/ErrorResponse.js\";\nimport { MakeListAggregator } from \"./MakeListAggregator.js\";\nimport { MakeSetAggregator } from \"./MakeSetAggregator.js\";\n\nexport function createAggregator(\n  aggregateType: AggregateType,\n):\n  | AverageAggregator\n  | CountAggregator\n  | MaxAggregator\n  | MinAggregator\n  | SumAggregator\n  | StaticValueAggregator {\n  switch (aggregateType) {\n    case \"Average\":\n      return new AverageAggregator();\n    case \"Count\":\n    case \"CountIf\":\n      return new CountAggregator();\n    case \"Max\":\n      return new MaxAggregator();\n    case \"Min\":\n      return new MinAggregator();\n    case \"Sum\":\n      return new SumAggregator();\n    case \"MakeList\":\n      return new MakeListAggregator();\n    case \"MakeSet\":\n      return new MakeSetAggregator();\n    default:\n      return new StaticValueAggregator();\n  }\n}\n\nexport { AverageAggregator, CountAggregator, MaxAggregator, MinAggregator, SumAggregator };\nexport type { Aggregator } from \"./Aggregator.js\";\n"]}

package/dist/esm/queryExecutionContext/ContinuationTokenManager/BaseContinuationTokenManager.js

@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 import { parseBaseContinuationToken, } from "../../documents/ContinuationToken/CompositeQueryContinuationToken.js";
-import { PartitionRangeManager } from "../PartitionRangeManager.js";
+import { PartitionRangeManager, isPartitionExhausted } from "../PartitionRangeManager.js";
 /**
  * Base abstract class for continuation token management.
  * Provides common functionality shared between parallel and ORDER BY query token managers.
@@ -10,12 +10,18 @@ import { PartitionRangeManager } from "../PartitionRangeManager.js";
 export class BaseContinuationTokenManager {
     ranges = [];
     partitionRangeManager = new PartitionRangeManager();
-    constructor(initialContinuationToken) {
+    collectionLink;
+    offset;
+    limit;
+    constructor(collectionLink, initialContinuationToken) {
+        this.collectionLink = collectionLink;
         if (initialContinuationToken) {
             const token = parseBaseContinuationToken(initialContinuationToken);
             if (token?.rangeMappings) {
-                this.ranges = token.rangeMappings;
+                this.ranges.push(...token.rangeMappings);
             }
+            this.offset = token?.offset;
+            this.limit = token?.limit;
         }
     }
     /**
@@ -42,7 +48,6 @@ export class BaseContinuationTokenManager {
      * @returns Object containing the end index for slicing results and optional continuation token for next page
      */
     paginateResults(pageSize, isResponseEmpty, responseResult) {
-        // Process response data
         if (responseResult) {
             this.processResponseResult(responseResult);
         }
@@ -97,23 +102,26 @@ export class BaseContinuationTokenManager {
         }
         this.processQuerySpecificResponse(responseResult);
     }
-    isPartitionExhausted(continuationToken) {
-        return (!continuationToken ||
-            continuationToken === "" ||
-            continuationToken === "null" ||
-            continuationToken.toLowerCase() === "null");
+    /**
+     * Compacts the ranges array in place by keeping only items that satisfy the predicate.
+     * This preserves the same array reference while removing unwanted entries.
+     */
+    compactRangesInPlace(shouldKeep) {
+        let writeIndex = 0;
+        for (let i = 0; i < this.ranges.length; i++) {
+            const mapping = this.ranges[i];
+            if (!shouldKeep(mapping)) {
+                continue;
+            }
+            this.ranges[writeIndex++] = mapping;
+        }
+        this.ranges.length = writeIndex;
     }
     removeExhaustedRangesFromRanges() {
         if (!this.ranges || !Array.isArray(this.ranges)) {
             return;
         }
-        this.ranges = this.ranges.filter((mapping) => {
-            if (!mapping) {
-                return false;
-            }
-            const isExhausted = this.isPartitionExhausted(mapping.continuationToken);
-            return !isExhausted;
-        });
+        this.compactRangesInPlace((mapping) => !!mapping && !isPartitionExhausted(mapping.continuationToken));
     }
     handlePartitionRangeChanges(updatedContinuationRanges) {
         if (updatedContinuationRanges && Object.keys(updatedContinuationRanges).length === 0) {
@@ -150,8 +158,8 @@ export class BaseContinuationTokenManager {
         existingMapping.continuationToken = continuationToken;
     }
     handleRangeSplit(oldRange, newRanges, continuationToken) {
-        // Remove the old range mapping from the common ranges array
-        this.ranges = this.ranges.filter((mapping) => !(mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max));
+        // Remove the old range mapping from the common ranges array (in-place, no reinitialization)
+        this.compactRangesInPlace((mapping) => !(mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max));
         // Add new range mappings for each split range
         newRanges.forEach((newRange) => {
             this.createNewRangeMapping(newRange, continuationToken);

package/dist/esm/queryExecutionContext/ContinuationTokenManager/BaseContinuationTokenManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"BaseContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/BaseContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EACL,0BAA0B,GAG3B,MAAM,sEAAsE,CAAC;AAK9E,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AAGpE;;;;GAIG;AACH,MAAM,OAAgB,4BAA4B;IACxC,MAAM,GAAsC,EAAE,CAAC;IACtC,qBAAqB,GAA0B,IAAI,qBAAqB,EAAE,CAAC;IAE5F,YAAY,wBAAiC;QAC3C,IAAI,wBAAwB,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,0BAA0B,CAAC,wBAAwB,CAAC,CAAC;YACnE,IAAI,KAAK,EAAE,aAAa,EAAE,CAAC;gBACzB,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,IAAc,gBAAgB;QAC5B,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,IAAc,SAAS;QACrB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAQD;;;;;;;;OAQG;IACI,eAAe,CACpB,QAAgB,EAChB,eAAwB,EACxB,cAAoC;QAKpC,wBAAwB;QACxB,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,+BAA+B,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;QAE3D,4BAA4B;QAC5B,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,eAAe,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAEhE,OAAO;YACL,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,iBAAiB,EAAE,WAAW;SAC/B,CAAC;IACJ,CAAC;IAqBD;;;OAGG;IACK,+BAA+B;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACjD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,iBAAiB,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAC1D,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;;OAGG;IACK,iBAAiB,CAAC,eAAyB,EAAE,QAAgB;QACnE,eAAe,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAClC,IAAI,CAAC,qBAAqB,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,8CAA8C;QAC9C,IAAI,CAAC,4BAA4B,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;IAC/D,CAAC;IAEO,uBAAuB,CAAC,oBAAoD;QAClF,IAAI,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,oBAAoB,CAAC,CAAC;IAC3E,CAAC;IAED;;;OAGG;IACK,qBAAqB,CAAC,cAAmC;QAC/D,iCAAiC;QACjC,IAAI,cAAc,CAAC,oBAAoB,EAAE,CAAC;YACxC,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;QACpE,CAAC;QAED,iCAAiC;QACjC,IAAI,cAAc,CAAC,yBAAyB,EAAE,CAAC;YAC7C,IAAI,CAAC,2BAA2B,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;QAC7E,CAAC;QAED,IAAI,CAAC,4BAA4B,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC;IAEO,oBAAoB,CAAC,iBAAqC;QAChE,OAAO,CACL,CAAC,iBAAiB;YAClB,iBAAiB,KAAK,EAAE;YACxB,iBAAiB,KAAK,MAAM;YAC5B,iBAAiB,CAAC,WAAW,EAAE,KAAK,MAAM,CAC3C,CAAC;IACJ,CAAC;IAEO,+BAA+B;QACrC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;YAC3C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YACzE,OAAO,CAAC,WAAW,CAAC;QACtB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,2BAA2B,CAAC,yBAAgD;QAClF,IAAI,yBAAyB,IAAI,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrF,OAAO;QACT,CAAC;QACD,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,EAAE;YAC5E,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,SAAiB,EAAE,WAAiC;QAC7E,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,WAAW,CAAC;QAC/D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,EAAE,iBAAiB,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,QAAa,EAAE,QAAa,EAAE,iBAAyB;QAC9E,mEAAmE;QACnE,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAChD,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,CACrF,CAAC;QAEF,IAAI,oBAAoB,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,oDAAoD;QACpD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAE1D,2DAA2D;QAC3D,MAAM,iBAAiB,GAAkB;YACvC,GAAG,EAAE,QAAQ,CAAC,GAAG;YACjB,GAAG,EAAE,QAAQ,CAAC,GAAG;SAClB,CAAC;QAEF,qBAAqB;QACrB,eAAe,CAAC,UAAU,GAAG,iBAAiB,CAAC;QAC/C,eAAe,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IACxD,CAAC;IAEO,gBAAgB,CAAC,QAAa,EAAE,SAAgB,EAAE,iBAAyB;QACjF,4DAA4D;QAC5D,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAC9B,CAAC,OAAO,EAAE,EAAE,CACV,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,CAAC,CACxF,CAAC;QAEF,8CAA8C;QAC9C,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7B,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,iBAAsB,EAAE,iBAAyB;QAC7E,mCAAmC;QACnC,MAAM,UAAU,GAAkB;YAChC,GAAG,EAAE,iBAAiB,CAAC,GAAG;YAC1B,GAAG,EAAE,iBAAiB,CAAC,GAAG;SAC3B,CAAC;QAEF,6CAA6C;QAC7C,MAAM,iBAAiB,GAAoC;YACzD,UAAU,EAAE,UAAU;YACtB,iBAAiB,EAAE,iBAAiB;SACrC,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACtC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { QueryRangeMapping } from \"../queryRangeMapping.js\";\nimport {\n  parseBaseContinuationToken,\n  type QueryRangeWithContinuationToken,\n  type RangeBoundary,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport type {\n  PartitionRangeUpdate,\n  PartitionRangeUpdates,\n} from \"../../documents/ContinuationToken/PartitionRangeUpdate.js\";\nimport { PartitionRangeManager } from \"../PartitionRangeManager.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\n\n/**\n * Base abstract class for continuation token management.\n * Provides common functionality shared between parallel and ORDER BY query token managers.\n * @internal\n */\nexport abstract class BaseContinuationTokenManager {\n  private ranges: QueryRangeWithContinuationToken[] = [];\n  private readonly partitionRangeManager: PartitionRangeManager = new PartitionRangeManager();\n\n  constructor(initialContinuationToken?: string) {\n    if (initialContinuationToken) {\n      const token = parseBaseContinuationToken(initialContinuationToken);\n      if (token?.rangeMappings) {\n        this.ranges = token.rangeMappings;\n      }\n    }\n  }\n\n  /**\n   * Provides controlled access to partition range manager for subclasses.\n   * This is the only protected access point needed.\n   */\n  protected get partitionManager(): PartitionRangeManager {\n    return this.partitionRangeManager;\n  }\n\n  /**\n   * Provides controlled access to ranges for subclasses.\n   * Made protected to allow subclass range management.\n   */\n  protected get rangeList(): QueryRangeWithContinuationToken[] {\n    return this.ranges;\n  }\n\n  /**\n   * Gets the current continuation token for serialization.\n   * Returns undefined if no token exists or query is exhausted.\n   */\n  protected abstract getCurrentContinuationToken(): any;\n\n  /**\n   * Processes query results and generates continuation tokens for pagination.\n   * Handles response data processing, range management, and token generation.\n   *\n   * @param pageSize - Maximum number of items to return in this page\n   * @param isResponseEmpty - Whether the current response contains no data\n   * @param responseResult - Optional response data containing partition mappings and query-specific data\n   * @returns Object containing the end index for slicing results and optional continuation token for next page\n   */\n  public paginateResults(\n    pageSize: number,\n    isResponseEmpty: boolean,\n    responseResult?: ParallelQueryResult,\n  ): {\n    endIndex: number;\n    continuationToken?: string;\n  } {\n    // Process response data\n    if (responseResult) {\n      this.processResponseResult(responseResult);\n    }\n\n    this.removeExhaustedRangesFromRanges();\n    const result = this.processRangesForPagination(pageSize, isResponseEmpty);\n    const tokenString = this.generateContinuationTokenString();\n\n    // Clean up processed ranges\n    this.trimProcessedData(result.processedRanges, result.endIndex);\n\n    return {\n      endIndex: result.endIndex,\n      continuationToken: tokenString,\n    };\n  }\n\n  protected abstract processRangesForPagination(\n    pageSize: number,\n    isResponseEmpty: boolean,\n  ): {\n    endIndex: number;\n    processedRanges: string[];\n  };\n\n  protected abstract processQuerySpecificResponse(responseResult: ParallelQueryResult): void;\n  protected abstract performQuerySpecificDataTrim(\n    processedRanges: string[],\n    endIndex: number,\n  ): void;\n\n  /**\n   * Gets the serialization function for the specific continuation token type.\n   */\n  protected abstract getSerializationFunction(): (token: any) => string;\n\n  /**\n   * Generates continuation token string using the appropriate serialization function.\n   * This provides a common implementation that delegates to query-specific logic.\n   */\n  private generateContinuationTokenString(): string | undefined {\n    const token = this.getCurrentContinuationToken();\n    if (!token) {\n      return undefined;\n    }\n\n    const serializeFunction = this.getSerializationFunction();\n    return serializeFunction(token);\n  }\n\n  /**\n   * Cleans up processed data after a page has been returned.\n   * Handles both common and query-specific cleanup.\n   */\n  private trimProcessedData(processedRanges: string[], endIndex: number): void {\n    processedRanges.forEach((rangeId) => {\n      this.partitionRangeManager.removePartitionRangeMapping(rangeId);\n    });\n\n    // Delegate query-specific cleanup to subclass\n    this.performQuerySpecificDataTrim(processedRanges, endIndex);\n  }\n\n  private addPartitionKeyRangeMap(partitionKeyRangeMap: Map<string, QueryRangeMapping>): void {\n    this.partitionRangeManager.addPartitionKeyRangeMap(partitionKeyRangeMap);\n  }\n\n  /**\n   * Processes the entire response result and updates the continuation token manager state.\n   * This encapsulates all response handling logic in one place.\n   */\n  private processResponseResult(responseResult: ParallelQueryResult): void {\n    // Handle partition key range map\n    if (responseResult.partitionKeyRangeMap) {\n      this.addPartitionKeyRangeMap(responseResult.partitionKeyRangeMap);\n    }\n\n    // Handle partition range updates\n    if (responseResult.updatedContinuationRanges) {\n      this.handlePartitionRangeChanges(responseResult.updatedContinuationRanges);\n    }\n\n    this.processQuerySpecificResponse(responseResult);\n  }\n\n  private isPartitionExhausted(continuationToken: string | undefined): boolean {\n    return (\n      !continuationToken ||\n      continuationToken === \"\" ||\n      continuationToken === \"null\" ||\n      continuationToken.toLowerCase() === \"null\"\n    );\n  }\n\n  private removeExhaustedRangesFromRanges(): void {\n    if (!this.ranges || !Array.isArray(this.ranges)) {\n      return;\n    }\n    this.ranges = this.ranges.filter((mapping) => {\n      if (!mapping) {\n        return false;\n      }\n      const isExhausted = this.isPartitionExhausted(mapping.continuationToken);\n      return !isExhausted;\n    });\n  }\n\n  private handlePartitionRangeChanges(updatedContinuationRanges: PartitionRangeUpdates): void {\n    if (updatedContinuationRanges && Object.keys(updatedContinuationRanges).length === 0) {\n      return;\n    }\n    Object.entries(updatedContinuationRanges).forEach(([rangeKey, rangeChange]) => {\n      this.processRangeChange(rangeKey, rangeChange);\n    });\n  }\n\n  private processRangeChange(_rangeKey: string, rangeChange: PartitionRangeUpdate): void {\n    const { oldRange, newRanges, continuationToken } = rangeChange;\n    if (newRanges.length === 1) {\n      this.handleRangeMerge(oldRange, newRanges[0], continuationToken);\n    } else {\n      this.handleRangeSplit(oldRange, newRanges, continuationToken);\n    }\n  }\n\n  private handleRangeMerge(oldRange: any, newRange: any, continuationToken: string): void {\n    // Find existing range mapping to update in the common ranges array\n    const existingMappingIndex = this.ranges.findIndex(\n      (mapping) =>\n        mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max,\n    );\n\n    if (existingMappingIndex < 0) {\n      return;\n    }\n\n    // Update existing mapping with new range properties\n    const existingMapping = this.ranges[existingMappingIndex];\n\n    // Create new simplified QueryRange with updated boundaries\n    const updatedQueryRange: RangeBoundary = {\n      min: newRange.min,\n      max: newRange.max,\n    };\n\n    // Update the mapping\n    existingMapping.queryRange = updatedQueryRange;\n    existingMapping.continuationToken = continuationToken;\n  }\n\n  private handleRangeSplit(oldRange: any, newRanges: any[], continuationToken: string): void {\n    // Remove the old range mapping from the common ranges array\n    this.ranges = this.ranges.filter(\n      (mapping) =>\n        !(mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max),\n    );\n\n    // Add new range mappings for each split range\n    newRanges.forEach((newRange) => {\n      this.createNewRangeMapping(newRange, continuationToken);\n    });\n  }\n\n  private createNewRangeMapping(partitionKeyRange: any, continuationToken: string): void {\n    // Create new simplified QueryRange\n    const queryRange: RangeBoundary = {\n      min: partitionKeyRange.min,\n      max: partitionKeyRange.max,\n    };\n\n    // Create new QueryRangeWithContinuationToken\n    const newRangeWithToken: QueryRangeWithContinuationToken = {\n      queryRange: queryRange,\n      continuationToken: continuationToken,\n    };\n\n    this.ranges.push(newRangeWithToken);\n  }\n}\n"]}
+{"version":3,"file":"BaseContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/BaseContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EACL,0BAA0B,GAG3B,MAAM,sEAAsE,CAAC;AAK9E,OAAO,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AAG1F;;;;GAIG;AACH,MAAM,OAAgB,4BAA4B;IAC/B,MAAM,GAAsC,EAAE,CAAC;IAC/C,qBAAqB,GAA0B,IAAI,qBAAqB,EAAE,CAAC;IACzE,cAAc,CAAS;IAChC,MAAM,CAAU;IAChB,KAAK,CAAU;IAEzB,YAAY,cAAsB,EAAE,wBAAiC;QACnE,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,wBAAwB,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,0BAA0B,CAAC,wBAAwB,CAAC,CAAC;YACnE,IAAI,KAAK,EAAE,aAAa,EAAE,CAAC;gBACzB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,aAAa,CAAC,CAAC;YAC3C,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,KAAK,EAAE,MAAM,CAAC;YAC5B,IAAI,CAAC,KAAK,GAAG,KAAK,EAAE,KAAK,CAAC;QAC5B,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,IAAc,gBAAgB;QAC5B,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,IAAc,SAAS;QACrB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAQD;;;;;;;;OAQG;IACI,eAAe,CACpB,QAAgB,EAChB,eAAwB,EACxB,cAAoC;QAKpC,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,qBAAqB,CAAC,cAAc,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,+BAA+B,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,+BAA+B,EAAE,CAAC;QAE3D,4BAA4B;QAC5B,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,eAAe,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAEhE,OAAO;YACL,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,iBAAiB,EAAE,WAAW;SAC/B,CAAC;IACJ,CAAC;IAqBD;;;OAGG;IACK,+BAA+B;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACjD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,iBAAiB,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAC1D,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED;;;OAGG;IACK,iBAAiB,CAAC,eAAyB,EAAE,QAAgB;QACnE,eAAe,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAClC,IAAI,CAAC,qBAAqB,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,8CAA8C;QAC9C,IAAI,CAAC,4BAA4B,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;IAC/D,CAAC;IAEO,uBAAuB,CAAC,oBAAoD;QAClF,IAAI,CAAC,qBAAqB,CAAC,uBAAuB,CAAC,oBAAoB,CAAC,CAAC;IAC3E,CAAC;IAED;;;OAGG;IACK,qBAAqB,CAAC,cAAmC;QAC/D,iCAAiC;QACjC,IAAI,cAAc,CAAC,oBAAoB,EAAE,CAAC;YACxC,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,oBAAoB,CAAC,CAAC;QACpE,CAAC;QAED,iCAAiC;QACjC,IAAI,cAAc,CAAC,yBAAyB,EAAE,CAAC;YAC7C,IAAI,CAAC,2BAA2B,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;QAC7E,CAAC;QAED,IAAI,CAAC,4BAA4B,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC;IAED;;;OAGG;IACK,oBAAoB,CAC1B,UAAiE;QAEjE,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBACzB,SAAS;YACX,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,GAAG,OAAO,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,UAAU,CAAC;IAClC,CAAC;IAEO,+BAA+B;QACrC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,oBAAoB,CACvB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAC3E,CAAC;IACJ,CAAC;IAEO,2BAA2B,CAAC,yBAAgD;QAClF,IAAI,yBAAyB,IAAI,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrF,OAAO;QACT,CAAC;QACD,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,EAAE,EAAE;YAC5E,IAAI,CAAC,kBAAkB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,SAAiB,EAAE,WAAiC;QAC7E,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,WAAW,CAAC;QAC/D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;QACnE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,SAAS,EAAE,iBAAiB,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,QAAa,EAAE,QAAa,EAAE,iBAAyB;QAC9E,mEAAmE;QACnE,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAChD,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,CACrF,CAAC;QAEF,IAAI,oBAAoB,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,oDAAoD;QACpD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAE1D,2DAA2D;QAC3D,MAAM,iBAAiB,GAAkB;YACvC,GAAG,EAAE,QAAQ,CAAC,GAAG;YACjB,GAAG,EAAE,QAAQ,CAAC,GAAG;SAClB,CAAC;QAEF,qBAAqB;QACrB,eAAe,CAAC,UAAU,GAAG,iBAAiB,CAAC;QAC/C,eAAe,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IACxD,CAAC;IAEO,gBAAgB,CAAC,QAAa,EAAE,SAAgB,EAAE,iBAAyB;QACjF,4FAA4F;QAC5F,IAAI,CAAC,oBAAoB,CACvB,CAAC,OAAO,EAAE,EAAE,CACV,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,GAAG,CAAC,CACxF,CAAC;QAEF,8CAA8C;QAC9C,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7B,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,iBAAsB,EAAE,iBAAyB;QAC7E,mCAAmC;QACnC,MAAM,UAAU,GAAkB;YAChC,GAAG,EAAE,iBAAiB,CAAC,GAAG;YAC1B,GAAG,EAAE,iBAAiB,CAAC,GAAG;SAC3B,CAAC;QAEF,6CAA6C;QAC7C,MAAM,iBAAiB,GAAoC;YACzD,UAAU,EAAE,UAAU;YACtB,iBAAiB,EAAE,iBAAiB;SACrC,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACtC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { QueryRangeMapping } from \"../queryRangeMapping.js\";\nimport {\n  parseBaseContinuationToken,\n  type QueryRangeWithContinuationToken,\n  type RangeBoundary,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport type {\n  PartitionRangeUpdate,\n  PartitionRangeUpdates,\n} from \"../../documents/ContinuationToken/PartitionRangeUpdate.js\";\nimport { PartitionRangeManager, isPartitionExhausted } from \"../PartitionRangeManager.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\n\n/**\n * Base abstract class for continuation token management.\n * Provides common functionality shared between parallel and ORDER BY query token managers.\n * @internal\n */\nexport abstract class BaseContinuationTokenManager {\n  private readonly ranges: QueryRangeWithContinuationToken[] = [];\n  private readonly partitionRangeManager: PartitionRangeManager = new PartitionRangeManager();\n  protected readonly collectionLink: string;\n  protected offset?: number;\n  protected limit?: number;\n\n  constructor(collectionLink: string, initialContinuationToken?: string) {\n    this.collectionLink = collectionLink;\n    if (initialContinuationToken) {\n      const token = parseBaseContinuationToken(initialContinuationToken);\n      if (token?.rangeMappings) {\n        this.ranges.push(...token.rangeMappings);\n      }\n      this.offset = token?.offset;\n      this.limit = token?.limit;\n    }\n  }\n\n  /**\n   * Provides controlled access to partition range manager for subclasses.\n   * This is the only protected access point needed.\n   */\n  protected get partitionManager(): PartitionRangeManager {\n    return this.partitionRangeManager;\n  }\n\n  /**\n   * Provides controlled access to ranges for subclasses.\n   * Made protected to allow subclass range management.\n   */\n  protected get rangeList(): QueryRangeWithContinuationToken[] {\n    return this.ranges;\n  }\n\n  /**\n   * Gets the current continuation token for serialization.\n   * Returns undefined if no token exists or query is exhausted.\n   */\n  protected abstract getCurrentContinuationToken(): any;\n\n  /**\n   * Processes query results and generates continuation tokens for pagination.\n   * Handles response data processing, range management, and token generation.\n   *\n   * @param pageSize - Maximum number of items to return in this page\n   * @param isResponseEmpty - Whether the current response contains no data\n   * @param responseResult - Optional response data containing partition mappings and query-specific data\n   * @returns Object containing the end index for slicing results and optional continuation token for next page\n   */\n  public paginateResults(\n    pageSize: number,\n    isResponseEmpty: boolean,\n    responseResult?: ParallelQueryResult,\n  ): {\n    endIndex: number;\n    continuationToken?: string;\n  } {\n    if (responseResult) {\n      this.processResponseResult(responseResult);\n    }\n\n    this.removeExhaustedRangesFromRanges();\n    const result = this.processRangesForPagination(pageSize, isResponseEmpty);\n    const tokenString = this.generateContinuationTokenString();\n\n    // Clean up processed ranges\n    this.trimProcessedData(result.processedRanges, result.endIndex);\n\n    return {\n      endIndex: result.endIndex,\n      continuationToken: tokenString,\n    };\n  }\n\n  protected abstract processRangesForPagination(\n    pageSize: number,\n    isResponseEmpty: boolean,\n  ): {\n    endIndex: number;\n    processedRanges: string[];\n  };\n\n  protected abstract processQuerySpecificResponse(responseResult: ParallelQueryResult): void;\n  protected abstract performQuerySpecificDataTrim(\n    processedRanges: string[],\n    endIndex: number,\n  ): void;\n\n  /**\n   * Gets the serialization function for the specific continuation token type.\n   */\n  protected abstract getSerializationFunction(): (token: any) => string;\n\n  /**\n   * Generates continuation token string using the appropriate serialization function.\n   * This provides a common implementation that delegates to query-specific logic.\n   */\n  private generateContinuationTokenString(): string | undefined {\n    const token = this.getCurrentContinuationToken();\n    if (!token) {\n      return undefined;\n    }\n\n    const serializeFunction = this.getSerializationFunction();\n    return serializeFunction(token);\n  }\n\n  /**\n   * Cleans up processed data after a page has been returned.\n   * Handles both common and query-specific cleanup.\n   */\n  private trimProcessedData(processedRanges: string[], endIndex: number): void {\n    processedRanges.forEach((rangeId) => {\n      this.partitionRangeManager.removePartitionRangeMapping(rangeId);\n    });\n\n    // Delegate query-specific cleanup to subclass\n    this.performQuerySpecificDataTrim(processedRanges, endIndex);\n  }\n\n  private addPartitionKeyRangeMap(partitionKeyRangeMap: Map<string, QueryRangeMapping>): void {\n    this.partitionRangeManager.addPartitionKeyRangeMap(partitionKeyRangeMap);\n  }\n\n  /**\n   * Processes the entire response result and updates the continuation token manager state.\n   * This encapsulates all response handling logic in one place.\n   */\n  private processResponseResult(responseResult: ParallelQueryResult): void {\n    // Handle partition key range map\n    if (responseResult.partitionKeyRangeMap) {\n      this.addPartitionKeyRangeMap(responseResult.partitionKeyRangeMap);\n    }\n\n    // Handle partition range updates\n    if (responseResult.updatedContinuationRanges) {\n      this.handlePartitionRangeChanges(responseResult.updatedContinuationRanges);\n    }\n\n    this.processQuerySpecificResponse(responseResult);\n  }\n\n  /**\n   * Compacts the ranges array in place by keeping only items that satisfy the predicate.\n   * This preserves the same array reference while removing unwanted entries.\n   */\n  private compactRangesInPlace(\n    shouldKeep: (mapping: QueryRangeWithContinuationToken) => boolean,\n  ): void {\n    let writeIndex = 0;\n    for (let i = 0; i < this.ranges.length; i++) {\n      const mapping = this.ranges[i];\n      if (!shouldKeep(mapping)) {\n        continue;\n      }\n      this.ranges[writeIndex++] = mapping;\n    }\n    this.ranges.length = writeIndex;\n  }\n\n  private removeExhaustedRangesFromRanges(): void {\n    if (!this.ranges || !Array.isArray(this.ranges)) {\n      return;\n    }\n    this.compactRangesInPlace(\n      (mapping) => !!mapping && !isPartitionExhausted(mapping.continuationToken),\n    );\n  }\n\n  private handlePartitionRangeChanges(updatedContinuationRanges: PartitionRangeUpdates): void {\n    if (updatedContinuationRanges && Object.keys(updatedContinuationRanges).length === 0) {\n      return;\n    }\n    Object.entries(updatedContinuationRanges).forEach(([rangeKey, rangeChange]) => {\n      this.processRangeChange(rangeKey, rangeChange);\n    });\n  }\n\n  private processRangeChange(_rangeKey: string, rangeChange: PartitionRangeUpdate): void {\n    const { oldRange, newRanges, continuationToken } = rangeChange;\n    if (newRanges.length === 1) {\n      this.handleRangeMerge(oldRange, newRanges[0], continuationToken);\n    } else {\n      this.handleRangeSplit(oldRange, newRanges, continuationToken);\n    }\n  }\n\n  private handleRangeMerge(oldRange: any, newRange: any, continuationToken: string): void {\n    // Find existing range mapping to update in the common ranges array\n    const existingMappingIndex = this.ranges.findIndex(\n      (mapping) =>\n        mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max,\n    );\n\n    if (existingMappingIndex < 0) {\n      return;\n    }\n\n    // Update existing mapping with new range properties\n    const existingMapping = this.ranges[existingMappingIndex];\n\n    // Create new simplified QueryRange with updated boundaries\n    const updatedQueryRange: RangeBoundary = {\n      min: newRange.min,\n      max: newRange.max,\n    };\n\n    // Update the mapping\n    existingMapping.queryRange = updatedQueryRange;\n    existingMapping.continuationToken = continuationToken;\n  }\n\n  private handleRangeSplit(oldRange: any, newRanges: any[], continuationToken: string): void {\n    // Remove the old range mapping from the common ranges array (in-place, no reinitialization)\n    this.compactRangesInPlace(\n      (mapping) =>\n        !(mapping.queryRange.min === oldRange.min && mapping.queryRange.max === oldRange.max),\n    );\n\n    // Add new range mappings for each split range\n    newRanges.forEach((newRange) => {\n      this.createNewRangeMapping(newRange, continuationToken);\n    });\n  }\n\n  private createNewRangeMapping(partitionKeyRange: any, continuationToken: string): void {\n    // Create new simplified QueryRange\n    const queryRange: RangeBoundary = {\n      min: partitionKeyRange.min,\n      max: partitionKeyRange.max,\n    };\n\n    // Create new QueryRangeWithContinuationToken\n    const newRangeWithToken: QueryRangeWithContinuationToken = {\n      queryRange: queryRange,\n      continuationToken: continuationToken,\n    };\n\n    this.ranges.push(newRangeWithToken);\n  }\n}\n"]}

package/dist/esm/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.js

@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 import { BaseContinuationTokenManager } from "./BaseContinuationTokenManager.js";
-import { createOrderByQueryContinuationToken, parseOrderByQueryContinuationToken, serializeOrderByQueryContinuationToken, } from "../../documents/ContinuationToken/OrderByQueryContinuationToken.js";
+import { parseOrderByQueryContinuationToken, serializeOrderByQueryContinuationToken, } from "../../documents/ContinuationToken/OrderByQueryContinuationToken.js";
 import { convertRangeMappingToQueryRange } from "../../documents/ContinuationToken/CompositeQueryContinuationToken.js";
 /**
  * Manages continuation tokens for ORDER BY queries using single-range sequential processing.
@@ -9,15 +9,23 @@ import { convertRangeMappingToQueryRange } from "../../documents/ContinuationTok
  * @internal
  */
 export class OrderByQueryContinuationTokenManager extends BaseContinuationTokenManager {
-    continuationToken;
     orderByItemsArray;
-    collectionLink;
+    // OrderBy-specific token state (computed into token on demand)
+    tokenRangeMappings;
+    tokenOrderByItems;
+    tokenSkipCount = 0;
+    tokenDocumentRid = "";
+    tokenHashedLastResult;
     constructor(collectionLink, initialContinuationToken) {
-        super(initialContinuationToken);
-        this.collectionLink = collectionLink;
+        super(collectionLink, initialContinuationToken);
         this.orderByItemsArray = [];
         if (initialContinuationToken) {
-            this.continuationToken = parseOrderByQueryContinuationToken(initialContinuationToken);
+            const parsed = parseOrderByQueryContinuationToken(initialContinuationToken);
+            this.tokenRangeMappings = parsed.rangeMappings;
+            this.tokenOrderByItems = parsed.orderByItems;
+            this.tokenSkipCount = parsed.skipCount;
+            this.tokenDocumentRid = parsed.documentRid;
+            this.tokenHashedLastResult = parsed.hashedLastResult;
         }
     }
     processQuerySpecificResponse(responseResult) {
@@ -41,8 +49,8 @@ export class OrderByQueryContinuationTokenManager extends BaseContinuationTokenM
         }
     }
     processRangesForPagination(pageSize, isResponseEmpty = false) {
-        // Handle empty response case - update the previous valid continuation token
-        if (isResponseEmpty && this.continuationToken) {
+        // Handle empty response case - preserve previous token state with updated range
+        if (isResponseEmpty && this.tokenRangeMappings) {
             let rangeProcessingResult;
             if (this.rangeList.length === 0) {
                 rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);
@@ -52,14 +60,10 @@ export class OrderByQueryContinuationTokenManager extends BaseContinuationTokenM
             }
             const { lastRangeBeforePageLimit } = rangeProcessingResult;
             if (lastRangeBeforePageLimit) {
-                // Use the range matching the continuation token for empty response
-                this.continuationToken.rangeMappings = [
-                    convertRangeMappingToQueryRange(lastRangeBeforePageLimit),
-                ];
+                this.tokenRangeMappings = [convertRangeMappingToQueryRange(lastRangeBeforePageLimit)];
             }
             else {
-                // Range is exhausted - clear the continuation token
-                this.continuationToken = undefined;
+                this.clearTokenState();
             }
             return {
                 endIndex: rangeProcessingResult.endIndex,
@@ -71,7 +75,7 @@ export class OrderByQueryContinuationTokenManager extends BaseContinuationTokenM
         const { lastRangeBeforePageLimit } = rangeProcessingResult;
         // Check if we have a valid range to continue with
         if (!lastRangeBeforePageLimit) {
-            this.continuationToken = undefined;
+            this.clearTokenState();
             return {
                 endIndex: rangeProcessingResult.endIndex,
                 processedRanges: rangeProcessingResult.processedRanges,
@@ -100,30 +104,49 @@ export class OrderByQueryContinuationTokenManager extends BaseContinuationTokenM
         // If we don't have valid ORDER BY items, we cannot create a proper continuation token
         // This can happen when the response doesn't contain ORDER BY metadata or when there are no results
         if (!lastOrderByItems || lastOrderByItems.length === 0) {
-            this.continuationToken = undefined;
+            this.clearTokenState();
             return {
                 endIndex: rangeProcessingResult.endIndex,
                 processedRanges: rangeProcessingResult.processedRanges,
             };
         }
-        const rangeMappings = [queryRange];
-        // Create new ORDER BY continuation token
-        this.continuationToken = createOrderByQueryContinuationToken(rangeMappings, lastOrderByItems, this.collectionLink, // Container RID/link
-        skipCount, // Number of documents with the same RID already processed
-        documentRid, // Document RID from the last item in the page
-        lastRangeBeforePageLimit.offset, // Current offset for OFFSET/LIMIT queries
-        lastRangeBeforePageLimit.limit, // Current limit for OFFSET/LIMIT queries
-        lastRangeBeforePageLimit.hashedLastResult);
+        // Update token state fields
+        this.tokenRangeMappings = [queryRange];
+        this.tokenOrderByItems = lastOrderByItems;
+        this.tokenSkipCount = skipCount;
+        this.tokenDocumentRid = documentRid;
+        this.offset = lastRangeBeforePageLimit.offset;
+        this.limit = lastRangeBeforePageLimit.limit;
+        this.tokenHashedLastResult = lastRangeBeforePageLimit.hashedLastResult;
         return {
             endIndex: rangeProcessingResult.endIndex,
             processedRanges: rangeProcessingResult.processedRanges,
         };
     }
     getCurrentContinuationToken() {
-        return this.continuationToken;
+        if (!this.tokenRangeMappings || !this.tokenOrderByItems) {
+            return undefined;
+        }
+        return {
+            rid: this.collectionLink,
+            rangeMappings: this.tokenRangeMappings,
+            orderByItems: this.tokenOrderByItems,
+            skipCount: this.tokenSkipCount,
+            documentRid: this.tokenDocumentRid,
+            offset: this.offset,
+            limit: this.limit,
+            hashedLastResult: this.tokenHashedLastResult,
+        };
     }
     getSerializationFunction() {
         return serializeOrderByQueryContinuationToken;
     }
+    clearTokenState() {
+        this.tokenRangeMappings = undefined;
+        this.tokenOrderByItems = undefined;
+        this.tokenSkipCount = 0;
+        this.tokenDocumentRid = "";
+        this.tokenHashedLastResult = undefined;
+    }
 }
 //# sourceMappingURL=OrderByQueryContinuationTokenManager.js.map

package/dist/esm/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"OrderByQueryContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAGjF,OAAO,EACL,mCAAmC,EACnC,kCAAkC,EAClC,sCAAsC,GACvC,MAAM,oEAAoE,CAAC;AAC5E,OAAO,EAAE,+BAA+B,EAAE,MAAM,sEAAsE,CAAC;AAEvH;;;;GAIG;AACH,MAAM,OAAO,oCAAqC,SAAQ,4BAA4B;IAC5E,iBAAiB,CAA4C;IACpD,iBAAiB,CAAuB;IACxC,cAAc,CAAS;IAExC,YAAY,cAAsB,EAAE,wBAAiC;QACnE,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;QAC5B,IAAI,wBAAwB,EAAE,CAAC;YAC7B,IAAI,CAAC,iBAAiB,GAAG,kCAAkC,CAAC,wBAAwB,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;IAES,4BAA4B,CAAC,cAAmC;QACxE,gFAAgF;QAChF,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;QAClC,IAAI,cAAc,CAAC,YAAY,EAAE,CAAC;YAChC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAES,4BAA4B,CAAC,gBAA0B,EAAE,QAAgB;QACjF,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAEO,sBAAsB,CAAC,QAAgB;QAC7C,IAAI,QAAQ,KAAK,CAAC,IAAI,QAAQ,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC;YAChE,sCAAsC;YACtC,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,yEAAyE;YACzE,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAES,0BAA0B,CAClC,QAAgB,EAChB,kBAA2B,KAAK;QAEhC,4EAA4E;QAC5E,IAAI,eAAe,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC9C,IAAI,qBAAqB,CAAC;YAE1B,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAChC,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAC/E,CAAC;iBAAM,CAAC;gBACN,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1F,CAAC;YAED,MAAM,EAAE,wBAAwB,EAAE,GAAG,qBAAqB,CAAC;YAC3D,IAAI,wBAAwB,EAAE,CAAC;gBAC7B,mEAAmE;gBACnE,IAAI,CAAC,iBAAiB,CAAC,aAAa,GAAG;oBACrC,+BAA+B,CAAC,wBAAwB,CAAC;iBAC1D,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,oDAAoD;gBACpD,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACrC,CAAC;YACD,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,MAAM,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACnF,MAAM,EAAE,wBAAwB,EAAE,GAAG,qBAAqB,CAAC;QAE3D,kDAAkD;QAClD,IAAI,CAAC,wBAAwB,EAAE,CAAC;YAC9B,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACnC,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,+BAA+B,CAAC,wBAAwB,CAAC,CAAC;QAE7E,wDAAwD;QACxD,IAAI,gBAAmC,CAAC;QACxC,IAAI,WAAmB,CAAC;QACxB,IAAI,SAAS,GAAW,CAAC,CAAC;QAC1B,IAAI,qBAAqB,CAAC,QAAQ,GAAG,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5E,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,QAAQ,GAAG,CAAC,CAAC;YAE/D,IAAI,mBAAmB,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC;gBACxD,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC,YAAY,CAAC;gBAC5E,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC;gBAC/D,+EAA+E;gBAC/E,2EAA2E;gBAC3E,SAAS,GAAG,CAAC,CAAC;gBACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,mBAAmB,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC9C,IAAI,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACnD,SAAS,EAAE,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,sFAAsF;QACtF,mGAAmG;QACnG,IAAI,CAAC,gBAAgB,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;YACnC,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,CAAC,UAAU,CAAC,CAAC;QAEnC,yCAAyC;QACzC,IAAI,CAAC,iBAAiB,GAAG,mCAAmC,CAC1D,aAAa,EACb,gBAAgB,EAChB,IAAI,CAAC,cAAc,EAAE,qBAAqB;QAC1C,SAAS,EAAE,0DAA0D;QACrE,WAAW,EAAE,8CAA8C;QAC3D,wBAAwB,CAAC,MAAM,EAAE,0CAA0C;QAC3E,wBAAwB,CAAC,KAAK,EAAE,yCAAyC;QACzE,wBAAwB,CAAC,gBAAgB,CAC1C,CAAC;QAEF,OAAO;YACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;YACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;SACvD,CAAC;IACJ,CAAC;IAES,2BAA2B;QACnC,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAES,wBAAwB;QAChC,OAAO,sCAAsC,CAAC;IAChD,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { BaseContinuationTokenManager } from \"./BaseContinuationTokenManager.js\";\nimport type { ParallelQueryResult, OrderByItemWithRid } from \"../parallelQueryResult.js\";\nimport type { OrderByQueryContinuationToken } from \"../../documents/ContinuationToken/OrderByQueryContinuationToken.js\";\nimport {\n  createOrderByQueryContinuationToken,\n  parseOrderByQueryContinuationToken,\n  serializeOrderByQueryContinuationToken,\n} from \"../../documents/ContinuationToken/OrderByQueryContinuationToken.js\";\nimport { convertRangeMappingToQueryRange } from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\n\n/**\n * Manages continuation tokens for ORDER BY queries using single-range sequential processing.\n * Uses OrderByQueryContinuationToken for tracking ORDER BY items and skip counts.\n * @internal\n */\nexport class OrderByQueryContinuationTokenManager extends BaseContinuationTokenManager {\n  private continuationToken: OrderByQueryContinuationToken | undefined;\n  private readonly orderByItemsArray: OrderByItemWithRid[];\n  private readonly collectionLink: string;\n\n  constructor(collectionLink: string, initialContinuationToken?: string) {\n    super(initialContinuationToken);\n    this.collectionLink = collectionLink;\n    this.orderByItemsArray = [];\n    if (initialContinuationToken) {\n      this.continuationToken = parseOrderByQueryContinuationToken(initialContinuationToken);\n    }\n  }\n\n  protected processQuerySpecificResponse(responseResult: ParallelQueryResult): void {\n    // Clear existing items and add new ones without reassigning the array reference\n    this.orderByItemsArray.length = 0;\n    if (responseResult.orderByItems) {\n      this.orderByItemsArray.push(...responseResult.orderByItems);\n    }\n  }\n\n  protected performQuerySpecificDataTrim(_processedRanges: string[], endIndex: number): void {\n    this.sliceOrderByItemsArray(endIndex);\n  }\n\n  private sliceOrderByItemsArray(endIndex: number): void {\n    if (endIndex === 0 || endIndex >= this.orderByItemsArray.length) {\n      // Clear the array without reassigning\n      this.orderByItemsArray.length = 0;\n    } else {\n      // Remove items from 0 to endIndex-1, keeping items from endIndex onwards\n      this.orderByItemsArray.splice(0, endIndex);\n    }\n  }\n\n  protected processRangesForPagination(\n    pageSize: number,\n    isResponseEmpty: boolean = false,\n  ): { endIndex: number; processedRanges: string[] } {\n    // Handle empty response case - update the previous valid continuation token\n    if (isResponseEmpty && this.continuationToken) {\n      let rangeProcessingResult;\n\n      if (this.rangeList.length === 0) {\n        rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);\n      } else {\n        rangeProcessingResult = this.partitionManager.processEmptyOrderByRanges(this.rangeList);\n      }\n\n      const { lastRangeBeforePageLimit } = rangeProcessingResult;\n      if (lastRangeBeforePageLimit) {\n        // Use the range matching the continuation token for empty response\n        this.continuationToken.rangeMappings = [\n          convertRangeMappingToQueryRange(lastRangeBeforePageLimit),\n        ];\n      } else {\n        // Range is exhausted - clear the continuation token\n        this.continuationToken = undefined;\n      }\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    // Normal processing path - handle non-empty responses\n    const rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);\n    const { lastRangeBeforePageLimit } = rangeProcessingResult;\n\n    // Check if we have a valid range to continue with\n    if (!lastRangeBeforePageLimit) {\n      this.continuationToken = undefined;\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    const queryRange = convertRangeMappingToQueryRange(lastRangeBeforePageLimit);\n\n    // Extract ORDER BY items from the last item on the page\n    let lastOrderByItems: any[] | undefined;\n    let documentRid: string;\n    let skipCount: number = 0;\n    if (rangeProcessingResult.endIndex > 0 && this.orderByItemsArray.length > 0) {\n      const lastItemIndexOnPage = rangeProcessingResult.endIndex - 1;\n\n      if (lastItemIndexOnPage < this.orderByItemsArray.length) {\n        lastOrderByItems = this.orderByItemsArray[lastItemIndexOnPage].orderByItems;\n        documentRid = this.orderByItemsArray[lastItemIndexOnPage]._rid;\n        // Calculate skip count: count how many documents in the page have the same RID\n        // This handles JOIN queries where multiple documents can have the same RID\n        skipCount = 0;\n        for (let i = 0; i <= lastItemIndexOnPage; i++) {\n          if (this.orderByItemsArray[i]._rid === documentRid) {\n            skipCount++;\n          }\n        }\n      }\n    }\n\n    // If we don't have valid ORDER BY items, we cannot create a proper continuation token\n    // This can happen when the response doesn't contain ORDER BY metadata or when there are no results\n    if (!lastOrderByItems || lastOrderByItems.length === 0) {\n      this.continuationToken = undefined;\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    const rangeMappings = [queryRange];\n\n    // Create new ORDER BY continuation token\n    this.continuationToken = createOrderByQueryContinuationToken(\n      rangeMappings,\n      lastOrderByItems,\n      this.collectionLink, // Container RID/link\n      skipCount, // Number of documents with the same RID already processed\n      documentRid, // Document RID from the last item in the page\n      lastRangeBeforePageLimit.offset, // Current offset for OFFSET/LIMIT queries\n      lastRangeBeforePageLimit.limit, // Current limit for OFFSET/LIMIT queries\n      lastRangeBeforePageLimit.hashedLastResult, // Hash for distinct queries\n    );\n\n    return {\n      endIndex: rangeProcessingResult.endIndex,\n      processedRanges: rangeProcessingResult.processedRanges,\n    };\n  }\n\n  protected getCurrentContinuationToken(): OrderByQueryContinuationToken | undefined {\n    return this.continuationToken;\n  }\n\n  protected getSerializationFunction(): (token: OrderByQueryContinuationToken) => string {\n    return serializeOrderByQueryContinuationToken;\n  }\n}\n"]}
+{"version":3,"file":"OrderByQueryContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/OrderByQueryContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAGjF,OAAO,EACL,kCAAkC,EAClC,sCAAsC,GACvC,MAAM,oEAAoE,CAAC;AAE5E,OAAO,EAAE,+BAA+B,EAAE,MAAM,sEAAsE,CAAC;AAEvH;;;;GAIG;AACH,MAAM,OAAO,oCAAqC,SAAQ,4BAA4B;IACnE,iBAAiB,CAAuB;IAEzD,+DAA+D;IACvD,kBAAkB,CAAqC;IACvD,iBAAiB,CAAS;IAC1B,cAAc,GAAW,CAAC,CAAC;IAC3B,gBAAgB,GAAW,EAAE,CAAC;IAC9B,qBAAqB,CAAU;IAEvC,YAAY,cAAsB,EAAE,wBAAiC;QACnE,KAAK,CAAC,cAAc,EAAE,wBAAwB,CAAC,CAAC;QAChD,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;QAC5B,IAAI,wBAAwB,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,kCAAkC,CAAC,wBAAwB,CAAC,CAAC;YAC5E,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,aAAa,CAAC;YAC/C,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,YAAY,CAAC;YAC7C,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,SAAS,CAAC;YACvC,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,WAAW,CAAC;YAC3C,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,gBAAgB,CAAC;QACvD,CAAC;IACH,CAAC;IAES,4BAA4B,CAAC,cAAmC;QACxE,gFAAgF;QAChF,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;QAClC,IAAI,cAAc,CAAC,YAAY,EAAE,CAAC;YAChC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAES,4BAA4B,CAAC,gBAA0B,EAAE,QAAgB;QACjF,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAEO,sBAAsB,CAAC,QAAgB;QAC7C,IAAI,QAAQ,KAAK,CAAC,IAAI,QAAQ,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC;YAChE,sCAAsC;YACtC,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACN,yEAAyE;YACzE,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAES,0BAA0B,CAClC,QAAgB,EAChB,kBAA2B,KAAK;QAEhC,gFAAgF;QAChF,IAAI,eAAe,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC/C,IAAI,qBAAqB,CAAC;YAE1B,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAChC,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAC/E,CAAC;iBAAM,CAAC;gBACN,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1F,CAAC;YAED,MAAM,EAAE,wBAAwB,EAAE,GAAG,qBAAqB,CAAC;YAC3D,IAAI,wBAAwB,EAAE,CAAC;gBAC7B,IAAI,CAAC,kBAAkB,GAAG,CAAC,+BAA+B,CAAC,wBAAwB,CAAC,CAAC,CAAC;YACxF,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,CAAC;YACD,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,MAAM,qBAAqB,GAAG,IAAI,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACnF,MAAM,EAAE,wBAAwB,EAAE,GAAG,qBAAqB,CAAC;QAE3D,kDAAkD;QAClD,IAAI,CAAC,wBAAwB,EAAE,CAAC;YAC9B,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,+BAA+B,CAAC,wBAAwB,CAAC,CAAC;QAE7E,wDAAwD;QACxD,IAAI,gBAAmC,CAAC;QACxC,IAAI,WAAmB,CAAC;QACxB,IAAI,SAAS,GAAW,CAAC,CAAC;QAC1B,IAAI,qBAAqB,CAAC,QAAQ,GAAG,CAAC,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5E,MAAM,mBAAmB,GAAG,qBAAqB,CAAC,QAAQ,GAAG,CAAC,CAAC;YAE/D,IAAI,mBAAmB,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC;gBACxD,gBAAgB,GAAG,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC,YAAY,CAAC;gBAC5E,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC;gBAC/D,+EAA+E;gBAC/E,2EAA2E;gBAC3E,SAAS,GAAG,CAAC,CAAC;gBACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,mBAAmB,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC9C,IAAI,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACnD,SAAS,EAAE,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,sFAAsF;QACtF,mGAAmG;QACnG,IAAI,CAAC,gBAAgB,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,OAAO;gBACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;gBACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;aACvD,CAAC;QACJ,CAAC;QAED,4BAA4B;QAC5B,IAAI,CAAC,kBAAkB,GAAG,CAAC,UAAU,CAAC,CAAC;QACvC,IAAI,CAAC,iBAAiB,GAAG,gBAAgB,CAAC;QAC1C,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,WAAY,CAAC;QACrC,IAAI,CAAC,MAAM,GAAG,wBAAwB,CAAC,MAAM,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,wBAAwB,CAAC,KAAK,CAAC;QAC5C,IAAI,CAAC,qBAAqB,GAAG,wBAAwB,CAAC,gBAAgB,CAAC;QAEvE,OAAO;YACL,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;YACxC,eAAe,EAAE,qBAAqB,CAAC,eAAe;SACvD,CAAC;IACJ,CAAC;IAES,2BAA2B;QACnC,IAAI,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACxD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,GAAG,EAAE,IAAI,CAAC,cAAc;YACxB,aAAa,EAAE,IAAI,CAAC,kBAAkB;YACtC,YAAY,EAAE,IAAI,CAAC,iBAAiB;YACpC,SAAS,EAAE,IAAI,CAAC,cAAc;YAC9B,WAAW,EAAE,IAAI,CAAC,gBAAgB;YAClC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,gBAAgB,EAAE,IAAI,CAAC,qBAAqB;SAC7C,CAAC;IACJ,CAAC;IAES,wBAAwB;QAChC,OAAO,sCAAsC,CAAC;IAChD,CAAC;IAEO,eAAe;QACrB,IAAI,CAAC,kBAAkB,GAAG,SAAS,CAAC;QACpC,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,qBAAqB,GAAG,SAAS,CAAC;IACzC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { BaseContinuationTokenManager } from \"./BaseContinuationTokenManager.js\";\nimport type { ParallelQueryResult, OrderByItemWithRid } from \"../parallelQueryResult.js\";\nimport type { OrderByQueryContinuationToken } from \"../../documents/ContinuationToken/OrderByQueryContinuationToken.js\";\nimport {\n  parseOrderByQueryContinuationToken,\n  serializeOrderByQueryContinuationToken,\n} from \"../../documents/ContinuationToken/OrderByQueryContinuationToken.js\";\nimport type { QueryRangeWithContinuationToken } from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport { convertRangeMappingToQueryRange } from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\n\n/**\n * Manages continuation tokens for ORDER BY queries using single-range sequential processing.\n * Uses OrderByQueryContinuationToken for tracking ORDER BY items and skip counts.\n * @internal\n */\nexport class OrderByQueryContinuationTokenManager extends BaseContinuationTokenManager {\n  private readonly orderByItemsArray: OrderByItemWithRid[];\n\n  // OrderBy-specific token state (computed into token on demand)\n  private tokenRangeMappings?: QueryRangeWithContinuationToken[];\n  private tokenOrderByItems?: any[];\n  private tokenSkipCount: number = 0;\n  private tokenDocumentRid: string = \"\";\n  private tokenHashedLastResult?: string;\n\n  constructor(collectionLink: string, initialContinuationToken?: string) {\n    super(collectionLink, initialContinuationToken);\n    this.orderByItemsArray = [];\n    if (initialContinuationToken) {\n      const parsed = parseOrderByQueryContinuationToken(initialContinuationToken);\n      this.tokenRangeMappings = parsed.rangeMappings;\n      this.tokenOrderByItems = parsed.orderByItems;\n      this.tokenSkipCount = parsed.skipCount;\n      this.tokenDocumentRid = parsed.documentRid;\n      this.tokenHashedLastResult = parsed.hashedLastResult;\n    }\n  }\n\n  protected processQuerySpecificResponse(responseResult: ParallelQueryResult): void {\n    // Clear existing items and add new ones without reassigning the array reference\n    this.orderByItemsArray.length = 0;\n    if (responseResult.orderByItems) {\n      this.orderByItemsArray.push(...responseResult.orderByItems);\n    }\n  }\n\n  protected performQuerySpecificDataTrim(_processedRanges: string[], endIndex: number): void {\n    this.sliceOrderByItemsArray(endIndex);\n  }\n\n  private sliceOrderByItemsArray(endIndex: number): void {\n    if (endIndex === 0 || endIndex >= this.orderByItemsArray.length) {\n      // Clear the array without reassigning\n      this.orderByItemsArray.length = 0;\n    } else {\n      // Remove items from 0 to endIndex-1, keeping items from endIndex onwards\n      this.orderByItemsArray.splice(0, endIndex);\n    }\n  }\n\n  protected processRangesForPagination(\n    pageSize: number,\n    isResponseEmpty: boolean = false,\n  ): { endIndex: number; processedRanges: string[] } {\n    // Handle empty response case - preserve previous token state with updated range\n    if (isResponseEmpty && this.tokenRangeMappings) {\n      let rangeProcessingResult;\n\n      if (this.rangeList.length === 0) {\n        rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);\n      } else {\n        rangeProcessingResult = this.partitionManager.processEmptyOrderByRanges(this.rangeList);\n      }\n\n      const { lastRangeBeforePageLimit } = rangeProcessingResult;\n      if (lastRangeBeforePageLimit) {\n        this.tokenRangeMappings = [convertRangeMappingToQueryRange(lastRangeBeforePageLimit)];\n      } else {\n        this.clearTokenState();\n      }\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    // Normal processing path - handle non-empty responses\n    const rangeProcessingResult = this.partitionManager.processOrderByRanges(pageSize);\n    const { lastRangeBeforePageLimit } = rangeProcessingResult;\n\n    // Check if we have a valid range to continue with\n    if (!lastRangeBeforePageLimit) {\n      this.clearTokenState();\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    const queryRange = convertRangeMappingToQueryRange(lastRangeBeforePageLimit);\n\n    // Extract ORDER BY items from the last item on the page\n    let lastOrderByItems: any[] | undefined;\n    let documentRid: string;\n    let skipCount: number = 0;\n    if (rangeProcessingResult.endIndex > 0 && this.orderByItemsArray.length > 0) {\n      const lastItemIndexOnPage = rangeProcessingResult.endIndex - 1;\n\n      if (lastItemIndexOnPage < this.orderByItemsArray.length) {\n        lastOrderByItems = this.orderByItemsArray[lastItemIndexOnPage].orderByItems;\n        documentRid = this.orderByItemsArray[lastItemIndexOnPage]._rid;\n        // Calculate skip count: count how many documents in the page have the same RID\n        // This handles JOIN queries where multiple documents can have the same RID\n        skipCount = 0;\n        for (let i = 0; i <= lastItemIndexOnPage; i++) {\n          if (this.orderByItemsArray[i]._rid === documentRid) {\n            skipCount++;\n          }\n        }\n      }\n    }\n\n    // If we don't have valid ORDER BY items, we cannot create a proper continuation token\n    // This can happen when the response doesn't contain ORDER BY metadata or when there are no results\n    if (!lastOrderByItems || lastOrderByItems.length === 0) {\n      this.clearTokenState();\n      return {\n        endIndex: rangeProcessingResult.endIndex,\n        processedRanges: rangeProcessingResult.processedRanges,\n      };\n    }\n\n    // Update token state fields\n    this.tokenRangeMappings = [queryRange];\n    this.tokenOrderByItems = lastOrderByItems;\n    this.tokenSkipCount = skipCount;\n    this.tokenDocumentRid = documentRid!;\n    this.offset = lastRangeBeforePageLimit.offset;\n    this.limit = lastRangeBeforePageLimit.limit;\n    this.tokenHashedLastResult = lastRangeBeforePageLimit.hashedLastResult;\n\n    return {\n      endIndex: rangeProcessingResult.endIndex,\n      processedRanges: rangeProcessingResult.processedRanges,\n    };\n  }\n\n  protected getCurrentContinuationToken(): OrderByQueryContinuationToken | undefined {\n    if (!this.tokenRangeMappings || !this.tokenOrderByItems) {\n      return undefined;\n    }\n    return {\n      rid: this.collectionLink,\n      rangeMappings: this.tokenRangeMappings,\n      orderByItems: this.tokenOrderByItems,\n      skipCount: this.tokenSkipCount,\n      documentRid: this.tokenDocumentRid,\n      offset: this.offset,\n      limit: this.limit,\n      hashedLastResult: this.tokenHashedLastResult,\n    };\n  }\n\n  protected getSerializationFunction(): (token: OrderByQueryContinuationToken) => string {\n    return serializeOrderByQueryContinuationToken;\n  }\n\n  private clearTokenState(): void {\n    this.tokenRangeMappings = undefined;\n    this.tokenOrderByItems = undefined;\n    this.tokenSkipCount = 0;\n    this.tokenDocumentRid = \"\";\n    this.tokenHashedLastResult = undefined;\n  }\n}\n"]}

package/dist/esm/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.js

@@ -1,42 +1,36 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 import { BaseContinuationTokenManager } from "./BaseContinuationTokenManager.js";
-import { createCompositeQueryContinuationToken, serializeCompositeToken, parseCompositeQueryContinuationToken, convertRangeMappingToQueryRange, } from "../../documents/ContinuationToken/CompositeQueryContinuationToken.js";
+import { serializeCompositeToken, convertRangeMappingToQueryRange, } from "../../documents/ContinuationToken/CompositeQueryContinuationToken.js";
 /**
  * Manages continuation tokens for parallel queries using multi-range aggregation.
  * Uses CompositeQueryContinuationToken for range tracking across multiple partitions.
  * @internal
  */
 export class ParallelQueryContinuationTokenManager extends BaseContinuationTokenManager {
-    continuationToken;
-    collectionLink;
-    constructor(collectionLink, initialContinuationToken) {
-        super(initialContinuationToken);
-        this.collectionLink = collectionLink;
-        if (initialContinuationToken) {
-            this.continuationToken = parseCompositeQueryContinuationToken(initialContinuationToken);
-        }
-    }
     processRangesForPagination(pageSize, _isResponseEmpty) {
         const result = this.partitionManager.processParallelRanges(pageSize);
         if (!result || !result.processedRangeMappings || result.processedRangeMappings.length === 0) {
             return { endIndex: 0, processedRanges: [] };
         }
         const rangeMappings = result.processedRangeMappings.map((mapping) => convertRangeMappingToQueryRange(mapping));
-        if (!this.continuationToken) {
-            this.continuationToken = createCompositeQueryContinuationToken(this.collectionLink, rangeMappings);
-        }
-        else {
-            this.updateExistingCompositeContinuationToken(rangeMappings);
-        }
+        this.updateRangeList(rangeMappings);
         if (result.lastPartitionBeforeCutoff && result.lastPartitionBeforeCutoff.mapping) {
-            this.continuationToken.offset = result.lastPartitionBeforeCutoff.mapping.offset;
-            this.continuationToken.limit = result.lastPartitionBeforeCutoff.mapping.limit;
+            this.offset = result.lastPartitionBeforeCutoff.mapping.offset;
+            this.limit = result.lastPartitionBeforeCutoff.mapping.limit;
         }
         return { endIndex: result.endIndex, processedRanges: result.processedRanges };
     }
     getCurrentContinuationToken() {
-        return this.continuationToken;
+        if (this.rangeList.length === 0) {
+            return undefined;
+        }
+        return {
+            rid: this.collectionLink,
+            rangeMappings: this.rangeList,
+            offset: this.offset,
+            limit: this.limit,
+        };
     }
     getSerializationFunction() {
         return serializeCompositeToken;
@@ -47,18 +41,15 @@ export class ParallelQueryContinuationTokenManager extends BaseContinuationToken
     performQuerySpecificDataTrim(_processedRanges, _endIndex) {
         // Parallel queries don't need additional cleanup
     }
-    updateExistingCompositeContinuationToken(rangeMappings) {
+    updateRangeList(rangeMappings) {
         for (const newRange of rangeMappings) {
-            // Check if this range already exists in the token
-            const existingRangeIndex = this.continuationToken.rangeMappings.findIndex((existingRange) => existingRange.queryRange.min === newRange.queryRange.min &&
+            const existingRangeIndex = this.rangeList.findIndex((existingRange) => existingRange.queryRange.min === newRange.queryRange.min &&
                 existingRange.queryRange.max === newRange.queryRange.max);
             if (existingRangeIndex >= 0) {
-                // Range exists - update the continuation token
-                this.continuationToken.rangeMappings[existingRangeIndex] = newRange;
+                this.rangeList[existingRangeIndex] = newRange;
             }
             else {
-                // New range - add to the rangeMappings array
-                this.continuationToken.rangeMappings.push(newRange);
+                this.rangeList.push(newRange);
             }
         }
     }

package/dist/esm/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"ParallelQueryContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAMjF,OAAO,EACL,qCAAqC,EACrC,uBAAuB,EACvB,oCAAoC,EACpC,+BAA+B,GAChC,MAAM,sEAAsE,CAAC;AAG9E;;;;GAIG;AACH,MAAM,OAAO,qCAAsC,SAAQ,4BAA4B;IAC7E,iBAAiB,CAA8C;IACtD,cAAc,CAAS;IAExC,YAAY,cAAsB,EAAE,wBAAiC;QACnE,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,wBAAwB,EAAE,CAAC;YAC7B,IAAI,CAAC,iBAAiB,GAAG,oCAAoC,CAAC,wBAAwB,CAAC,CAAC;QAC1F,CAAC;IACH,CAAC;IAES,0BAA0B,CAClC,QAAgB,EAChB,gBAAyB;QAEzB,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACrE,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,sBAAsB,IAAI,MAAM,CAAC,sBAAsB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5F,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;QAC9C,CAAC;QAED,MAAM,aAAa,GAAsC,MAAM,CAAC,sBAAsB,CAAC,GAAG,CACxF,CAAC,OAA0B,EAAE,EAAE,CAAC,+BAA+B,CAAC,OAAO,CAAC,CACzE,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5B,IAAI,CAAC,iBAAiB,GAAG,qCAAqC,CAC5D,IAAI,CAAC,cAAc,EACnB,aAAa,CACd,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,wCAAwC,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC;QAED,IAAI,MAAM,CAAC,yBAAyB,IAAI,MAAM,CAAC,yBAAyB,CAAC,OAAO,EAAE,CAAC;YACjF,IAAI,CAAC,iBAAkB,CAAC,MAAM,GAAG,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,MAAM,CAAC;YACjF,IAAI,CAAC,iBAAkB,CAAC,KAAK,GAAG,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,KAAK,CAAC;QACjF,CAAC;QACD,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;IAChF,CAAC;IAES,2BAA2B;QACnC,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAES,wBAAwB;QAChC,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAES,4BAA4B,CAAC,eAAoC;QACzE,6DAA6D;IAC/D,CAAC;IAES,4BAA4B,CAAC,gBAA0B,EAAE,SAAiB;QAClF,iDAAiD;IACnD,CAAC;IAEO,wCAAwC,CAC9C,aAAgD;QAEhD,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;YACrC,kDAAkD;YAClD,MAAM,kBAAkB,GAAG,IAAI,CAAC,iBAAkB,CAAC,aAAa,CAAC,SAAS,CACxE,CAAC,aAAa,EAAE,EAAE,CAChB,aAAa,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,UAAU,CAAC,GAAG;gBACxD,aAAa,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,UAAU,CAAC,GAAG,CAC3D,CAAC;YAEF,IAAI,kBAAkB,IAAI,CAAC,EAAE,CAAC;gBAC5B,+CAA+C;gBAC/C,IAAI,CAAC,iBAAkB,CAAC,aAAa,CAAC,kBAAkB,CAAC,GAAG,QAAQ,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACN,6CAA6C;gBAC7C,IAAI,CAAC,iBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { BaseContinuationTokenManager } from \"./BaseContinuationTokenManager.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\nimport type {\n  CompositeQueryContinuationToken,\n  QueryRangeWithContinuationToken,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport {\n  createCompositeQueryContinuationToken,\n  serializeCompositeToken,\n  parseCompositeQueryContinuationToken,\n  convertRangeMappingToQueryRange,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport type { QueryRangeMapping } from \"../queryRangeMapping.js\";\n\n/**\n * Manages continuation tokens for parallel queries using multi-range aggregation.\n * Uses CompositeQueryContinuationToken for range tracking across multiple partitions.\n * @internal\n */\nexport class ParallelQueryContinuationTokenManager extends BaseContinuationTokenManager {\n  private continuationToken: CompositeQueryContinuationToken | undefined;\n  private readonly collectionLink: string;\n\n  constructor(collectionLink: string, initialContinuationToken?: string) {\n    super(initialContinuationToken);\n    this.collectionLink = collectionLink;\n    if (initialContinuationToken) {\n      this.continuationToken = parseCompositeQueryContinuationToken(initialContinuationToken);\n    }\n  }\n\n  protected processRangesForPagination(\n    pageSize: number,\n    _isResponseEmpty: boolean,\n  ): { endIndex: number; processedRanges: string[] } {\n    const result = this.partitionManager.processParallelRanges(pageSize);\n    if (!result || !result.processedRangeMappings || result.processedRangeMappings.length === 0) {\n      return { endIndex: 0, processedRanges: [] };\n    }\n\n    const rangeMappings: QueryRangeWithContinuationToken[] = result.processedRangeMappings.map(\n      (mapping: QueryRangeMapping) => convertRangeMappingToQueryRange(mapping),\n    );\n\n    if (!this.continuationToken) {\n      this.continuationToken = createCompositeQueryContinuationToken(\n        this.collectionLink,\n        rangeMappings,\n      );\n    } else {\n      this.updateExistingCompositeContinuationToken(rangeMappings);\n    }\n\n    if (result.lastPartitionBeforeCutoff && result.lastPartitionBeforeCutoff.mapping) {\n      this.continuationToken!.offset = result.lastPartitionBeforeCutoff.mapping.offset;\n      this.continuationToken!.limit = result.lastPartitionBeforeCutoff.mapping.limit;\n    }\n    return { endIndex: result.endIndex, processedRanges: result.processedRanges };\n  }\n\n  protected getCurrentContinuationToken(): CompositeQueryContinuationToken | undefined {\n    return this.continuationToken;\n  }\n\n  protected getSerializationFunction(): (token: CompositeQueryContinuationToken) => string {\n    return serializeCompositeToken;\n  }\n\n  protected processQuerySpecificResponse(_responseResult: ParallelQueryResult): void {\n    // Parallel queries don't need additional response processing\n  }\n\n  protected performQuerySpecificDataTrim(_processedRanges: string[], _endIndex: number): void {\n    // Parallel queries don't need additional cleanup\n  }\n\n  private updateExistingCompositeContinuationToken(\n    rangeMappings: QueryRangeWithContinuationToken[],\n  ): void {\n    for (const newRange of rangeMappings) {\n      // Check if this range already exists in the token\n      const existingRangeIndex = this.continuationToken!.rangeMappings.findIndex(\n        (existingRange) =>\n          existingRange.queryRange.min === newRange.queryRange.min &&\n          existingRange.queryRange.max === newRange.queryRange.max,\n      );\n\n      if (existingRangeIndex >= 0) {\n        // Range exists - update the continuation token\n        this.continuationToken!.rangeMappings[existingRangeIndex] = newRange;\n      } else {\n        // New range - add to the rangeMappings array\n        this.continuationToken!.rangeMappings.push(newRange);\n      }\n    }\n  }\n}\n"]}
+{"version":3,"file":"ParallelQueryContinuationTokenManager.js","sourceRoot":"","sources":["../../../../src/queryExecutionContext/ContinuationTokenManager/ParallelQueryContinuationTokenManager.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAMjF,OAAO,EACL,uBAAuB,EACvB,+BAA+B,GAChC,MAAM,sEAAsE,CAAC;AAG9E;;;;GAIG;AACH,MAAM,OAAO,qCAAsC,SAAQ,4BAA4B;IAC3E,0BAA0B,CAClC,QAAgB,EAChB,gBAAyB;QAEzB,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QACrE,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,sBAAsB,IAAI,MAAM,CAAC,sBAAsB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5F,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;QAC9C,CAAC;QAED,MAAM,aAAa,GAAsC,MAAM,CAAC,sBAAsB,CAAC,GAAG,CACxF,CAAC,OAA0B,EAAE,EAAE,CAAC,+BAA+B,CAAC,OAAO,CAAC,CACzE,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;QAEpC,IAAI,MAAM,CAAC,yBAAyB,IAAI,MAAM,CAAC,yBAAyB,CAAC,OAAO,EAAE,CAAC;YACjF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,MAAM,CAAC;YAC9D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,yBAAyB,CAAC,OAAO,CAAC,KAAK,CAAC;QAC9D,CAAC;QACD,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;IAChF,CAAC;IAES,2BAA2B;QACnC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,GAAG,EAAE,IAAI,CAAC,cAAc;YACxB,aAAa,EAAE,IAAI,CAAC,SAAS;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;IACJ,CAAC;IAES,wBAAwB;QAChC,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAES,4BAA4B,CAAC,eAAoC;QACzE,6DAA6D;IAC/D,CAAC;IAES,4BAA4B,CAAC,gBAA0B,EAAE,SAAiB;QAClF,iDAAiD;IACnD,CAAC;IAEO,eAAe,CAAC,aAAgD;QACtE,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;YACrC,MAAM,kBAAkB,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CACjD,CAAC,aAAa,EAAE,EAAE,CAChB,aAAa,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,UAAU,CAAC,GAAG;gBACxD,aAAa,CAAC,UAAU,CAAC,GAAG,KAAK,QAAQ,CAAC,UAAU,CAAC,GAAG,CAC3D,CAAC;YAEF,IAAI,kBAAkB,IAAI,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,GAAG,QAAQ,CAAC;YAChD,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;IACH,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { BaseContinuationTokenManager } from \"./BaseContinuationTokenManager.js\";\nimport type { ParallelQueryResult } from \"../parallelQueryResult.js\";\nimport type {\n  CompositeQueryContinuationToken,\n  QueryRangeWithContinuationToken,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport {\n  serializeCompositeToken,\n  convertRangeMappingToQueryRange,\n} from \"../../documents/ContinuationToken/CompositeQueryContinuationToken.js\";\nimport type { QueryRangeMapping } from \"../queryRangeMapping.js\";\n\n/**\n * Manages continuation tokens for parallel queries using multi-range aggregation.\n * Uses CompositeQueryContinuationToken for range tracking across multiple partitions.\n * @internal\n */\nexport class ParallelQueryContinuationTokenManager extends BaseContinuationTokenManager {\n  protected processRangesForPagination(\n    pageSize: number,\n    _isResponseEmpty: boolean,\n  ): { endIndex: number; processedRanges: string[] } {\n    const result = this.partitionManager.processParallelRanges(pageSize);\n    if (!result || !result.processedRangeMappings || result.processedRangeMappings.length === 0) {\n      return { endIndex: 0, processedRanges: [] };\n    }\n\n    const rangeMappings: QueryRangeWithContinuationToken[] = result.processedRangeMappings.map(\n      (mapping: QueryRangeMapping) => convertRangeMappingToQueryRange(mapping),\n    );\n\n    this.updateRangeList(rangeMappings);\n\n    if (result.lastPartitionBeforeCutoff && result.lastPartitionBeforeCutoff.mapping) {\n      this.offset = result.lastPartitionBeforeCutoff.mapping.offset;\n      this.limit = result.lastPartitionBeforeCutoff.mapping.limit;\n    }\n    return { endIndex: result.endIndex, processedRanges: result.processedRanges };\n  }\n\n  protected getCurrentContinuationToken(): CompositeQueryContinuationToken | undefined {\n    if (this.rangeList.length === 0) {\n      return undefined;\n    }\n    return {\n      rid: this.collectionLink,\n      rangeMappings: this.rangeList,\n      offset: this.offset,\n      limit: this.limit,\n    };\n  }\n\n  protected getSerializationFunction(): (token: CompositeQueryContinuationToken) => string {\n    return serializeCompositeToken;\n  }\n\n  protected processQuerySpecificResponse(_responseResult: ParallelQueryResult): void {\n    // Parallel queries don't need additional response processing\n  }\n\n  protected performQuerySpecificDataTrim(_processedRanges: string[], _endIndex: number): void {\n    // Parallel queries don't need additional cleanup\n  }\n\n  private updateRangeList(rangeMappings: QueryRangeWithContinuationToken[]): void {\n    for (const newRange of rangeMappings) {\n      const existingRangeIndex = this.rangeList.findIndex(\n        (existingRange) =>\n          existingRange.queryRange.min === newRange.queryRange.min &&\n          existingRange.queryRange.max === newRange.queryRange.max,\n      );\n\n      if (existingRangeIndex >= 0) {\n        this.rangeList[existingRangeIndex] = newRange;\n      } else {\n        this.rangeList.push(newRange);\n      }\n    }\n  }\n}\n"]}

package/dist/esm/queryExecutionContext/PartitionRangeManager.d.ts

@@ -1,5 +1,12 @@
-import { QueryRangeWithContinuationToken } from "../documents/ContinuationToken/CompositeQueryContinuationToken.js";
+import type { QueryRangeWithContinuationToken } from "../documents/ContinuationToken/CompositeQueryContinuationToken.js";
 import type { QueryRangeMapping } from "./queryRangeMapping.js";
+/**
+ * Checks if a continuation token indicates an exhausted partition.
+ * @param continuationToken - The continuation token to check
+ * @returns true if the partition is exhausted (null, empty, or "null" string)
+ * @hidden
+ */
+export declare function isPartitionExhausted(continuationToken: string | undefined): boolean;
 /**
  * Manages partition key range mappings for query execution.
  * Handles range operations, offset/limit processing, and distinct query logic.
@@ -12,12 +19,6 @@ export declare class PartitionRangeManager {
      * Gets a copy of the current partition key range map for constructor pattern
      */
     getPartitionKeyRangeMap(): Map<string, QueryRangeMapping>;
-    /**
-     * Checks if a continuation token indicates an exhausted partition
-     * @param continuationToken - The continuation token to check
-     * @returns true if the partition is exhausted (null, empty, or "null" string)
-     */
-    private isPartitionExhausted;
     /**
      * Adds a range mapping to the partition key range map
      * Does not allow updates to existing keys - only new additions

package/dist/esm/queryExecutionContext/PartitionRangeManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"PartitionRangeManager.d.ts","sourceRoot":"","sources":["../../../src/queryExecutionContext/PartitionRangeManager.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,+BAA+B,EAAE,MAAM,mEAAmE,CAAC;AACpH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAEhE;;;;GAIG;AACH,qBAAa,qBAAqB;IAChC,OAAO,CAAC,oBAAoB,CAA6C;gBAE7D,2BAA2B,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC;IAMxE;;OAEG;IACI,uBAAuB,IAAI,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC;IAIhE;;;;OAIG;IACH,OAAO,CAAC,oBAAoB;IAS5B;;;;;OAKG;IACH,OAAO,CAAC,wBAAwB;IAMhC;;OAEG;IACI,2BAA2B,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAIzD;;;OAGG;IACI,uBAAuB,CAAC,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,GAAG,IAAI;IAQ1F;;OAEG;IACI,oBAAoB,IAAI,OAAO;IAItC;;;;OAIG;IACI,qBAAqB,CAAC,aAAa,EAAE,iBAAiB,EAAE,GAAG,iBAAiB,EAAE;IAkBrF;;OAEG;IACI,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG;QAC7C,QAAQ,EAAE,MAAM,CAAC;QACjB,eAAe,EAAE,MAAM,EAAE,CAAC;QAC1B,wBAAwB,EAAE,iBAAiB,GAAG,IAAI,CAAC;KACpD;IAqBM,yBAAyB,CAAC,MAAM,EAAE,+BAA+B,EAAE,GAAG;QAC3E,QAAQ,EAAE,MAAM,CAAC;QACjB,eAAe,EAAE,MAAM,EAAE,CAAC;QAC1B,wBAAwB,EAAE,iBAAiB,GAAG,SAAS,CAAC;KACzD;IAsBD;;OAEG;IACI,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG;QAC9C,QAAQ,EAAE,MAAM,CAAC;QACjB,eAAe,EAAE,MAAM,EAAE,CAAC;QAC1B,sBAAsB,EAAE,iBAAiB,EAAE,CAAC;QAC5C,yBAAyB,CAAC,EAAE;YAAE,OAAO,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,iBAAiB,CAAA;SAAE,CAAC;KAC7E;CAkCF"}
+{"version":3,"file":"PartitionRangeManager.d.ts","sourceRoot":"","sources":["../../../src/queryExecutionContext/PartitionRangeManager.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,mEAAmE,CAAC;AACzH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAEhE;;;;;GAKG;AACH,wBAAgB,oBAAoB,CAAC,iBAAiB,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,CAOnF;AAED;;;;GAIG;AACH,qBAAa,qBAAqB;IAChC,OAAO,CAAC,oBAAoB,CAA6C;gBAE7D,2BAA2B,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC;IAMxE;;OAEG;IACI,uBAAuB,IAAI,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC;IAIhE;;;;;OAKG;IACH,OAAO,CAAC,wBAAwB;IAMhC;;OAEG;IACI,2BAA2B,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAIzD;;;OAGG;IACI,uBAAuB,CAAC,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,GAAG,IAAI;IAQ1F;;OAEG;IACI,oBAAoB,IAAI,OAAO;IAItC;;;;OAIG;IACI,qBAAqB,CAAC,aAAa,EAAE,iBAAiB,EAAE,GAAG,iBAAiB,EAAE;IAkBrF;;OAEG;IACI,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG;QAC7C,QAAQ,EAAE,MAAM,CAAC;QACjB,eAAe,EAAE,MAAM,EAAE,CAAC;QAC1B,wBAAwB,EAAE,iBAAiB,GAAG,IAAI,CAAC;KACpD;IAqBM,yBAAyB,CAAC,MAAM,EAAE,+BAA+B,EAAE,GAAG;QAC3E,QAAQ,EAAE,MAAM,CAAC;QACjB,eAAe,EAAE,MAAM,EAAE,CAAC;QAC1B,wBAAwB,EAAE,iBAAiB,GAAG,SAAS,CAAC;KACzD;IAsBD;;OAEG;IACI,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG;QAC9C,QAAQ,EAAE,MAAM,CAAC;QACjB,eAAe,EAAE,MAAM,EAAE,CAAC;QAC1B,sBAAsB,EAAE,iBAAiB,EAAE,CAAC;QAC5C,yBAAyB,CAAC,EAAE;YAAE,OAAO,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,iBAAiB,CAAA;SAAE,CAAC;KAC7E;CAkCF"}