Improve cleanup on dropping the logger (#15)

* adding flag on send to channel in case the logger guard is dropped

* increment version for publish

Author: Abso1ut3Zer0

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "inqjet"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 authors = ["Abso1ut3Zer0"]
 description = "Ultra-fast, low-latency logging for Rust applications"

Cargo.toml

@@ -549,8 +549,8 @@ where
         let parker = Parker::new();
         let notifier = parker.unparker().to_owned();
         let chan = Arc::new(Channel::new(capacity, parker.unparker().to_owned()));
-        let logger = Logger::new(chan.clone(), level);
         let flag = Arc::new(AtomicBool::new(true));
+        let logger = Logger::new(chan.clone(), level, flag.clone());
         let mut appender = Appender::new(chan, parker, flag.clone(), writer);
         log::set_boxed_logger(Box::new(logger))?;
         log::set_max_level(level);

src/lib.rs

@@ -90,6 +90,15 @@ pub struct Logger {
 
     /// Maximum log level that will be processed. Messages below this level are ignored.
     max_level: log::LevelFilter,
+
+    /// Flag to indicate if the logger is still running.
+    running: Arc<AtomicBool>,
+}
+
+impl Drop for Logger {
+    fn drop(&mut self) {
+        self.running.store(false, Ordering::Relaxed);
+    }
 }
 
 impl Logger {
@@ -122,8 +131,16 @@ impl Logger {
     /// // Only log INFO and above (INFO, WARN, ERROR)
     /// let logger = Logger::new(channel, LevelFilter::Info);
     /// ```
-    pub(crate) fn new(chan: Arc<Channel>, max_level: log::LevelFilter) -> Self {
-        Self { chan, max_level }
+    pub(crate) fn new(
+        chan: Arc<Channel>,
+        max_level: log::LevelFilter,
+        running: Arc<AtomicBool>,
+    ) -> Self {
+        Self {
+            chan,
+            max_level,
+            running,
+        }
     }
 }
 
@@ -273,7 +290,9 @@ impl log::Log for Logger {
         s.write_str("\n").ok();
 
         // Send to consumer thread (transfers ownership)
-        self.chan.push(s);
+        if self.running.load(Ordering::Relaxed) {
+            self.chan.push(s);
+        }
     }
 
     /// Flushes any buffered log records.
@@ -492,7 +511,8 @@ mod tests {
         // Create ring buffer and logger
         let parker = Parker::new();
         let chan = Arc::new(Channel::new(4, parker.unparker().to_owned()));
-        let logger = Logger::new(chan.clone(), log::LevelFilter::Info);
+        let running = Arc::new(AtomicBool::new(true));
+        let logger = Logger::new(chan.clone(), log::LevelFilter::Info, running);
         let running = Arc::new(AtomicBool::new(true));
 
         // Create appender with a Vec as writer (for testing)