Merge pull request #283 from MindFlavor/issue/278

Implemented stream for ListAttachmentsBuilder

Author: Francesco Cogno

README.md

@@ -5,7 +5,7 @@
 
 [![Build Status](https://travis-ci.org/MindFlavor/AzureSDKForRust.svg?branch=master)](https://travis-ci.org/MindFlavor/AzureSDKForRust) [![Coverage Status](https://coveralls.io/repos/MindFlavor/AzureSDKForRust/badge.svg?branch=master&service=github)](https://coveralls.io/github/MindFlavor/AzureSDKForRust?branch=master) ![stability-unstable](https://img.shields.io/badge/stability-unstable-yellow.svg)
 
-[![tag](https://img.shields.io/github/tag/mindflavor/AzureSDKForRust.svg)](https://github.com/MindFlavor/AzureSDKForRust/tree/cosmos_0.100.0) [![release](https://img.shields.io/github/release/mindflavor/AzureSDKForRust.svg)](https://github.com/MindFlavor/AzureSDKForRust/releases/tag/cosmos_0.100.0) [![commitssince](https://img.shields.io/github/commits-since/mindflavor/AzureSDKForRust/cosmos_0.100.0)](https://github.com/MindFlavor/AzureSDKForRust/commits/master)
+[![tag](https://img.shields.io/github/tag/mindflavor/AzureSDKForRust.svg)](https://github.com/MindFlavor/AzureSDKForRust/tree/cosmos_0.100.1) [![release](https://img.shields.io/github/release/mindflavor/AzureSDKForRust.svg)](https://github.com/MindFlavor/AzureSDKForRust/releases/tag/cosmos_0.100.1) [![commitssince](https://img.shields.io/github/commits-since/mindflavor/AzureSDKForRust/cosmos_0.100.1)](https://github.com/MindFlavor/AzureSDKForRust/commits/master)
 
 [![GitHub contributors](https://img.shields.io/github/contributors/MindFlavor/AzureSDKForRust.svg)](https://github.com/MindFlavor/AzureSDKForRust/graphs/contributors)
 

azure_sdk_cosmos/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name          = "azure_sdk_cosmos"
-version       = "0.100.0"
+version       = "0.100.1"
 description   = "Rust wrappers around Microsoft Azure REST APIs - Azure Cosmos DB crate"
 readme        = "README.md"
 authors       = ["Francesco Cogno <francesco.cogno@outlook.com>", "Max Gortman <mgortman@microsoft.com>"]

azure_sdk_cosmos/src/requests/list_attachments_builder.rs

@@ -4,10 +4,11 @@ use crate::DocumentClientRequired;
 use crate::{DocumentClient, ResourceType};
 use azure_sdk_core::errors::{check_status_extract_headers_and_body, AzureError};
 use azure_sdk_core::prelude::*;
+use futures::stream::{unfold, Stream};
 use hyper::StatusCode;
 use std::convert::TryInto;
 
-#[derive(Debug, Clone)]
+#[derive(Debug)]
 pub struct ListAttachmentsBuilder<'a, 'b, C, D, COLL>
 where
     C: CosmosClient,
@@ -47,6 +48,26 @@ where
     }
 }
 
+impl<'a, 'b, C, D, COLL> Clone for ListAttachmentsBuilder<'a, 'b, C, D, COLL>
+where
+    C: CosmosClient,
+    D: DatabaseClient<C>,
+    COLL: CollectionClient<C, D>,
+{
+    fn clone(&self) -> Self {
+        Self {
+            document_client: self.document_client,
+            if_match_condition: self.if_match_condition,
+            user_agent: self.user_agent,
+            activity_id: self.activity_id,
+            consistency_level: self.consistency_level.clone(),
+            continuation: self.continuation,
+            max_item_count: self.max_item_count,
+            a_im: self.a_im,
+        }
+    }
+}
+
 impl<'a, 'b, C, D, COLL> DocumentClientRequired<'a, C, D, COLL>
     for ListAttachmentsBuilder<'a, 'b, C, D, COLL>
 where
@@ -351,4 +372,45 @@ where
 
         Ok((&headers, &whole_body as &[u8]).try_into()?)
     }
+
+    pub fn stream(&self) -> impl Stream<Item = Result<ListAttachmentsResponse, AzureError>> + '_ {
+        #[derive(Debug, Clone, PartialEq)]
+        enum States {
+            Init,
+            Continuation(String),
+        };
+
+        unfold(
+            Some(States::Init),
+            move |continuation_token: Option<States>| {
+                async move {
+                    debug!("continuation_token == {:?}", &continuation_token);
+                    let response = match continuation_token {
+                        Some(States::Init) => self.execute().await,
+                        Some(States::Continuation(continuation_token)) => {
+                            self.clone()
+                                .with_continuation(&continuation_token)
+                                .execute()
+                                .await
+                        }
+                        None => return None,
+                    };
+
+                    // the ? operator does not work in async move (yet?)
+                    // so we have to resort to this boilerplate
+                    let response = match response {
+                        Ok(response) => response,
+                        Err(err) => return Some((Err(err), None)),
+                    };
+
+                    let continuation_token = match &response.continuation_token {
+                        Some(ct) => Some(States::Continuation(ct.to_owned())),
+                        None => None,
+                    };
+
+                    Some((Ok(response), continuation_token))
+                }
+            },
+        )
+    }
 }

azure_sdk_cosmos/src/responses/list_attachments_response.rs

@@ -1,7 +1,9 @@
 use crate::from_headers::*;
 use crate::{Attachment, ResourceQuota};
 use azure_sdk_core::errors::AzureError;
-use azure_sdk_core::{session_token_from_headers, SessionToken};
+use azure_sdk_core::{
+    continuation_token_from_headers_optional, session_token_from_headers, SessionToken,
+};
 use chrono::{DateTime, Utc};
 use hyper::header::HeaderMap;
 
@@ -40,6 +42,7 @@ pub struct ListAttachmentsResponse {
     pub activity_id: uuid::Uuid,
     pub gateway_version: String,
     pub date: DateTime<Utc>,
+    pub continuation_token: Option<String>,
 }
 
 impl std::convert::TryFrom<(&HeaderMap, &[u8])> for ListAttachmentsResponse {
@@ -77,6 +80,7 @@ impl std::convert::TryFrom<(&HeaderMap, &[u8])> for ListAttachmentsResponse {
             activity_id: activity_id_from_headers(headers)?,
             gateway_version: gateway_version_from_headers(headers)?.to_owned(),
             date: date_from_headers(headers)?,
+            continuation_token: continuation_token_from_headers_optional(headers)?,
         })
     }
 }