chore: log informative err msg on startup failures

kasbuunk · Dec 14, 2024 · f3847fa · f3847fa
1 parent 8d90f52
commit f3847fa
Showing 1 changed file with 31 additions and 5 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -3,7 +3,7 @@ use std::process;
 use std::sync::Arc;
 
 use chrono::prelude::*;
-use log::info;
+use log::{error, info};
 use tokio::signal::unix::{signal, SignalKind};
 use tokio_util::sync::CancellationToken;
 
@@ -46,7 +46,14 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     // Construct transmitter.
     let transmitter: Arc<dyn contract::Transmitter> = match config.transmitter {
         config::Transmitter::Nats(nats_config) => {
-            let nats_client = nats::connect_to_nats(nats_config).await?;
+            let nats_client = match nats::connect_to_nats(nats_config).await {
+                Ok(client) => client,
+                Err(err) => {
+                    error!("Failed to initialise nats connection: {}", err);
+                    process::exit(1);
+                }
+            };
+
             let transmitter = transmitter_nats::NatsPublisher::new(nats_client);
             info!("Initialised nats transmitter.");
 
@@ -57,19 +64,38 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     // Construct repository.
     let repository: Arc<dyn contract::Repository> = match config.repository {
         config::Repository::Postgres(postgres_config) => {
-            let postgres_connection = postgres::connect_to_database(postgres_config).await?;
+            let postgres_connection = match postgres::connect_to_database(postgres_config).await {
+                Ok(client) => client,
+                Err(err) => {
+                    error!("Failed to initialise postgres connection: {}", err);
+                    process::exit(1);
+                }
+            };
+
             let repository = repository_postgres::RepositoryPostgres::new(postgres_connection);
 
             info!("Initialised postgres repository.");
 
             if config.automigrate {
                 info!("Running migrations.");
 
-                repository.migrate().await?;
+                match repository.migrate().await {
+                    Ok(_) => (),
+                    Err(err) => {
+                        error!("Failed to run sql migrations: {}", err);
+                        process::exit(1);
+                    }
+                };
             }
 
             if config.reset_state {
-                repository.clear_all().await?;
+                match repository.clear_all().await {
+                    Ok(_) => (),
+                    Err(err) => {
+                        error!("Failed to reset repository state: {}", err);
+                        process::exit(1);
+                    }
+                };
             }
 
             Arc::new(repository)