aws_utils_ssm 0.4.0

AWS SSM utilities for getting parameter values from AWS Systems Manager Parameter Store
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

use std::time::Duration;

use aws_config::{
    BehaviorVersion,
    timeout::{TimeoutConfig, TimeoutConfigBuilder},
};
use aws_sdk_ssm::{Client, config::SharedInterceptor};

pub mod error;
pub mod ssm;

pub use aws_sdk_ssm;

pub async fn make_client_with_timeout_default(endpoint_url: Option<String>) -> Client {
    make_client_with_timeout(
        endpoint_url,
        Some(Duration::from_secs(3100)),
        Some(Duration::from_secs(60)),
        Some(Duration::from_secs(55)),
        Some(Duration::from_secs(50)),
    )
    .await
}

pub async fn make_client_with_timeout(
    endpoint_url: Option<String>,
    connect_timeout: Option<Duration>,
    operation_timeout: Option<Duration>,
    operation_attempt_timeout: Option<Duration>,
    read_timeout: Option<Duration>,
) -> Client {
    let mut timeout_config = TimeoutConfigBuilder::new();
    timeout_config
        .set_connect_timeout(connect_timeout)
        .set_operation_timeout(operation_timeout)
        .set_operation_attempt_timeout(operation_attempt_timeout)
        .set_read_timeout(read_timeout);
    make_client(endpoint_url, Some(timeout_config.build()), None).await
}

pub async fn make_client(
    endpoint_url: Option<String>,
    timeout_config: Option<TimeoutConfig>,
    interceptor: Option<SharedInterceptor>,
) -> Client {
    let mut config_loader = aws_config::defaults(BehaviorVersion::latest());
    if let Some(timeout_config) = timeout_config {
        config_loader = config_loader.timeout_config(timeout_config);
    }
    let config = config_loader.load().await;
    let mut builder = aws_sdk_ssm::config::Builder::from(&config);
    if let Some(aws_endpoint_url) = endpoint_url {
        builder = builder.endpoint_url(aws_endpoint_url)
    }
    if let Some(interceptor) = interceptor {
        builder.push_interceptor(interceptor);
    }
    Client::from_conf(builder.build())
}