Unverified Commit d6e82f5f authored by Doug's avatar Doug Committed by GitHub
Browse files

Added Kinesis code examples (#306) 



* Added Kinesis code examples

* Cleanup region handling

* Update Cargo.toml

* Update Cargo.toml for new crate names

* Updated Kinesis DeleteStream code example

* Updated Kinesis code examples after running clippy

* Updated Kinesis code examples with latest changes

* Updated Kinesis DescribeStream example.

* Update aws/sdk/examples/kinesis-put-record/src/main.rs

* Update aws/sdk/examples/kinesis-put-record/Cargo.toml

* Update aws/sdk/examples/kinesis-put-record/src/main.rs

Co-authored-by: default avatarRussell Cohen <russell.r.cohen@gmail.com>
Co-authored-by: default avatarRussell Cohen <rcoh@amazon.com>
parent 409b6b99

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

+6 −0
Original line number Diff line number Diff line
@@ -46,6 +46,12 @@ impl ProvideRegion for Region { 
    }

}



impl<'a> ProvideRegion for &'a Region {

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

        Some((*self).clone())

    }

}



pub fn default_provider() -> impl ProvideRegion {

    EnvironmentProvider

}

aws/sdk/examples/kinesis-create-stream/Cargo.toml

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

name = "kinesis-create-stream"

version = "0.1.0"

authors = ["Doug Schwartz <dougsch@amazon.com>"]

edition = "2018"



# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html



[dependencies]

kinesis = { package = "aws-sdk-kinesis", path = "../../build/aws-sdk/kinesis" }

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/kinesis-create-stream/src/main.rs

0 → 100644
+65 −0
Original line number Diff line number Diff line 
/*

 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.

 * SPDX-License-Identifier: Apache-2.0.

 */



use std::process;



use kinesis::{Client, Config, Region};



use aws_types::region::{EnvironmentProvider, ProvideRegion};



use structopt::StructOpt;

use tracing_subscriber::fmt::format::FmtSpan;

use tracing_subscriber::fmt::SubscriberBuilder;



#[derive(Debug, StructOpt)]

struct Opt {

    /// The region

    #[structopt(short, long)]

    region: Option<String>,



    #[structopt(short, long)]

    name: String,



    #[structopt(short, long)]

    verbose: bool,

}



#[tokio::main]

async fn main() {

    let Opt {

        name,

        region,

        verbose,

    } = Opt::from_args();



    let region = EnvironmentProvider::new()

        .region()

        .or_else(|| region.as_ref().map(|region| Region::new(region.clone())))

        .unwrap_or_else(|| Region::new("us-west-2"));



    if verbose {

        println!("Kinesis client version: {}\n", kinesis::PKG_VERSION);

        println!("Region:      {:?}", &region);

        println!("Stream name: {}", name);



        SubscriberBuilder::default()

            .with_env_filter("info")

            .with_span_events(FmtSpan::CLOSE)

            .init();

    }



    let config = Config::builder().region(region).build();



    let client = Client::from_conf(config);



    match client.create_stream().stream_name(name).send().await {

        Ok(_) => println!("Created stream"),

        Err(e) => {

            println!("Got an error creating stream");

            println!("{}", e);

            process::exit(1);

        }

    };

}

aws/sdk/examples/kinesis-delete-stream/Cargo.toml

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

name = "kinesis-delete-stream"

version = "0.1.0"

authors = ["Doug Schwartz <dougsch@amazon.com>"]

edition = "2018"



# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html



[dependencies]

kinesis = { package = "aws-sdk-kinesis", path = "../../build/aws-sdk/kinesis" }

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/kinesis-delete-stream/src/main.rs

0 → 100644
+67 −0
Original line number Diff line number Diff line 
/*

 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.

 * SPDX-License-Identifier: Apache-2.0.

 */



use std::process;



use kinesis::{Client, Config, Region};



use aws_types::region::{EnvironmentProvider, ProvideRegion};



use structopt::StructOpt;

use tracing_subscriber::fmt::format::FmtSpan;

use tracing_subscriber::fmt::SubscriberBuilder;



#[derive(Debug, StructOpt)]

struct Opt {

    /// The region

    #[structopt(short, long)]

    region: Option<String>,



    /// The name of the stream to delete

    #[structopt(short, long)]

    name: String,



    /// Whether to display additional information

    #[structopt(short, long)]

    verbose: bool,

}



#[tokio::main]

async fn main() {

    let Opt {

        name,

        region,

        verbose,

    } = Opt::from_args();



    let region = EnvironmentProvider::new()

        .region()

        .or_else(|| region.as_ref().map(|region| Region::new(region.clone())))

        .unwrap_or_else(|| Region::new("us-west-2"));



    if verbose {

        println!("Kinesis client version: {}\n", kinesis::PKG_VERSION);

        println!("Region:      {:?}", &region);

        println!("Stream name: {}", name);



        SubscriberBuilder::default()

            .with_env_filter("info")

            .with_span_events(FmtSpan::CLOSE)

            .init();

    }



    let config = Config::builder().region(region).build();



    let client = Client::from_conf(config);



    match client.delete_stream().stream_name(name).send().await {

        Ok(_) => println!("Deleted stream."),

        Err(e) => {

            println!("Got an error deleting the stream:");

            println!("{}", e);

            process::exit(1);

        }

    };

}