DeleteDatabase::into_future

Author: rylev

sdk/data_cosmos/examples/create_delete_database.rs

@@ -76,7 +76,8 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
 
     let resp = client
         .into_database_client(database_name)
-        .delete_database(Context::new(), DeleteDatabaseOptions::new())
+        .delete_database()
+        .into_future()
         .await?;
     println!("database deleted. resp == {:#?}", resp);
 

sdk/data_cosmos/examples/document_00.rs

@@ -196,7 +196,8 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
     // And then we delete the database.
     client
         .into_database_client(database.id)
-        .delete_database(Context::new(), DeleteDatabaseOptions::new())
+        .delete_database()
+        .into_future()
         .await?;
     println!("database deleted");
 

sdk/data_cosmos/src/clients/database_client.rs

@@ -77,23 +77,8 @@ impl DatabaseClient {
     }
 
     /// Delete the database
-    pub async fn delete_database(
-        &self,
-        ctx: Context,
-        options: DeleteDatabaseOptions,
-    ) -> crate::Result<DeleteDatabaseResponse> {
-        let mut request = self.cosmos_client().prepare_request_pipeline(
-            &format!("dbs/{}", self.database_name()),
-            http::Method::DELETE,
-        );
-
-        options.decorate_request(&mut request)?;
-        let response = self
-            .pipeline()
-            .send(ctx.clone().insert(ResourceType::Databases), &mut request)
-            .await?;
-
-        Ok(DeleteDatabaseResponse::try_from(response).await?)
+    pub fn delete_database(&self) -> DeleteDatabaseBuilder {
+        DeleteDatabaseBuilder::new(self.clone())
     }
 
     /// List collections in the database

sdk/data_cosmos/src/operations/create_collection.rs

@@ -37,6 +37,7 @@ impl CreateCollectionBuilder {
         consistency_level: ConsistencyLevel => Some(consistency_level),
         indexing_policy: IndexingPolicy => Some(indexing_policy),
         offer: Offer => Some(offer),
+        context: Context => context,
     }
 
     pub fn into_future(self) -> CreateCollection {

sdk/data_cosmos/src/operations/create_document.rs

@@ -49,6 +49,7 @@ impl<D: Serialize + CosmosEntity + Send + 'static> CreateDocumentBuilder<D> {
         allow_tentative_writes: TentativeWritesAllowance,
         is_upsert: bool => if is_upsert { IsUpsert::Yes } else { IsUpsert::No },
         indexing_directive: IndexingDirective,
+        context: Context => context,
     }
 
     pub fn partition_key<PK: Serialize>(
@@ -95,7 +96,9 @@ impl<D: Serialize + CosmosEntity + Send + 'static> CreateDocumentBuilder<D> {
 type CreateDocument = futures::future::BoxFuture<'static, crate::Result<CreateDocumentResponse>>;
 
 #[cfg(feature = "into_future")]
-impl <D: Serialize + CosmosEntity + Send + 'static> std::future::IntoFuture for CreateDocumentBuilder<D> {
+impl<D: Serialize + CosmosEntity + Send + 'static> std::future::IntoFuture
+    for CreateDocumentBuilder<D>
+{
     type Future = CreateDocument;
     type Output = <CreateDocument as std::future::Future>::Output;
     fn into_future(self) -> Self::Future {

sdk/data_cosmos/src/operations/create_permission.rs

@@ -26,6 +26,7 @@ impl CreatePermissionBuilder {
     setters! {
         expiry_seconds: u64 => Some(ExpirySeconds::new(expiry_seconds)),
         consistency_level: ConsistencyLevel => Some(consistency_level),
+        context: Context => context,
     }
 
     pub fn into_future(self) -> CreatePermission {

sdk/data_cosmos/src/operations/create_user.rs

@@ -19,6 +19,7 @@ impl CreateUserBuilder {
 
     setters! {
         consistency_level: ConsistencyLevel => Some(consistency_level),
+        context: Context => context,
     }
 
     pub fn into_future(self) -> CreateUser {

sdk/data_cosmos/src/operations/delete_collection.rs

@@ -22,6 +22,7 @@ impl DeleteCollectionBuilder {
 
     setters! {
         consistency_level: ConsistencyLevel => Some(consistency_level),
+        context: Context => context,
     }
 
     pub fn into_future(self) -> DeleteCollection {

sdk/data_cosmos/src/operations/delete_database.rs

@@ -2,28 +2,60 @@ use crate::headers::from_headers::*;
 use crate::prelude::*;
 use crate::ResourceQuota;
 use azure_core::headers::session_token_from_headers;
-use azure_core::Request as HttpRequest;
+use azure_core::Context;
 use azure_core::Response as HttpResponse;
 
 #[derive(Debug, Clone)]
-pub struct DeleteDatabaseOptions {
+pub struct DeleteDatabaseBuilder {
+    client: DatabaseClient,
     consistency_level: Option<ConsistencyLevel>,
+    context: Context,
 }
 
-impl DeleteDatabaseOptions {
-    pub fn new() -> Self {
+impl DeleteDatabaseBuilder {
+    pub(crate) fn new(client: DatabaseClient) -> Self {
         Self {
+            client,
             consistency_level: None,
+            context: Context::new(),
         }
     }
 
     setters! {
         consistency_level: ConsistencyLevel => Some(consistency_level),
+        context: Context => context,
     }
 
-    pub fn decorate_request(&self, request: &mut HttpRequest) -> crate::Result<()> {
-        azure_core::headers::add_optional_header2(&self.consistency_level, request)?;
-        Ok(())
+    pub fn into_future(self) -> DeleteDatabase {
+        Box::pin(async move {
+            let mut request = self.client.cosmos_client().prepare_request_pipeline(
+                &format!("dbs/{}", self.client.database_name()),
+                http::Method::DELETE,
+            );
+
+            azure_core::headers::add_optional_header2(&self.consistency_level, &mut request)?;
+            let response = self
+                .client
+                .pipeline()
+                .send(
+                    self.context.clone().insert(ResourceType::Databases),
+                    &mut request,
+                )
+                .await?;
+
+            Ok(DeleteDatabaseResponse::try_from(response).await?)
+        })
+    }
+}
+
+type DeleteDatabase = futures::future::BoxFuture<'static, crate::Result<DeleteDatabaseResponse>>;
+
+#[cfg(feature = "into_future")]
+impl std::future::IntoFuture for DeleteDatabaseBuilder {
+    type Future = DeleteDatabase;
+    type Output = <DeleteDatabase as std::future::Future>::Output;
+    fn into_future(self) -> Self::Future {
+        Self::into_future(self)
     }
 }
 

sdk/data_cosmos/tests/cosmos_document.rs

@@ -1,7 +1,7 @@
 #![cfg(all(test, feature = "test_e2e"))]
 use azure_core::Context;
 use azure_data_cosmos::prelude::{
-    CreateDocumentBuilder, DeleteDatabaseOptions, GetDocumentOptions,
+    CreateDocumentBuilder, DeleteDatabaseBuilder, GetDocumentOptions,
 };
 use serde::{Deserialize, Serialize};