DeletePermission::into_future

Author: rylev

sdk/data_cosmos/examples/permission_00.rs

@@ -138,12 +138,11 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
     );
 
     let delete_permission_response = permission_client
-        .delete_permission(
-            Context::new(),
-            DeletePermissionOptions::new().consistency_level(ConsistencyLevel::Session(
-                replace_permission_response.session_token,
-            )),
-        )
+        .delete_permission()
+        .consistency_level(ConsistencyLevel::Session(
+            replace_permission_response.session_token,
+        ))
+        .into_future()
         .await
         .unwrap();
 

sdk/data_cosmos/examples/user_permission_token.rs

@@ -126,9 +126,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
         Err(error) => println!("Insert failed: {:#?}", error),
     }
 
-    permission_client
-        .delete_permission(Context::new(), DeletePermissionOptions::new())
-        .await?;
+    permission_client.delete_permission().into_future().await?;
 
     // All includes read and write.
     let permission_mode = get_collection_response.collection.all_permission();

sdk/data_cosmos/src/clients/permission_client.rs

@@ -86,21 +86,8 @@ impl PermissionClient {
     }
 
     /// Delete the permission
-    pub async fn delete_permission(
-        &self,
-        ctx: Context,
-        options: DeletePermissionOptions,
-    ) -> crate::Result<DeletePermissionResponse> {
-        let mut request = self.prepare_request_with_permission_name(http::Method::DELETE);
-
-        options.decorate_request(&mut request)?;
-
-        let response = self
-            .pipeline()
-            .send(ctx.clone().insert(ResourceType::Permissions), &mut request)
-            .await?;
-
-        Ok(DeletePermissionResponse::try_from(response).await?)
+    pub fn delete_permission(&self) -> DeletePermissionBuilder {
+        DeletePermissionBuilder::new(self.clone())
     }
 
     pub(crate) fn prepare_request_with_permission_name(&self, method: http::Method) -> Request {

sdk/data_cosmos/src/operations/delete_permission.rs

@@ -2,29 +2,61 @@ use crate::headers::from_headers::*;
 use crate::prelude::*;
 
 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 DeletePermissionOptions {
+pub struct DeletePermissionBuilder {
+    client: PermissionClient,
     consistency_level: Option<ConsistencyLevel>,
+    context: Context,
 }
 
-impl DeletePermissionOptions {
-    pub fn new() -> Self {
+impl DeletePermissionBuilder {
+    pub(crate) fn new(client: PermissionClient) -> Self {
         Self {
+            client,
             consistency_level: None,
+            context: Context::new(),
         }
     }
 
     setters! {
         consistency_level: ConsistencyLevel => Some(consistency_level),
+        context: Context => context,
     }
 
-    pub(crate) fn decorate_request(&self, request: &mut HttpRequest) -> crate::Result<()> {
-        azure_core::headers::add_optional_header2(&self.consistency_level, request)?;
+    pub fn into_future(self) -> DeletePermission {
+        Box::pin(async move {
+            let mut request = self
+                .client
+                .prepare_request_with_permission_name(http::Method::DELETE);
 
-        Ok(())
+            azure_core::headers::add_optional_header2(&self.consistency_level, &mut request)?;
+
+            let response = self
+                .client
+                .pipeline()
+                .send(
+                    self.context.clone().insert(ResourceType::Permissions),
+                    &mut request,
+                )
+                .await?;
+
+            Ok(DeletePermissionResponse::try_from(response).await?)
+        })
+    }
+}
+
+type DeletePermission =
+    futures::future::BoxFuture<'static, crate::Result<DeletePermissionResponse>>;
+
+#[cfg(feature = "into_future")]
+impl std::future::IntoFuture for DeletePermissionBuilder {
+    type Future = DeletePermission;
+    type Output = <DeletePermission as std::future::Future>::Output;
+    fn into_future(self) -> Self::Future {
+        Self::into_future(self)
     }
 }