fix: improve logs when show on aws

Author: fdaciuk

Diff for: rust-api-gateway-v1-2/functions/api/src/main.rs

@@ -12,12 +12,26 @@ use tracing_subscriber;
 
 #[tokio::main]
 async fn main() -> Result<(), LambdaError> {
-  tracing_subscriber::fmt::init();
+  setup_logs();
   let func = service_fn(handler);
   lambda_http::run(func).await?;
   Ok(())
 }
 
+fn setup_logs() {
+  match is_local() {
+    true => tracing_subscriber::fmt::init(),
+    false => tracing_subscriber::fmt().with_ansi(false).init(),
+  }
+}
+
+fn is_local() -> bool {
+  match std::env::var("ENVIRONMENT") {
+    Ok(value) => value == "local",
+    Err(_) => false,
+  }
+}
+
 async fn handler(request: Request) -> Result<impl IntoResponse, LambdaError> {
   match request.body() {
     Body::Text(value) => match serde_json::from_str::<UserData>(value) {

Diff for: rust-api-gateway-v1/functions/api/src/main.rs

@@ -10,12 +10,26 @@ use tracing_subscriber;
 
 #[tokio::main]
 async fn main() -> Result<(), LambdaError> {
-  tracing_subscriber::fmt::init();
+  setup_logs();
   let func = service_fn(handler);
   lambda_runtime::run(func).await?;
   Ok(())
 }
 
+fn setup_logs() {
+  match is_local() {
+    true => tracing_subscriber::fmt::init(),
+    false => tracing_subscriber::fmt().with_ansi(false).init(),
+  };
+}
+
+fn is_local() -> bool {
+  match std::env::var("ENVIRONMENT") {
+    Ok(value) => value == "local",
+    Err(_) => false,
+  }
+}
+
 async fn handler(
   event: LambdaEvent<ApiGatewayProxyRequest>,
 ) -> Result<ApiGatewayProxyResponse, LambdaError> {

Diff for: rust-api-gateway-v2-sqs/functions/api/src/main.rs

@@ -35,12 +35,19 @@ async fn main() -> Result<(), LambdaError> {
 }
 
 fn setup_logs() {
-  match is_running_on_lambda() {
-    false => tracing_subscriber::fmt::init(),
-    true => tracing_subscriber::fmt().with_ansi(false).init(),
+  match is_local() {
+    true => tracing_subscriber::fmt::init(),
+    false => tracing_subscriber::fmt().with_ansi(false).init(),
   };
 }
 
+fn is_local() -> bool {
+  match std::env::var("ENVIRONMENT") {
+    Ok(value) => value == "local",
+    Err(_) => false,
+  }
+}
+
 fn is_running_on_lambda() -> bool {
   std::env::var("OFFLINE").is_err()
 }

Diff for: rust-api-gateway-v2-sqs/functions/queue/src/main.rs

@@ -6,12 +6,26 @@ use tracing_subscriber;
 
 #[tokio::main]
 async fn main() -> Result<(), LambdaError> {
-  tracing_subscriber::fmt::init();
+  setup_logs();
   let func = service_fn(handler);
   lambda_runtime::run(func).await?;
   Ok(())
 }
 
+fn setup_logs() {
+  match is_local() {
+    true => tracing_subscriber::fmt::init(),
+    false => tracing_subscriber::fmt().with_ansi(false).init(),
+  }
+}
+
+fn is_local() -> bool {
+  match std::env::var("ENVIRONMENT") {
+    Ok(value) => value == "local",
+    Err(_) => false,
+  }
+}
+
 #[derive(Debug, Deserialize)]
 struct Body {
   first_name: String,

Diff for: rust-api-gateway-v2/functions/api/src/main.rs

@@ -14,15 +14,16 @@ use tracing_subscriber;
 
 #[actix_web::main]
 async fn main() -> Result<(), LambdaError> {
-  tracing_subscriber::fmt::init();
+  setup_logs();
   let factory = || App::new().service(hello_handler).service(create_user);
 
   match is_running_on_lambda() {
     true => run_actix_on_lambda(factory).await?,
     false => {
-      info!("Server is running on http://localhost:3000");
+      let port = 3008;
+      info!("Server is running on http://localhost:{}", port);
       HttpServer::new(factory)
-        .bind(("127.0.0.1", 3000))?
+        .bind(("127.0.0.1", port))?
         .run()
         .await?
     }
@@ -31,6 +32,20 @@ async fn main() -> Result<(), LambdaError> {
   Ok(())
 }
 
+fn setup_logs() {
+  match is_local() {
+    true => tracing_subscriber::fmt::init(),
+    false => tracing_subscriber::fmt().with_ansi(false).init(),
+  }
+}
+
+fn is_local() -> bool {
+  match std::env::var("ENVIRONMENT") {
+    Ok(value) => value == "local",
+    Err(_) => false,
+  }
+}
+
 fn is_running_on_lambda() -> bool {
   std::env::var("OFFLINE").is_err()
 }

Diff for: rust-lambda-sqs/functions/api/src/main.rs

@@ -6,12 +6,26 @@ use tracing_subscriber;
 
 #[tokio::main]
 async fn main() -> Result<(), LambdaError> {
-  tracing_subscriber::fmt::init();
+  setup_logs();
   let func = service_fn(handler);
   lambda_runtime::run(func).await?;
   Ok(())
 }
 
+fn setup_logs() {
+  match is_local() {
+    true => tracing_subscriber::fmt::init(),
+    false => tracing_subscriber::fmt().with_ansi(false).init(),
+  }
+}
+
+fn is_local() -> bool {
+  match std::env::var("ENVIRONMENT") {
+    Ok(value) => value == "local",
+    Err(_) => false,
+  }
+}
+
 #[derive(Debug, Deserialize)]
 struct Body {
   first_name: String,

Diff for: rust-lambda/functions/api/src/main.rs

@@ -5,12 +5,26 @@ use tracing_subscriber;
 
 #[tokio::main]
 async fn main() -> Result<(), LambdaError> {
-  tracing_subscriber::fmt::init();
+  setup_logs();
   let func = service_fn(handler);
   lambda_runtime::run(func).await?;
   Ok(())
 }
 
+fn setup_logs() {
+  match is_local() {
+    true => tracing_subscriber::fmt::init(),
+    false => tracing_subscriber::fmt().with_ansi(false).init(),
+  }
+}
+
+fn is_local() -> bool {
+  match std::env::var("ENVIRONMENT") {
+    Ok(value) => value == "local",
+    Err(_) => false,
+  }
+}
+
 async fn handler(event: LambdaEvent<Value>) -> Result<Value, LambdaError> {
   let (event, _context) = event.into_parts();
   let first_name = event["firstName"].as_str().unwrap_or("world");