aws_credential_types/
token_fn.rs

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
130
131
132
133
/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

//! Types that allow an access token provider to be created from a closure

use crate::provider::{future, token::ProvideToken};
use std::fmt::{self, Debug, Formatter};
use std::future::Future;
use std::marker::PhantomData;

/// A [`ProvideToken`] implemented by a closure.
///
/// See [`provide_token_fn`] for more details.
#[derive(Copy, Clone)]
pub struct ProvideTokenFn<'c, T> {
    f: T,
    phantom: PhantomData<&'c T>,
}

impl<T> Debug for ProvideTokenFn<'_, T> {
    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
        write!(f, "ProvideTokenFn")
    }
}

impl<'c, T, F> ProvideToken for ProvideTokenFn<'c, T>
where
    T: Fn() -> F + Send + Sync + 'c,
    F: Future<Output = crate::provider::token::Result> + Send + 'static,
{
    fn provide_token<'a>(&'a self) -> future::ProvideToken<'a>
    where
        Self: 'a,
    {
        future::ProvideToken::new((self.f)())
    }
}

/// Returns a new token provider built with the given closure. This allows you
/// to create an [`ProvideToken`] implementation from an async block that returns
/// a [`crate::provider::token::Result`].
///
/// # Examples
///
/// ```no_run
/// use aws_credential_types::Token;
/// use aws_credential_types::token_fn::provide_token_fn;
///
/// async fn load_token() -> Token {
///     todo!()
/// }
///
/// provide_token_fn(|| async {
///     // Async process to retrieve a token goes here
///     let token = load_token().await;
///     Ok(token)
/// });
/// ```
pub fn provide_token_fn<'c, T, F>(f: T) -> ProvideTokenFn<'c, T>
where
    T: Fn() -> F + Send + Sync + 'c,
    F: Future<Output = crate::provider::token::Result> + Send + 'static,
{
    ProvideTokenFn {
        f,
        phantom: Default::default(),
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::Token;
    use async_trait::async_trait;
    use std::fmt::{Debug, Formatter};

    fn assert_send_sync<T: Send + Sync>() {}

    #[test]
    fn creds_are_send_sync() {
        assert_send_sync::<Token>()
    }

    #[async_trait]
    trait AnotherTrait: Send + Sync {
        async fn token(&self) -> Token;
    }

    struct AnotherTraitWrapper<T> {
        inner: T,
    }

    impl<T> Debug for AnotherTraitWrapper<T> {
        fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
            write!(f, "wrapper")
        }
    }

    impl<T: AnotherTrait> ProvideToken for AnotherTraitWrapper<T> {
        fn provide_token<'a>(&'a self) -> future::ProvideToken<'a>
        where
            Self: 'a,
        {
            future::ProvideToken::new(async move { Ok(self.inner.token().await) })
        }
    }

    // Test that the closure passed to `provide_token_fn` is allowed to borrow things
    #[tokio::test]
    async fn provide_token_fn_closure_can_borrow() {
        fn check_is_str_ref(_input: &str) {}
        async fn test_async_provider(input: String) -> crate::provider::token::Result {
            Ok(Token::new(input, None))
        }

        let things_to_borrow = vec!["one".to_string(), "two".to_string()];

        let mut providers = Vec::new();
        for thing in &things_to_borrow {
            let provider = provide_token_fn(move || {
                check_is_str_ref(thing);
                test_async_provider(thing.into())
            });
            providers.push(provider);
        }

        let (two, one) = (providers.pop().unwrap(), providers.pop().unwrap());
        assert_eq!("one", one.provide_token().await.unwrap().token());
        assert_eq!("two", two.provide_token().await.unwrap().token());
    }
}