Add init checking to PrettyLogger

src/logger/mod.rs

@@ -12,6 +12,7 @@ use std::time::SystemTime;
 pub use pretty_logger::PrettyLogger;
 pub use pretty_logger::log;
 
+pub static mut IS_DEFAULT_LOGGER: bool = true;
 static mut LOGGER: &dyn Logger = &DefaultLogger;
 
 #[derive(Debug, serde::Serialize, serde::Deserialize)]
@@ -46,12 +47,14 @@ pub fn set_logger_box(logger: Box<dyn Logger>) {
     #[cfg(feature = "log")]
     unsafe {
         LOGGER = Box::leak(logger);
+        IS_DEFAULT_LOGGER = false;
     }
 }
 
 pub fn set_logger(logger: &'static dyn Logger) {
     unsafe {
         LOGGER = logger;
+        IS_DEFAULT_LOGGER = false;
     }
 }
 

src/logger/pretty_logger.rs

@@ -52,15 +52,19 @@ pub fn log(message: &Record) {
 
 impl PrettyLogger {
     pub fn init() {
-        crate::logger::set_logger_box(Box::new(PrettyLogger { output: None }));
+        if !unsafe { crate::logger::IS_DEFAULT_LOGGER } {
+            crate::logger::set_logger_box(Box::new(PrettyLogger { output: None }));
+        }
     }
 
     pub fn init_output<T>(output: T)
     where
         T: Fn(&Record) + 'static,
     {
-        crate::logger::set_logger_box(Box::new(PrettyLogger {
-            output: Some(Box::new(output)),
-        }));
+        if !unsafe { crate::logger::IS_DEFAULT_LOGGER } {
+            crate::logger::set_logger_box(Box::new(PrettyLogger {
+                output: Some(Box::new(output)),
+            }));
+        }
     }
 }