allow users to provide continuation token (#1537)

Author: demoray

sdk/data_cosmos/src/operations/list_attachments.rs

@@ -19,7 +19,8 @@ operation! {
     ?max_item_count: MaxItemCount,
     ?a_im: ChangeFeed,
     ?if_match_condition: IfMatchCondition,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListAttachmentsBuilder {
@@ -49,6 +50,7 @@ impl ListAttachmentsBuilder {
                     &mut request,
                 );
 
+                let continuation = continuation.or(this.continuation);
                 request.insert_headers(&continuation);
 
                 let response = this

sdk/data_cosmos/src/operations/list_collections.rs

@@ -11,7 +11,8 @@ operation! {
     ListCollections,
     client: DatabaseClient,
     ?max_item_count: MaxItemCount,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListCollectionsBuilder {
@@ -26,6 +27,7 @@ impl ListCollectionsBuilder {
                 }
                 request.insert_headers(&this.max_item_count.unwrap_or_default());
 
+                let continuation = continuation.or(this.continuation);
                 request.insert_headers(&continuation);
 
                 let response = this

sdk/data_cosmos/src/operations/list_databases.rs

@@ -11,7 +11,8 @@ operation! {
     ListDatabases,
     client: CosmosClient,
     ?max_item_count: MaxItemCount,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListDatabasesBuilder {
@@ -27,6 +28,9 @@ impl ListDatabasesBuilder {
                 request.insert_headers(&this.max_item_count.unwrap_or_default());
                 request.insert_headers(&continuation);
 
+                let continuation = continuation.or(this.continuation);
+                request.insert_headers(&continuation);
+
                 let response = this
                     .client
                     .pipeline()

sdk/data_cosmos/src/operations/list_documents.rs

@@ -19,7 +19,8 @@ operation! {
     ?a_im: ChangeFeed,
     ?if_match_condition: IfMatchCondition,
     ?consistency_level: ConsistencyLevel,
-    ?partition_range_id: PartitionRangeId
+    ?partition_range_id: PartitionRangeId,
+    ?continuation: Continuation,
 }
 
 impl ListDocumentsBuilder {
@@ -47,6 +48,8 @@ impl ListDocumentsBuilder {
                 req.insert_headers(&this.max_item_count.unwrap_or_default());
                 req.insert_headers(&this.a_im.unwrap_or_default());
                 req.insert_headers(&this.partition_range_id);
+
+                let continuation = continuation.or(this.continuation);
                 req.insert_headers(&continuation);
 
                 let response = this

sdk/data_cosmos/src/operations/list_permissions.rs

@@ -10,7 +10,8 @@ operation! {
     ListPermissions,
     client: UserClient,
     ?max_item_count: MaxItemCount,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListPermissionsBuilder {
@@ -33,6 +34,7 @@ impl ListPermissionsBuilder {
                 }
                 request.insert_headers(&this.max_item_count.unwrap_or_default());
 
+                let continuation = continuation.or(this.continuation);
                 request.insert_headers(&continuation);
 
                 let response = this

sdk/data_cosmos/src/operations/list_stored_procedures.rs

@@ -14,7 +14,8 @@ operation! {
     ListStoredProcedures,
     client: CollectionClient,
     ?max_item_count: MaxItemCount,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListStoredProceduresBuilder {
@@ -37,6 +38,7 @@ impl ListStoredProceduresBuilder {
                 }
                 request.insert_headers(&this.max_item_count.unwrap_or_default());
 
+                let continuation = continuation.or(this.continuation);
                 request.insert_headers(&continuation);
 
                 let response = this

sdk/data_cosmos/src/operations/list_triggers.rs

@@ -13,7 +13,8 @@ operation! {
     client: CollectionClient,
     ?if_match_condition: IfMatchCondition,
     ?max_item_count: MaxItemCount,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListTriggersBuilder {
@@ -37,6 +38,7 @@ impl ListTriggersBuilder {
                 }
                 request.insert_headers(&this.max_item_count.unwrap_or_default());
 
+                let continuation = continuation.or(this.continuation);
                 request.insert_headers(&continuation);
 
                 let response = this

sdk/data_cosmos/src/operations/list_user_defined_functions.rs

@@ -18,7 +18,8 @@ operation! {
     client: CollectionClient,
     ?if_match_condition: IfMatchCondition,
     ?max_item_count: MaxItemCount,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListUserDefinedFunctionsBuilder {
@@ -42,6 +43,7 @@ impl ListUserDefinedFunctionsBuilder {
                 }
                 request.insert_headers(&this.max_item_count.unwrap_or_default());
 
+                let continuation = continuation.or(this.continuation);
                 request.insert_headers(&continuation);
 
                 let response = this

sdk/data_cosmos/src/operations/list_users.rs

@@ -15,7 +15,8 @@ operation! {
     ListUsers,
     client: DatabaseClient,
     ?max_item_count: MaxItemCount,
-    ?consistency_level: ConsistencyLevel
+    ?consistency_level: ConsistencyLevel,
+    ?continuation: Continuation,
 }
 
 impl ListUsersBuilder {
@@ -34,9 +35,8 @@ impl ListUsersBuilder {
                 }
                 request.insert_headers(&this.max_item_count.unwrap_or_default());
 
-                if let Some(ref c) = continuation {
-                    request.insert_headers(c);
-                }
+                let continuation = continuation.or(this.continuation);
+                request.insert_headers(&continuation);
 
                 let response = this
                     .client

sdk/data_cosmos/src/operations/query_documents.rs

@@ -29,6 +29,7 @@ operation! {
     ?parallelize_cross_partition_query: ParallelizeCrossPartition,
     ?query_cross_partition: QueryCrossPartition,
     ?partition_range_id: PartitionRangeId,
+    ?continuation: Continuation,
     #[skip]
     partition_key_serialized: String
 }
@@ -85,9 +86,8 @@ impl QueryDocumentsBuilder {
                     );
                 }
 
-                if let Some(ref c) = continuation {
-                    request.insert_headers(c);
-                }
+                let continuation = continuation.or(this.continuation);
+                request.insert_headers(&continuation);
 
                 let response = this
                     .client