DeleteDocument::into_future

Author: rylev

sdk/data_cosmos/examples/document_entries_00.rs

@@ -173,10 +173,8 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
         client
             .clone()
             .into_document_client(id.clone(), &id)?
-            .delete_document(
-                Context::new(),
-                DeleteDocumentOptions::new().consistency_level(&response),
-            )
+            .delete_document()
+            .into_future()
             .await?;
     }
     println!("Cleaned up");

sdk/data_cosmos/examples/readme.rs

@@ -1,4 +1,3 @@
-use azure_core::Context;
 use serde::{Deserialize, Serialize};
 // Using the prelude module of the Cosmos crate makes easier to use the Rust Azure SDK for Cosmos.
 use azure_data_cosmos::prelude::*;
@@ -147,12 +146,10 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
         collection_client
             .clone()
             .into_document_client(document.result.id.clone(), &document.result.a_number)?
-            .delete_document(
-                Context::new(),
-                DeleteDocumentOptions::new()
-                    .consistency_level(session_token.clone())
-                    .if_match_condition(&document.document_attributes),
-            )
+            .delete_document()
+            .consistency_level(session_token.clone())
+            .if_match_condition(&document.document_attributes)
+            .into_future()
             .await?;
     }
 

sdk/data_cosmos/examples/remove_all_documents.rs

@@ -1,4 +1,3 @@
-use azure_core::prelude::*;
 use azure_data_cosmos::prelude::*;
 use futures::stream::StreamExt;
 use serde_json::Value;
@@ -74,7 +73,8 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
         client
             .clone()
             .into_document_client(id.clone(), &partition_key)?
-            .delete_document(Context::new(), DeleteDocumentOptions::new())
+            .delete_document()
+            .into_future()
             .await?;
     }
 

sdk/data_cosmos/src/clients/document_client.rs

@@ -101,22 +101,8 @@ impl DocumentClient {
     }
 
     /// Delete a document
-    pub async fn delete_document(
-        &self,
-        ctx: Context,
-        options: DeleteDocumentOptions,
-    ) -> crate::Result<DeleteDocumentResponse> {
-        let mut request = self.prepare_request_pipeline_with_document_name(http::Method::DELETE);
-
-        options.decorate_request(&mut request, self.partition_key_serialized())?;
-
-        let response = self
-            .cosmos_client()
-            .pipeline()
-            .send(ctx.clone().insert(ResourceType::Documents), &mut request)
-            .await?;
-
-        DeleteDocumentResponse::try_from(response).await
+    pub fn delete_document(&self) -> DeleteDocumentBuilder {
+        DeleteDocumentBuilder::new(self.clone())
     }
 
     /// List all attachments for a document
@@ -132,7 +118,10 @@ impl DocumentClient {
         AttachmentClient::new(self, attachment_name)
     }
 
-    fn prepare_request_pipeline_with_document_name(&self, method: http::Method) -> Request {
+    pub(crate) fn prepare_request_pipeline_with_document_name(
+        &self,
+        method: http::Method,
+    ) -> Request {
         self.cosmos_client().prepare_request_pipeline(
             &format!(
                 "dbs/{}/colls/{}/docs/{}",

sdk/data_cosmos/src/operations/delete_document.rs

@@ -1,24 +1,30 @@
+use crate::headers::from_headers::*;
 use crate::prelude::*;
 
+use azure_core::headers::session_token_from_headers;
 use azure_core::prelude::*;
-use azure_core::{Request as HttpRequest, Response as HttpResponse};
+use azure_core::Response as HttpResponse;
 use chrono::{DateTime, Utc};
 
 #[derive(Debug, Clone)]
-pub struct DeleteDocumentOptions {
+pub struct DeleteDocumentBuilder {
+    client: DocumentClient,
     if_match_condition: Option<IfMatchCondition>,
     if_modified_since: Option<IfModifiedSince>,
     consistency_level: Option<ConsistencyLevel>,
     allow_tentative_writes: TentativeWritesAllowance,
+    context: Context,
 }
 
-impl DeleteDocumentOptions {
-    pub fn new() -> DeleteDocumentOptions {
+impl DeleteDocumentBuilder {
+    pub(crate) fn new(client: DocumentClient) -> DeleteDocumentBuilder {
         Self {
+            client,
             if_match_condition: None,
             if_modified_since: None,
             consistency_level: None,
             allow_tentative_writes: TentativeWritesAllowance::Deny,
+            context: Context::new(),
         }
     }
 
@@ -27,29 +33,50 @@ impl DeleteDocumentOptions {
         if_match_condition: IfMatchCondition => Some(if_match_condition),
         allow_tentative_writes: TentativeWritesAllowance,
         if_modified_since: DateTime<Utc> => Some(IfModifiedSince::new(if_modified_since)),
+        context: Context => context,
     }
 
-    pub fn decorate_request(
-        &self,
-        request: &mut HttpRequest,
-        serialized_partition_key: &str,
-    ) -> crate::Result<()> {
-        azure_core::headers::add_optional_header2(&self.if_match_condition, request)?;
-        azure_core::headers::add_optional_header2(&self.if_modified_since, request)?;
-        azure_core::headers::add_optional_header2(&self.consistency_level, request)?;
-        azure_core::headers::add_mandatory_header2(&self.allow_tentative_writes, request)?;
+    pub fn into_future(self) -> DeleteDocument {
+        Box::pin(async move {
+            let mut request = self
+                .client
+                .prepare_request_pipeline_with_document_name(http::Method::DELETE);
+
+            azure_core::headers::add_optional_header2(&self.if_match_condition, &mut request)?;
+            azure_core::headers::add_optional_header2(&self.if_modified_since, &mut request)?;
+            azure_core::headers::add_optional_header2(&self.consistency_level, &mut request)?;
+            azure_core::headers::add_mandatory_header2(&self.allow_tentative_writes, &mut request)?;
 
-        crate::cosmos_entity::add_as_partition_key_header_serialized2(
-            serialized_partition_key,
-            request,
-        );
+            crate::cosmos_entity::add_as_partition_key_header_serialized2(
+                &self.client.partition_key_serialized(),
+                &mut request,
+            );
 
-        Ok(())
+            let response = self
+                .client
+                .cosmos_client()
+                .pipeline()
+                .send(
+                    self.context.clone().insert(ResourceType::Documents),
+                    &mut request,
+                )
+                .await?;
+
+            DeleteDocumentResponse::try_from(response).await
+        })
     }
 }
 
-use crate::headers::from_headers::*;
-use azure_core::headers::session_token_from_headers;
+type DeleteDocument = futures::future::BoxFuture<'static, crate::Result<DeleteDocumentResponse>>;
+
+#[cfg(feature = "into_future")]
+impl std::future::IntoFuture for DeleteDocumentBuilder {
+    type Future = DeleteDocument;
+    type Output = <DeleteDocument as std::future::Future>::Output;
+    fn into_future(self) -> Self::Future {
+        Self::into_future(self)
+    }
+}
 
 #[derive(Debug, Clone)]
 pub struct DeleteDocumentResponse {