Как я могу вернуться к учетным данным из файла, если переменные среды не найдены? - PullRequest
Новая
       39

Как я могу вернуться к учетным данным из файла, если переменные среды не найдены?

0 голосов
/ 02 октября 2019

Как я могу использовать учетные данные из проанализированного файла (config.yml), если переменные среды не были найдены? Для тестирования я использую этот пример : 

extern crate rusoto_core;
extern crate rusoto_s3;

use rusoto_core::credential::ChainProvider;
use rusoto_core::request::HttpClient;
use rusoto_core::Region;
use rusoto_s3::{S3, S3Client};
use std::time::{Duration, Instant};

fn main() {
    let mut chain = ChainProvider::new();
    chain.set_timeout(Duration::from_millis(200));
    let s3client = S3Client::new_with(
        HttpClient::new().expect("failed to create request dispatcher"),
        chain,
        Region::UsEast1,
    );

    let start = Instant::now();
    println!("Starting up at {:?}", start);

    match s3client.list_buckets().sync() {
        Err(e) => println!("Error listing buckets: {}", e),
        Ok(buckets) => println!("Buckets found: {:?}", buckets),
    };
    println!("Took {:?}", Instant::now().duration_since(start));
}

Он работает, но требует переменных окружения AWS_ACCESS_KEY_ID и AWS_SECRET_ACCESS_KEY. Я хотел бы расширить его, чтобы при отсутствии определенных переменных среды я мог бы использовать ключи, найденные в разобранном файле: 

// parse config file
let file = std::fs::File::open("config.yml").expect("Unable to open file");
let yml: Config = match serde_yaml::from_reader(file) {
    Err(err) => {
        println!("Error: {}", err);
        return;
    }
    Ok(yml) => yml,
};

Например, config.yml может выглядеть примерно так: 

---
endpoint: s3.provider
access_key: ACCESS_KEY_ID
secret_key: SECRET_ACCESS_KEY

Что я могу добавить к chain, чтобы использовать учетные данные, найденные в config.yml, что-то вроде: 

let config_provider = StaticProvider::new_minimal(yml.access_key, yml.secret_key);

Как отдать предпочтение среде и, если она не найденазатем используйте учетные данные, предоставленные StaticProvider?

1 Ответ

2 голосов
/ 02 октября 2019

ChainProvider на самом деле имеет четыре источника для проверки учетных данных AWS. Третий файл - это файл конфигурации AWS, расположенный в домашнем каталоге пользователя. Но его формат предопределен.

Если вы настаиваете на использовании своего собственного файла YAML, вы можете связать EnvironmentProvider и StaticProvider вместе, например, так: 

use futures::future::Future;
use rusoto_core::request::HttpClient;
use rusoto_core::Region;
use rusoto_credential::{
    AwsCredentials,
    CredentialsError,
    EnvironmentProvider,
    ProvideAwsCredentials,
    StaticProvider,
};
use rusoto_s3::{S3, S3Client};
use std::time::Instant;

struct MyChainProvider<'a> {
    access_key: &'a str,
    secret_key: &'a str,
}

impl<'a> ProvideAwsCredentials for MyChainProvider<'a> {
    type Future = Box<dyn Future<Item=AwsCredentials, Error=CredentialsError> + Send>;

    fn credentials(&self) -> Self::Future {
        let future = EnvironmentProvider::default().credentials()
            .or_else({
                let access_key = self.access_key.to_string();
                let secret_key = self.secret_key.to_string();

                move |_| -> Self::Future {
                    Box::new(StaticProvider::new_minimal(access_key, secret_key).credentials())
                }
            });

        Box::new(future)
    }
}

fn main() {
    let chain = MyChainProvider {
        access_key: ...,
        secret_key: ...,
    };

    ...
}
...