[Cosmos] cross partition continuation token

sdk/cosmosdb/cosmos/src/client/Item/Items.ts

@@ -147,6 +147,10 @@ export class Items {
    */
   public query<T>(query: string | SqlQuerySpec, options?: FeedOptions): QueryIterator<T>;
   public query<T>(query: string | SqlQuerySpec, options: FeedOptions = {}): QueryIterator<T> {
+    console.log("==========================================");
+    console.log("ITEMS.query() method called");
+    console.log("enableEncryption:", this.clientContext.enableEncryption);
+    console.log("==========================================");
     const path = getPathFromLink(this.container.url, ResourceType.item);
     const id = getIdFromLink(this.container.url);
 

sdk/cosmosdb/cosmos/src/documents/ContinuationToken/OrderByQueryContinuationToken.ts

@@ -0,0 +1,84 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+/**
+ * Continuation token for order by queries.
+ * @internal
+ */
+export interface OrderByQueryContinuationToken {
+  /**
+   * Composite token for the query continuation
+   */
+  compositeToken: string;
+
+  /**
+   * Order by items for the query
+   */
+  orderByItems: any[];
+
+  /**
+   * Resource ID of the container for which the continuation token is issued
+   */
+  rid: string;
+
+  /**
+   * Number of items to skip in the query
+   */
+  skipCount: number;
+
+  /**
+   * Current offset value for OFFSET/LIMIT queries
+   */
+  offset?: number;
+
+  /**
+   * Current limit value for OFFSET/LIMIT queries
+   */
+  limit?: number;
+
+  /**
+   * Hash of the last document result for distinct order queries
+   * Used to ensure duplicates are not returned across continuation boundaries
+   */
+  hashedLastResult?: string;
+}
+
+/**
+ * Creates an OrderByQueryContinuationToken
+ * @internal
+ */
+export function createOrderByQueryContinuationToken(
+  compositeToken: string,
+  orderByItems: any[],
+  rid: string,
+  skipCount: number,
+  offset?: number,
+  limit?: number,
+  hashedLastResult?: string
+): OrderByQueryContinuationToken {
+  return {
+    compositeToken,
+    orderByItems,
+    rid,
+    skipCount,
+    offset,
+    limit,
+    hashedLastResult,
+  };
+}
+
+/**
+ * Serializes an OrderByQueryContinuationToken to a JSON string
+ * @internal
+ */
+export function serializeOrderByQueryContinuationToken(token: OrderByQueryContinuationToken): string {
+  return JSON.stringify(token);
+}
+
+/**
+ * Deserializes a JSON string to an OrderByQueryContinuationToken
+ * @internal
+ */
+export function parseOrderByQueryContinuationToken(tokenString: string): OrderByQueryContinuationToken {
+  return JSON.parse(tokenString);
+}

sdk/cosmosdb/cosmos/src/queryExecutionContext/CompositeQueryContinuationToken.ts

@@ -0,0 +1,85 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import type { QueryRangeMapping } from "./QueryRangeMapping.js";
+
+/**
+ * @hidden
+ * Composite continuation token for parallel query execution across multiple partition ranges
+ */
+export interface CompositeQueryContinuationToken {
+  /**
+   * Resource ID of the container for which the continuation token is issued
+   */
+  readonly rid: string;
+
+  /**
+   * List of query range mappings part of the continuation token
+   */
+  rangeMappings: QueryRangeMapping[];
+
+  /**
+   * Current offset value for OFFSET/LIMIT queries
+   */
+  offset?: number;
+
+  /**
+   * Current limit value for OFFSET/LIMIT queries
+   */
+  limit?: number;
+
+}
+
+/**
+ * Creates a new CompositeQueryContinuationToken
+ * @hidden
+ */
+export function createCompositeQueryContinuationToken(
+  rid: string, 
+  rangeMappings: QueryRangeMapping[], 
+  offset?: number,
+  limit?: number
+): CompositeQueryContinuationToken {
+  return {
+    rid,
+    rangeMappings,
+    offset,
+    limit,
+  };
+}
+
+/**
+ * Adds a range mapping to the continuation token
+ * @hidden
+ */
+export function addRangeMappingToCompositeToken(token: CompositeQueryContinuationToken, rangeMapping: QueryRangeMapping): void {
+  token.rangeMappings.push(rangeMapping);
+}
+
+/**
+ * Serializes the composite continuation token to a JSON string
+ * @hidden
+ */
+export function compositeTokenToString(token: CompositeQueryContinuationToken): string {
+  return JSON.stringify({
+    rid: token.rid,
+    rangeMappings: token.rangeMappings,
+    offset: token.offset,
+    limit: token.limit,
+  });
+}
+
+/**
+ * Deserializes a JSON string to a CompositeQueryContinuationToken
+ * @hidden
+ */
+export function compositeTokenFromString(tokenString: string): CompositeQueryContinuationToken {
+  const parsed = JSON.parse(tokenString);
+  return createCompositeQueryContinuationToken(
+    parsed.rid,
+    parsed.rangeMappings,
+    parsed.globalContinuationToken,
+    parsed.offset,
+    parsed.limit,
+  );
+}