From 10498b9b5739be2d48274acdc105fa12a0612b04 Mon Sep 17 00:00:00 2001 From: Victor Mignot Date: Mon, 14 Nov 2022 22:53:06 -0500 Subject: [PATCH] Introducing env catcher function --- src/env_catcher.rs | 0 src/environment.rs | 15 +++++++++++++++ src/lib.rs | 1 + 3 files changed, 16 insertions(+) delete mode 100644 src/env_catcher.rs create mode 100644 src/environment.rs diff --git a/src/env_catcher.rs b/src/env_catcher.rs deleted file mode 100644 index e69de29..0000000 diff --git a/src/environment.rs b/src/environment.rs new file mode 100644 index 0000000..19b8a1a --- /dev/null +++ b/src/environment.rs @@ -0,0 +1,15 @@ +//! Utility function to deal wit environment variable + +use log::error; +use std::env; + +/// Get the environment variiable var_name or panic +pub fn get_env_variable(var_name: &String) -> String { + match env::var(var_name) { + Ok(v) => v, + Err(_) => { + error!(target: "bot_warn_errors", "Failed to fetch the {} environment variable.", var_name); + panic!("Exiting..."); + } + } +} diff --git a/src/lib.rs b/src/lib.rs index af10ddb..a7099e5 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -8,6 +8,7 @@ mod client; mod database; +mod environment; pub mod errors; pub use crate::{