deez 0.3.0

abstraction for DynamoDB
Documentation 
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

/// Convenience macro for batch write operations.
/// 
/// # Examples
/// 
/// ```
/// batch_write!(
///     client;
///     writes:
///         Task {
///             task_id: Some("87cb64a9-6431-406f-89d8-e91cb7ea944b".to_string()),
///             project: Some("foo_project".to_string()),
///             employee: Some("Mark".to_string()),
///             ..Default::default()
///         };
///     deletes:
///         Task {
///             task_id: Some("d9bb6b19-e306-4705-9773-960abe6c5078".to_string()),
///             project: Some("foo_project".to_string()),
///             employee: Some("Jimbo".to_string()),
///             ..Default::default()
///         },
///         Task {
///             task_id: Some("8ddbddaf-2072-4da1-b7c4-04aa31006b41".to_string()),
///             project: Some("foo_project".to_string()),
///             employee: Some("Steve".to_string()),
///             ..Default::default()
///         }
/// )?;
/// ```
#[macro_export]
macro_rules! batch_write {
    (
        $client:expr;
        writes: $( $w:expr ),*;
        deletes: $( $d:expr ),*
    ) => {{
        let mut m: HashMap<String, Vec<WriteRequest>> = HashMap::new();

        $({
            let w = $w;
            let table_name = w.table__name();
            let wr = WriteRequest::builder()
                .put_request(PutRequest::builder().set_item(Some(w.into())).build())
                .build();
            if let Some(x) = m.get_mut(&table_name) {
                x.push(wr);
            } else {
                m.insert(table_name, vec![wr]);
            }
        })*

        $({
            let d = $d;
            let keys = d.primary_keys();
            let dr = WriteRequest::builder()
                .delete_request(
                    DeleteRequest::builder()
                        .key(keys.hash.field(), keys.hash.av())
                        .key(keys.range.field(), keys.range.av())
                        .build(),
                )
                .build();
            if let Some(x) = m.get_mut(&d.table__name()) {
                x.push(dr);
            } else {
                m.insert(d.table__name(), vec![dr]);
            }
        })*

        $client
            .batch_write_item()
            .set_request_items(Some(m))
            .send()
            .await
    }};
}