azure_sdk_for_rust 0.12.0

Rust wrappers around Microsoft Azure REST APIs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

/* Table batch support
* Current limitation:
1. Only support single changeset in a batch request
2. Only allow PUT and GET in changeset
*/
use super::entity_path;
use serde::Serialize;
use serde_json;

const BATCH_BEGIN: &str = r#"--batch_a1e9d677-b28b-435e-a89e-87e6a768a431
Content-Type: multipart/mixed; boundary=changeset_8a28b620-b4bb-458c-a177-0959fb14c977

"#;
const BATCH_END: &str = "--batch_a1e9d677-b28b-435e-a89e-87e6a768a431\n";
const CHANGESET_BEGIN: &str = r#"--changeset_8a28b620-b4bb-458c-a177-0959fb14c977
Content-Type: application/http
Content-Transfer-Encoding: binary

"#;
const CHANGESET_END: &str = "--changeset_8a28b620-b4bb-458c-a177-0959fb14c977--\n";
const UPDATE_HEADER: &str = "Content-Type: application/json\n";
const ACCEPT_HEADER: &str = "Accept: application/json;odata=nometadata\n";
const IF_MATCH_HEADER: &str = "If-Match: *\n";

// RowKey, Payload. Payload None for deletion
pub struct BatchItem<T: Serialize>(String, Option<T>);

impl<T: Serialize> BatchItem<T> {
    pub fn new(row_key: String, value: Option<T>) -> Self {
        BatchItem(row_key, value)
    }
}

pub fn generate_batch_payload<T: Serialize>(uri_prefix: &str, table: &str, primary_key: &str, items: &[BatchItem<T>]) -> String {
    let mut payload: String = BATCH_BEGIN.to_owned();
    for item in items {
        payload.push_str(CHANGESET_BEGIN);
        payload.push_str(if item.1.is_some() { "PUT" } else { "DELETE" });
        payload.push_str(" ");
        payload.push_str(uri_prefix);
        payload.push_str(entity_path(table, primary_key, item.0.as_str()).as_str());
        payload.push_str(" HTTP/1.1\n");
        payload.push_str(ACCEPT_HEADER);
        if let Some(ref v) = item.1 {
            payload.push_str(UPDATE_HEADER);
            payload.push_str("\n");
            payload.push_str(serde_json::to_string(v).unwrap().as_str());
        } else {
            payload.push_str(IF_MATCH_HEADER);
        }

        payload.push_str("\n");
    }
    payload + CHANGESET_END + BATCH_END
}

#[cfg(test)]
mod test {
    use super::*;

    #[allow(non_snake_case)]
    #[derive(Serialize)]
    struct Entity {
        PartitionKey: String,
        RowKey: String,
        Rating: i32,
        Text: String,
    }

    #[test]
    fn verify_batch_payload() {
        let expected = r#"--batch_a1e9d677-b28b-435e-a89e-87e6a768a431
Content-Type: multipart/mixed; boundary=changeset_8a28b620-b4bb-458c-a177-0959fb14c977

--changeset_8a28b620-b4bb-458c-a177-0959fb14c977
Content-Type: application/http
Content-Transfer-Encoding: binary

PUT https://myaccount.table.core.windows.net/Blogs(PartitionKey='Channel_17',RowKey='3') HTTP/1.1
Accept: application/json;odata=nometadata
Content-Type: application/json

{"PartitionKey":"Channel_17","RowKey":"3","Rating":9,"Text":".NET..."}
--changeset_8a28b620-b4bb-458c-a177-0959fb14c977
Content-Type: application/http
Content-Transfer-Encoding: binary

PUT https://myaccount.table.core.windows.net/Blogs(PartitionKey='Channel_17',RowKey='3') HTTP/1.1
Accept: application/json;odata=nometadata
Content-Type: application/json

{"PartitionKey":"Channel_17","RowKey":"3","Rating":9,"Text":"PDC 2008..."}
--changeset_8a28b620-b4bb-458c-a177-0959fb14c977
Content-Type: application/http
Content-Transfer-Encoding: binary

DELETE https://myaccount.table.core.windows.net/Blogs(PartitionKey='Channel_17',RowKey='3') HTTP/1.1
Accept: application/json;odata=nometadata
If-Match: *

--changeset_8a28b620-b4bb-458c-a177-0959fb14c977--
--batch_a1e9d677-b28b-435e-a89e-87e6a768a431
"#;

        let items = vec![
            bupdate("Channel_17", "3", 9, ".NET..."),
            bupdate("Channel_17", "3", 9, "PDC 2008..."),
            bdelete("3"),
        ];
        let actual = generate_batch_payload("https://myaccount.table.core.windows.net/", "Blogs", "Channel_17", items.as_slice());
        assert_eq!(expected, actual);
    }

    fn bupdate(pk: &str, rk: &str, rating: i32, text: &str) -> BatchItem<Entity> {
        BatchItem(
            rk.to_owned(),
            Some(Entity {
                PartitionKey: pk.to_owned(),
                RowKey: rk.to_owned(),
                Rating: rating,
                Text: text.to_owned(),
            }),
        )
    }

    fn bdelete(rk: &str) -> BatchItem<Entity> {
        BatchItem(rk.to_owned(), None)
    }
}