dynoxide-rs 0.10.0

A lightweight, embeddable DynamoDB emulator backed by SQLite
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
77
78
79
80
81
82
83
84
85

use crate::errors::Result;
use crate::storage_backend::StorageBackend;
use crate::streams;
use serde::{Deserialize, Serialize};

#[derive(Debug, Default, Deserialize)]
pub struct ListStreamsRequest {
    #[serde(rename = "TableName", default)]
    pub table_name: Option<String>,
    #[serde(rename = "ExclusiveStartStreamArn", default)]
    pub exclusive_start_stream_arn: Option<String>,
    #[serde(rename = "Limit", default)]
    pub limit: Option<usize>,
}

#[derive(Debug, Default, Serialize)]
pub struct ListStreamsResponse {
    #[serde(rename = "Streams")]
    pub streams: Vec<StreamSummary>,
    #[serde(
        rename = "LastEvaluatedStreamArn",
        skip_serializing_if = "Option::is_none"
    )]
    pub last_evaluated_stream_arn: Option<String>,
}

#[derive(Debug, Default, Serialize)]
pub struct StreamSummary {
    #[serde(rename = "StreamArn")]
    pub stream_arn: String,
    #[serde(rename = "TableName")]
    pub table_name: String,
    #[serde(rename = "StreamLabel")]
    pub stream_label: String,
}

pub async fn execute<S: StorageBackend>(
    storage: &S,
    request: ListStreamsRequest,
) -> Result<ListStreamsResponse> {
    let tables = storage.list_stream_enabled_tables().await?;

    let mut summaries: Vec<StreamSummary> = tables
        .into_iter()
        .filter(|meta| {
            if let Some(ref filter_table) = request.table_name {
                &meta.table_name == filter_table
            } else {
                true
            }
        })
        .map(|meta| {
            let label = meta.stream_label.unwrap_or_default();
            StreamSummary {
                stream_arn: streams::stream_arn(&meta.table_name, &label),
                table_name: meta.table_name,
                stream_label: label,
            }
        })
        .collect();

    // Apply ExclusiveStartStreamArn pagination
    if let Some(ref start_arn) = request.exclusive_start_stream_arn {
        if let Some(pos) = summaries.iter().position(|s| &s.stream_arn == start_arn) {
            summaries = summaries.split_off(pos + 1);
        }
    }

    // Apply limit
    let last_arn = if let Some(limit) = request.limit {
        if summaries.len() > limit {
            summaries.truncate(limit);
            summaries.last().map(|s| s.stream_arn.clone())
        } else {
            None
        }
    } else {
        None
    };

    Ok(ListStreamsResponse {
        streams: summaries,
        last_evaluated_stream_arn: last_arn,
    })
}