Unverified Commit 082d05f0 authored by Russell Cohen's avatar Russell Cohen Committed by GitHub
Browse files

Add ECR, ecr example, and chain region provider (#557)

parent c3de3140

aws/rust-runtime/aws-types/src/region.rs

+50 −0
Original line number Diff line number Diff line
@@ -36,6 +36,56 @@ impl Region { 
    }

}



pub struct ChainProvider {

    providers: Vec<Box<dyn ProvideRegion>>,

}



/// Implement a region provider based on a series of region providers

///

/// # Example

/// ```rust

/// use aws_types::region::{ChainProvider, Region};

/// use std::env;

/// // region provider that first checks the `CUSTOM_REGION` environment variable,

/// // then checks the default provider chain, then falls back to us-east-2

/// let provider = ChainProvider::first_try(env::var("CUSTOM_REGION").ok().map(Region::new))

///     .or_default_provider()

///     .or_else(Region::new("us-east-2"));

/// ```

impl ChainProvider {

    pub fn first_try(provider: impl ProvideRegion + 'static) -> Self {

        ChainProvider {

            providers: vec![Box::new(provider)],

        }

    }

    pub fn or_else(mut self, fallback: impl ProvideRegion + 'static) -> Self {

        self.providers.push(Box::new(fallback));

        self

    }



    pub fn or_default_provider(mut self) -> Self {

        self.providers.push(Box::new(default_provider()));

        self

    }

}



impl ProvideRegion for Option<Region> {

    fn region(&self) -> Option<Region> {

        self.clone()

    }

}



impl ProvideRegion for ChainProvider {

    fn region(&self) -> Option<Region> {

        for provider in &self.providers {

            if let Some(region) = provider.region() {

                return Some(region);

            }

        }

        None

    }

}



/// Provide a [`Region`](Region) to use with AWS requests

///

/// For most cases [`default_provider`](default_provider) will be the best option, implementing

aws/sdk/build.gradle.kts

+1 −0
Original line number Diff line number Diff line
@@ -80,6 +80,7 @@ val tier1Services = setOf( 
    "sqs",

    "ssm",

    "sts",

    "ecr",

    "eks"

)

aws/sdk/examples/ecr/Cargo.toml

0 → 100644
+12 −0
Original line number Diff line number Diff line 
[package]

name = "ecr-examples"

version = "0.1.0"

authors = ["Russell Cohen <rcoh@amazon.com>"]

edition = "2018"



[dependencies]

aws-sdk-ecr = { path = "../../build/aws-sdk/ecr" }

tokio = { version = "1", features = ["full"]}

structopt = { version = "0.3", default-features = false }

tracing-subscriber = { version = "0.2.16", features = ["fmt"] }

aws-types = { path = "../../build/aws-sdk/aws-types" }

aws/sdk/examples/ecr/src/main.rs

0 → 100644
+46 −0
Original line number Diff line number Diff line 
use aws_sdk_ecr::{Config, Region};

use aws_types::region;

use structopt::StructOpt;



#[derive(Debug, StructOpt)]

struct Opt {

    /// The region

    #[structopt(short, long)]

    region: Option<String>,



    #[structopt(long)]

    repository: String,



    #[structopt(short, long)]

    verbose: bool,

}

#[tokio::main]

async fn main() -> Result<(), aws_sdk_ecr::Error> {

    let Opt {

        region,

        repository,

        verbose,

    } = Opt::from_args();

    if verbose {

        tracing_subscriber::fmt::init();

    }

    let provider = region::ChainProvider::first_try(region.map(Region::new))

        .or_default_provider()

        .or_else(Region::new("us-east-2"));

    let client = aws_sdk_ecr::Client::from_conf(Config::builder().region(provider).build());

    let rsp = client

        .list_images()

        .repository_name(&repository)

        .send()

        .await?;

    let images = rsp.image_ids.unwrap_or_default();

    println!("found {} images", images.len());

    for image in images {

        println!(

            "image: {}:{}",

            image.image_tag.unwrap(),

            image.image_digest.unwrap()

        );

    }

    Ok(())

}