-
Notifications
You must be signed in to change notification settings - Fork 5.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
f0298d0
commit e727634
Showing
7 changed files
with
137 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -4,13 +4,15 @@ version = "0.1.0" | |
authors = [ | ||
"John DiSanti <[email protected]>", | ||
"Doug Schwartz <[email protected]>", | ||
"David Souther <[email protected]>" | ||
] | ||
edition = "2021" | ||
|
||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html | ||
|
||
[dependencies] | ||
aws-config = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "next" } | ||
aws-credential-types = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "next" } | ||
aws-sdk-s3 = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "next" } | ||
tokio = { version = "1.20.1", features = ["full"] } | ||
clap = { version = "~4.4", features = ["derive"] } | ||
|
Empty file.
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,131 @@ | ||
/* | ||
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. | ||
* SPDX-License-Identifier: Apache-2.0. | ||
*/ | ||
|
||
#![allow(clippy::result_large_err)] | ||
|
||
use aws_config::meta::region::RegionProviderChain; | ||
use aws_credential_types::provider::{ProvideCredentials, SharedCredentialsProvider}; | ||
use aws_sdk_s3::config::retry::RetryConfig; | ||
use aws_sdk_s3::config::Credentials; | ||
use aws_sdk_s3::{config::Region, meta::PKG_VERSION, Client, Error}; | ||
use clap::Parser; | ||
|
||
// TODO(you): When running these examples, create or edit these files & update them with your values. | ||
const STATIC_ACCESS_KEY_ID: &str = include_str!("./testing/NO_ENV_STATIC_ACCESS_KEY_ID"); | ||
const STATIC_SECRET_ACCESS_KEY: &str = include_str!("./testing/NO_ENV_STATIC_ACCESS_KEY_ID"); | ||
|
||
#[derive(Debug, Parser)] | ||
struct Opt { | ||
/// The AWS Region. | ||
#[structopt(short, long)] | ||
region: Option<String>, | ||
|
||
/// The number of (re)tries. | ||
#[structopt(short, long, default_value = "2")] | ||
tries: u32, | ||
|
||
/// Whether to display additional information. | ||
#[structopt(short, long)] | ||
verbose: bool, | ||
} | ||
|
||
// Shows your buckets. | ||
async fn show_num_buckets(client: &Client) -> Result<(), Error> { | ||
let resp = client.list_buckets().send().await?; | ||
let buckets = resp.buckets(); | ||
|
||
println!("Found {} buckets in all regions.", buckets.len()); | ||
|
||
Ok(()) | ||
} | ||
|
||
// WARNING: This example is for demonstration purposes only. Your code should always | ||
// load credentials in a secure fashion. | ||
#[derive(Debug)] | ||
struct StaticCredentials { | ||
access_key_id: &'static str, | ||
secret_access_key: &'static str, | ||
} | ||
|
||
impl StaticCredentials { | ||
pub fn new(access_key_id: &'static str, secret_access_key: &'static str) -> Self { | ||
Self { | ||
access_key_id, | ||
secret_access_key, | ||
} | ||
} | ||
|
||
async fn load_credentials(&self) -> aws_credential_types::provider::Result { | ||
Ok(Credentials::new( | ||
self.access_key_id, | ||
self.secret_access_key, | ||
None, | ||
None, | ||
"StaticCredentials", | ||
)) | ||
} | ||
} | ||
|
||
impl ProvideCredentials for StaticCredentials { | ||
fn provide_credentials<'a>( | ||
&'a self, | ||
) -> aws_credential_types::provider::future::ProvideCredentials<'a> | ||
where | ||
Self: 'a, | ||
{ | ||
aws_credential_types::provider::future::ProvideCredentials::new(self.load_credentials()) | ||
} | ||
} | ||
|
||
/// Displays how many Amazon S3 buckets you have. | ||
/// # Arguments | ||
/// | ||
/// * `[-r REGION]` - The Region in which the client is created. | ||
/// If not supplied, uses the value of the **AWS_REGION** environment variable. | ||
/// If the environment variable is not set, defaults to **us-west-2**. | ||
/// * `[-t TRIES]` - The number of times to (re)try the request. | ||
/// * `[-v]` - Whether to display additional information. | ||
#[tokio::main] | ||
async fn main() -> Result<(), Error> { | ||
tracing_subscriber::fmt::init(); | ||
|
||
let Opt { | ||
region, | ||
tries, | ||
verbose, | ||
} = Opt::parse(); | ||
|
||
let region = RegionProviderChain::first_try(region.map(Region::new)) | ||
.or_else(Region::new("us-west-2")) | ||
.region() | ||
.await | ||
.expect("parsed region"); | ||
println!(); | ||
|
||
if verbose { | ||
println!("S3 client version: {PKG_VERSION}"); | ||
println!("Region: {region}"); | ||
println!("Retries: {tries}"); | ||
println!(); | ||
} | ||
|
||
assert_ne!(tries, 0, "You cannot set zero retries."); | ||
|
||
let shared_config = aws_config::SdkConfig::builder() | ||
.region(region) | ||
.credentials_provider(SharedCredentialsProvider::new(StaticCredentials::new( | ||
STATIC_ACCESS_KEY_ID, | ||
STATIC_SECRET_ACCESS_KEY, | ||
))) | ||
// Set max attempts. | ||
// If tries is 1, there are no retries. | ||
.retry_config(RetryConfig::standard().with_max_attempts(tries)) | ||
.build(); | ||
|
||
// Construct an S3 client with customized retry configuration. | ||
let client = Client::new(&shared_config); | ||
|
||
show_num_buckets(&client).await | ||
} |
2 changes: 2 additions & 0 deletions
2
rust_dev_preview/examples/sdk-config/src/bin/testing/.gitignore
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
NO_ENV_STATIC_ACCESS_KEY_ID | ||
NO_ENV_STATIC_SECRET_ACCESS_KEY |
1 change: 1 addition & 0 deletions
1
rust_dev_preview/examples/sdk-config/src/bin/testing/NO_ENV_STATIC_ACCESS_KEY_ID
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
REPLACE_WHEN_RUNNING_BUT_DO_NOT_COMMIT |
1 change: 1 addition & 0 deletions
1
rust_dev_preview/examples/sdk-config/src/bin/testing/NO_ENV_STATIC_SECRET_ACCESS_KEY
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
REPLACE_WHEN_RUNNING_BUT_DO_NOT_COMMIT |