From b7f76db2a02aba793a68ba35c2b4b51d5e069bdc Mon Sep 17 00:00:00 2001 From: Andinus Date: Wed, 29 Sep 2021 20:17:40 +0530 Subject: Rust: Semi Structured Logs: Implement Display trait to format log --- rust/semi-structured-logs/src/lib.rs | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) (limited to 'rust') diff --git a/rust/semi-structured-logs/src/lib.rs b/rust/semi-structured-logs/src/lib.rs index 4f06874..0160b1d 100644 --- a/rust/semi-structured-logs/src/lib.rs +++ b/rust/semi-structured-logs/src/lib.rs @@ -1,25 +1,30 @@ +use std::fmt::{Display, Formatter, Result}; + +#[derive(Debug)] pub enum LogLevel { Info, Warning, Error, } -pub fn log(level: LogLevel, message: &str) -> String { - match level { - LogLevel::Info => info(message), - LogLevel::Warning => warn(message), - LogLevel::Error => error(message), +impl Display for LogLevel { + fn fmt(&self, f: &mut Formatter) -> Result { + write!(f, "{:?}", self) } } +pub fn log(level: LogLevel, message: &str) -> String { + format!("[{}]: {}", level.to_string().to_uppercase(), message) +} + pub fn info(message: &str) -> String { - format!("[INFO]: {}", message) + log(LogLevel::Info, message) } pub fn warn(message: &str) -> String { - format!("[WARNING]: {}", message) + log(LogLevel::Warning, message) } pub fn error(message: &str) -> String { - format!("[ERROR]: {}", message) + log(LogLevel::Error, message) } -- cgit 1.4.1-2-gfad0