Cosmos/stored procedure (#238)

Implemented stored procedure functions

Author: Francesco Cogno

Cargo.toml.bak

@@ -5,15 +5,15 @@
 
 [![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/0.40.0) [![release](https://img.shields.io/github/release/mindflavor/AzureSDKForRust.svg)](https://github.com/MindFlavor/AzureSDKForRust/releases/tag/0.40.0) [![commitssince](https://img.shields.io/github/commits-since/mindflavor/AzureSDKForRust/0.40.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.41.0) [![release](https://img.shields.io/github/release/mindflavor/AzureSDKForRust.svg)](https://github.com/MindFlavor/AzureSDKForRust/releases/tag/cosmos_0.41.0) [![commitssince](https://img.shields.io/github/commits-since/mindflavor/AzureSDKForRust/cosmos_0.41.0)](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)
 
 Crate | Docs | Crates.io | Downloads | Downloads@Latest |
 -- | -- | -- | -- | -- |
 [azure_sdk_auth_aad](https://github.com/MindFlavor/AzureSDKForRust/tree/master/azure_sdk_auth_aad) | [![docs](https://docs.rs/azure_sdk_auth_aad/badge.svg)](https://docs.rs/azure_sdk_auth_aad/0.40.0/azure_sdk_auth_aad) | [![Crate](https://img.shields.io/crates/v/azure_sdk_auth_aad.svg)](https://crates.io/crates/azure_sdk_auth_aad) | [![cratedown](https://img.shields.io/crates/d/azure_sdk_auth_aad.svg)](https://crates.io/crates/azure_sdk_auth_aad) | [![cratelastdown](https://img.shields.io/crates/dv/azure_sdk_auth_aad.svg)](https://crates.io/crates/azure_sdk_auth_aad)
 [azure_sdk_core](https://github.com/MindFlavor/AzureSDKForRust/tree/master/azure_sdk_core) | [![docs](https://docs.rs/azure_sdk_core/badge.svg)](https://docs.rs/azure_sdk_core/0.40.0/azure_sdk_core) | [![Crate](https://img.shields.io/crates/v/azure_sdk_core.svg)](https://crates.io/crates/azure_sdk_core) | [![cratedown](https://img.shields.io/crates/d/azure_sdk_core.svg)](https://crates.io/crates/azure_sdk_core) | [![cratelastdown](https://img.shields.io/crates/dv/azure_sdk_core.svg)](https://crates.io/crates/azure_sdk_core)
-[azure_sdk_cosmos](https://github.com/MindFlavor/AzureSDKForRust/tree/master/azure_sdk_cosmos) | [![docs](https://docs.rs/azure_sdk_cosmos/badge.svg)](https://docs.rs/azure_sdk_cosmos/0.40.0/azure_sdk_cosmos) | [![Crate](https://img.shields.io/crates/v/azure_sdk_cosmos.svg)](https://crates.io/crates/azure_sdk_cosmos) | [![cratedown](https://img.shields.io/crates/d/azure_sdk_cosmos.svg)](https://crates.io/crates/azure_sdk_cosmos) | [![cratelastdown](https://img.shields.io/crates/dv/azure_sdk_cosmos.svg)](https://crates.io/crates/azure_sdk_cosmos)
+[azure_sdk_cosmos](https://github.com/MindFlavor/AzureSDKForRust/tree/master/azure_sdk_cosmos) | [![docs](https://docs.rs/azure_sdk_cosmos/badge.svg)](https://docs.rs/azure_sdk_cosmos/0.41.0/azure_sdk_cosmos) | [![Crate](https://img.shields.io/crates/v/azure_sdk_cosmos.svg)](https://crates.io/crates/azure_sdk_cosmos) | [![cratedown](https://img.shields.io/crates/d/azure_sdk_cosmos.svg)](https://crates.io/crates/azure_sdk_cosmos) | [![cratelastdown](https://img.shields.io/crates/dv/azure_sdk_cosmos.svg)](https://crates.io/crates/azure_sdk_cosmos)
 [azure_sdk_service_bus](https://github.com/MindFlavor/AzureSDKForRust/tree/master/azure_sdk_service_bus) | [![docs](https://docs.rs/azure_sdk_service_bus/badge.svg)](https://docs.rs/azure_sdk_service_bus/0.40.0/azure_sdk_service_bus) | [![Crate](https://img.shields.io/crates/v/azure_sdk_service_bus.svg)](https://crates.io/crates/azure_sdk_service_bus) | [![cratedown](https://img.shields.io/crates/d/azure_sdk_service_bus.svg)](https://crates.io/crates/azure_sdk_service_bus) | [![cratelastdown](https://img.shields.io/crates/dv/azure_sdk_service_bus.svg)](https://crates.io/crates/azure_sdk_service_bus)
 [azure_sdk_storage_account](https://github.com/MindFlavor/AzureSDKForRust/tree/master/azure_sdk_storage_account) | [![docs](https://docs.rs/azure_sdk_storage_account/badge.svg)](https://docs.rs/azure_sdk_storage_account/0.40.0/azure_sdk_storage_account) | [![Crate](https://img.shields.io/crates/v/azure_sdk_storage_account.svg)](https://crates.io/crates/azure_sdk_storage_account) | [![cratedown](https://img.shields.io/crates/d/azure_sdk_storage_account.svg)](https://crates.io/crates/azure_sdk_storage_account) | [![cratelastdown](https://img.shields.io/crates/dv/azure_sdk_storage_account.svg)](https://crates.io/crates/azure_sdk_storage_account)
 [azure_sdk_storage_blob](https://github.com/MindFlavor/AzureSDKForRust/tree/master/azure_sdk_storage_blob) | [![docs](https://docs.rs/azure_sdk_storage_blob/badge.svg)](https://docs.rs/azure_sdk_storage_blob/0.40.0/azure_sdk_storage_blob) | [![Crate](https://img.shields.io/crates/v/azure_sdk_storage_blob.svg)](https://crates.io/crates/azure_sdk_storage_blob) | [![cratedown](https://img.shields.io/crates/d/azure_sdk_storage_blob.svg)](https://crates.io/crates/azure_sdk_storage_blob) | [![cratelastdown](https://img.shields.io/crates/dv/azure_sdk_storage_blob.svg)](https://crates.io/crates/azure_sdk_storage_blob)

README.md

@@ -1,6 +1,6 @@
 [package]
 name          = "azure_sdk_cosmos"
-version       = "0.40.0"
+version       = "0.41.0"
 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/Cargo.toml

@@ -1,4 +1,5 @@
 use azure_sdk_cosmos::prelude::*;
+use futures::stream::StreamExt;
 use std::error::Error;
 
 #[tokio::main]
@@ -32,12 +33,15 @@ async fn main() -> Result<(), Box<dyn Error>> {
     // account. Database do not implement Display but deref to &str so you can pass it to methods
     // both as struct or id.
 
+    let list_databases_response = client.list_databases().execute().await?;
+    println!("list_databases_response = {:#?}", list_databases_response);
+
     let db = client
         .create_database()
         .with_database_name(&database_name)
         .execute()
         .await?;
-    println!("created database = {:?}", db);
+    println!("created database = {:#?}", db);
 
     // create collection!
     {
@@ -71,21 +75,32 @@ async fn main() -> Result<(), Box<dyn Error>> {
             .await?;
 
         println!(
-            "create_collection_response == {:?}",
+            "create_collection_response == {:#?}",
             create_collection_response
         );
 
         let db_collection = db_client.with_collection(&"panzadoro");
+
+        let get_collection_response = db_collection.get_collection().execute().await?;
+        println!("get_collection_response == {:#?}", get_collection_response);
+
+        let stream = db_client.list_collections();
+        let mut stream = Box::pin(stream.stream());
+        while let Some(res) = stream.next().await {
+            let res = res?;
+            println!("res == {:#?}", res);
+        }
+
         let delete_response = db_collection.delete_collection().execute().await?;
-        println!("collection deleted: {:?}", delete_response);
+        println!("collection deleted: {:#?}", delete_response);
     }
 
     let resp = client
         .with_database(&database_name)
         .delete_database()
         .execute()
         .await?;
-    println!("database deleted. resp == {:?}", resp);
+    println!("database deleted. resp == {:#?}", resp);
 
     Ok(())
 }

azure_sdk_cosmos/examples/create_delete_database.rs

@@ -4,6 +4,7 @@ extern crate serde_derive;
 // DB.
 use azure_sdk_core::prelude::*;
 use azure_sdk_cosmos::prelude::*;
+use azure_sdk_cosmos::responses::GetDocumentResponse;
 use std::borrow::Cow;
 use std::error::Error;
 
@@ -177,7 +178,7 @@ async fn main() -> Result<(), Box<dyn Error>> {
 
     // The document can be no longer there so the result is
     // an Option<Document<T>>
-    if let Some(document) = get_document_response.document {
+    if let GetDocumentResponse::Found(document) = get_document_response {
         // Now, for the sake of experimentation, we will update (replace) the
         // document created. We do this only if the original document has not been
         // modified in the meantime. This is called optimistic concurrency.
@@ -189,7 +190,7 @@ async fn main() -> Result<(), Box<dyn Error>> {
             .replace_document()
             .with_document(&doc)
             .with_partition_keys(&(&doc.document_attributes.id).into())
-            .with_if_match_condition(IfMatchCondition::Match(&document.document_attributes.etag))
+            .with_if_match_condition(IfMatchCondition::Match(&document.etag))
             .execute()
             .await?;
         println!(

azure_sdk_cosmos/examples/document00.rs

@@ -1,6 +1,7 @@
 use azure_sdk_core::modify_conditions::IfMatchCondition;
 use azure_sdk_core::prelude::*;
 use azure_sdk_cosmos::prelude::*;
+use azure_sdk_cosmos::responses::GetDocumentResponse;
 use futures::stream::StreamExt;
 use std::borrow::Cow;
 use std::error::Error;
@@ -74,12 +75,9 @@ async fn main() -> Result<(), Box<dyn Error>> {
 
     // we inserted 5 documents and retrieved the first 3.
     // continuation_token must be present
-    assert_eq!(
-        response.additional_headers.continuation_token.is_some(),
-        true
-    );
+    assert!(response.continuation_token.is_some());
 
-    let ct = response.additional_headers.continuation_token.unwrap();
+    let ct = response.continuation_token.unwrap();
     println!("ct == {}", ct);
 
     let response = client
@@ -93,10 +91,7 @@ async fn main() -> Result<(), Box<dyn Error>> {
 
     // we got the last 2 entries. Now continuation_token
     // must be absent
-    assert_eq!(
-        response.additional_headers.continuation_token.is_some(),
-        false
-    );
+    assert!(response.continuation_token.is_none());
 
     // we can have Rust pass the continuation_token for
     // us if we call the stream function. Here we
@@ -123,17 +118,24 @@ async fn main() -> Result<(), Box<dyn Error>> {
         .execute::<MySampleStruct>()
         .await?;
 
-    assert_eq!(response.document.is_some(), true);
+    assert!(match response {
+        GetDocumentResponse::Found(_) => true,
+        _ => false,
+    });
     println!("response == {:#?}", response);
-    let mut doc = response.clone().document.unwrap();
-    doc.document.a_string = "Something else here".into();
 
-    let etag = doc.document_attributes.etag().to_owned();
+    let mut doc = match response {
+        GetDocumentResponse::Found(ref resp) => resp.clone(),
+        GetDocumentResponse::NotFound(_) => panic!(),
+    };
+    doc.document.document.a_string = "Something else here".into();
+
+    let etag = doc.etag.to_owned();
 
     println!("\n\nReplacing document");
     let replace_document_response = client
         .replace_document()
-        .with_document(&doc)
+        .with_document(&doc.document)
         .with_partition_keys(PartitionKeys::new().push(&id)?)
         .with_consistency_level(ConsistencyLevel::from(&response))
         .with_if_match_condition(IfMatchCondition::Match(&etag)) // use optimistic concurrency check
@@ -157,8 +159,10 @@ async fn main() -> Result<(), Box<dyn Error>> {
         .execute::<MySampleStruct>()
         .await?;
 
-    assert_eq!(response.document.is_some(), false);
-    assert_eq!(response.has_been_found, false);
+    assert!(match response {
+        GetDocumentResponse::NotFound(_) => true,
+        _ => false,
+    });
     println!("response == {:#?}", response);
 
     for i in 0u64..5 {

azure_sdk_cosmos/examples/document_entries00.rs

@@ -0,0 +1,113 @@
+use azure_sdk_cosmos::prelude::*;
+use std::borrow::Cow;
+use std::error::Error;
+#[macro_use]
+extern crate serde_derive;
+
+#[derive(Serialize, Deserialize, Clone, Debug)]
+struct MySampleStruct<'a> {
+    a_string: Cow<'a, str>,
+    a_number: u64,
+    a_timestamp: i64,
+}
+
+// This example expects you to have created a collection
+// with partitionKey on "id".
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn Error>> {
+    let database_name = std::env::args()
+        .nth(1)
+        .expect("please specify database name as first command line parameter");
+    let collection_name = std::env::args()
+        .nth(2)
+        .expect("please specify collection name as second command line parameter");
+
+    let master_key =
+        std::env::var("COSMOS_MASTER_KEY").expect("Set env variable COSMOS_MASTER_KEY first!");
+    let account = std::env::var("COSMOS_ACCOUNT").expect("Set env variable COSMOS_ACCOUNT first!");
+
+    let authorization_token = AuthorizationToken::new_master(&master_key)?;
+
+    let client = ClientBuilder::new(account, authorization_token)?;
+    let client = client.with_database(&database_name);
+    let client = client.with_collection(&collection_name);
+
+    let mut doc = Document::new(
+        format!("unique_id{}", 500),
+        MySampleStruct {
+            a_string: Cow::Borrowed("Something here"),
+            a_number: 600,
+            a_timestamp: chrono::Utc::now().timestamp(),
+        },
+    );
+
+    let mut partition_keys = PartitionKeys::new();
+    partition_keys.push(doc.document_attributes.id())?;
+
+    // let's add an entity.
+    let create_document_response = client
+        .create_document()
+        .with_document(&doc)
+        .with_partition_keys(&partition_keys)
+        .with_is_upsert(true)
+        .execute()
+        .await?;
+
+    println!(
+        "create_document_response == {:#?}\n\n\n",
+        create_document_response
+    );
+
+    let document_client = client.with_document(&doc);
+
+    let get_document_response = document_client
+        .get_document()
+        .with_partition_keys(&partition_keys)
+        .execute::<serde_json::Value>()
+        .await?;
+    println!("get_document_response == {:#?}", get_document_response);
+
+    let document_client = client.with_document(&"ciccia");
+
+    let get_document_response = document_client
+        .get_document()
+        .with_partition_keys(&partition_keys)
+        .execute::<serde_json::Value>()
+        .await?;
+    println!(
+        "get_document_response == {:#?}\n\n\n",
+        get_document_response
+    );
+
+    let list_documents_response = client
+        .list_documents()
+        .execute::<serde_json::Value>()
+        .await?;
+    println!("list_documents_response == {:#?}", list_documents_response);
+
+    let query_documents_response = client
+        .query_documents()
+        .with_query(&("SELECT * FROM c WHERE c.a_number = 600".into()))
+        .with_query_cross_partition(true)
+        .execute::<serde_json::Value>()
+        .await?;
+    println!(
+        "query_documents_response == {:#?}",
+        query_documents_response
+    );
+
+    doc.document.a_number = 43;
+
+    let replace_document_response = client
+        .replace_document()
+        .with_document(&doc)
+        .with_partition_keys(&partition_keys)
+        .execute()
+        .await?;
+    println!(
+        "replace_document_response == {:#?}",
+        replace_document_response
+    );
+
+    Ok(())
+}