Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -254,7 +254,7 @@ X-Ray exporter.

| Option | Default | Options |
|------------------------------------|-----------|------------------|
| --awsxray-exporter-region | us-east-1 | aws region codes |
| --awsxray-exporter-region | `AWS_REGION` env-var if set, then `AWS_DEFAULT_REGION`, then `us-east-1` | aws region codes |
| --awsxray-exporter-custom-endpoint | | |

For a list of available AWS X-Ray region codes here: https://docs.aws.amazon.com/general/latest/gr/xray.html
Expand All @@ -272,7 +272,7 @@ exporter.

| Option | Default | Options |
|--------------------------------------------------------|------------------|------------------|
| --awsemf-exporter-region | us-east-1 | aws region codes |
| --awsemf-exporter-region | `AWS_REGION` env-var if set, then `AWS_DEFAULT_REGION`, then `us-east-1` | aws region codes |
| --awsemf-exporter-custom-endpoint | | |
| --awsemf-exporter-log-group-name | /metrics/default | |
| --awsemf-exporter-log-stream-name | otel-stream | |
Expand Down
36 changes: 31 additions & 5 deletions src/exporters/shared/aws.rs
Original file line number Diff line number Diff line change
Expand Up @@ -79,9 +79,11 @@ impl Display for Region {
}
}

impl From<String> for Region {
fn from(s: String) -> Self {
match s.as_str() {
impl core::str::FromStr for Region {
type Err = String;

fn from_str(s: &str) -> Result<Self, Self::Err> {
Ok(match s {
"us-east-1" => Region::UsEast1,
"us-east-2" => Region::UsEast2,
"us-west-1" => Region::UsWest1,
Expand Down Expand Up @@ -114,7 +116,31 @@ impl From<String> for Region {
"me-south-1" => Region::MeSouth1,
"me-central-1" => Region::MeCentral1,
"sa-east-1" => Region::SaEast1,
_ => panic!("Unknown region: {}", s),
}
_ => return Err(format!("Unknown region: {}", s)),
})
}
}

impl From<String> for Region {
fn from(s: String) -> Self {
s.parse().unwrap()
}
}

impl Default for Region {
fn default() -> Self {
use std::env;
// Try to read AWS_REGION
env::var("AWS_REGION")
// Else, try to read AWS_DEFAULT_REGION
.or_else(|_| env::var("AWS_DEFAULT_REGION"))
// Discard the error
.ok()
// Parse the value and discard the error
.map(|s| s.parse().ok())
// Flatten the result
.flatten()
// Unwrap it or fallback on us-east-1
.unwrap_or_else(|| Region::UsEast1)
}
}
2 changes: 1 addition & 1 deletion src/exporters/xray/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ pub struct XRayExporterConfigBuilder {
impl Default for XRayExporterConfigBuilder {
fn default() -> Self {
Self {
region: Region::UsEast1,
region: Default::default(),
custom_endpoint: None,
retry_config: Default::default(),
}
Expand Down
5 changes: 2 additions & 3 deletions src/init/awsemf_exporter.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,10 +9,9 @@ pub struct AwsEmfExporterArgs {
/// AWS EMF Exporter Region
#[arg(
id("AWSEMF_REGION"),
value_enum,
long("awsemf-exporter-region"),
env = "ROTEL_AWSEMF_EXPORTER_REGION",
default_value = "us-east-1"
default_value_t
)]
pub region: Region,

Expand Down Expand Up @@ -82,7 +81,7 @@ pub struct AwsEmfExporterArgs {
impl Default for AwsEmfExporterArgs {
fn default() -> Self {
Self {
region: Region::UsEast1,
region: Region::default(),
custom_endpoint: None,
log_group_name: "/metrics/default".to_string(),
log_stream_name: "otel-stream".to_string(),
Expand Down
14 changes: 2 additions & 12 deletions src/init/xray_exporter.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,14 @@ use serde::Deserialize;

use crate::exporters::shared::aws::Region;

#[derive(Debug, Clone, Args, Deserialize)]
#[derive(Debug, Default, Clone, Args, Deserialize)]
#[serde(default)]
pub struct XRayExporterArgs {
/// AWS X-Ray Exporter Region
#[arg(
value_enum,
long("awsxray-exporter-region"),
env = "ROTEL_AWSXRAY_EXPORTER_REGION",
default_value = "us-east-1"
default_value_t
)]
pub region: Region,

Expand All @@ -22,12 +21,3 @@ pub struct XRayExporterArgs {
)]
pub custom_endpoint: Option<String>,
}

impl Default for XRayExporterArgs {
fn default() -> Self {
Self {
region: Region::UsEast1,
custom_endpoint: None,
}
}
}