diff --git a/crates/datadog-serverless-compat/src/main.rs b/crates/datadog-serverless-compat/src/main.rs index d50798f..b227c15 100644 --- a/crates/datadog-serverless-compat/src/main.rs +++ b/crates/datadog-serverless-compat/src/main.rs @@ -50,7 +50,7 @@ pub async fn main() { .map(|val| val.to_lowercase()) .unwrap_or("info".to_string()); - let (_, env_type) = match read_cloud_env() { + let (_, mut env_type) = match read_cloud_env() { Some(value) => value, None => { error!("Unable to identify environment. Shutting down Mini Agent."); @@ -58,6 +58,17 @@ pub async fn main() { } }; + if let Ok(res) = env::var("DD_ENV_TYPE") { + match res.to_lowercase().as_str() { + "cloudfunction" => env_type = EnvironmentType::CloudFunction, + "azurefunction" => env_type = EnvironmentType::AzureFunction, + "azurespringapp" => env_type = EnvironmentType::AzureSpringApp, + _ => { + debug!("DD_ENV_TYPE value not recognized, env_type not overwritten"); + } + } + } + let dogstatsd_tags = match env_type { EnvironmentType::CloudFunction => "origin:cloudfunction,dd.origin:cloudfunction", EnvironmentType::AzureFunction => "origin:azurefunction,dd.origin:azurefunction",