ferrokinesis 0.1.0

A local AWS Kinesis mock server for testing, written in Rust
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

use crate::constants;
use crate::error::KinesisErrorResponse;
use crate::store::Store;
use serde_json::{Value, json};

pub async fn execute(store: &Store, data: Value) -> Result<Option<Value>, KinesisErrorResponse> {
    let stream_name = data[constants::STREAM_NAME].as_str().unwrap_or("");
    let stream = store.get_stream(stream_name).await?;

    let open_shard_count = stream
        .shards
        .iter()
        .filter(|s| s.sequence_number_range.ending_sequence_number.is_none())
        .count();

    let consumer_count = store
        .list_consumers_for_stream(&stream.stream_arn)
        .await
        .len();

    Ok(Some(json!({
        "StreamDescriptionSummary": {
            "RetentionPeriodHours": stream.retention_period_hours,
            "EnhancedMonitoring": stream.enhanced_monitoring,
            "EncryptionType": stream.encryption_type,
            "KeyId": stream.key_id,
            "StreamARN": stream.stream_arn,
            "StreamName": stream.stream_name,
            "StreamStatus": stream.stream_status,
            "StreamCreationTimestamp": stream.stream_creation_timestamp,
            "StreamModeDetails": stream.stream_mode_details,
            "OpenShardCount": open_shard_count,
            "ConsumerCount": consumer_count,
        }
    })))
}