refactor log

libs/hbb_common/Cargo.toml

@@ -7,6 +7,7 @@ edition = "2018"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+flexi_logger = { version = "0.22", features = ["async", "use_chrono_for_offset"] }
 protobuf = { version = "3.1", features = ["with-bytes"] }
 tokio = { version = "1.20", features = ["full"] }
 tokio-util = { version = "0.7", features = ["full"] }

libs/hbb_common/src/lib.rs

@@ -39,8 +39,8 @@ pub use tokio_socks::IntoTargetAddr;
 pub use tokio_socks::TargetAddr;
 pub mod password_security;
 pub use chrono;
-pub use libc;
 pub use directories_next;
+pub use libc;
 pub mod keyboard;
 pub use sysinfo;
 
@@ -312,6 +312,44 @@ pub fn is_domain_port_str(id: &str) -> bool {
     .is_match(id)
 }
 
+pub fn init_log(_is_async: bool, _name: &str) -> Option<flexi_logger::LoggerHandle> {
+    #[cfg(debug_assertions)]
+    {
+        use env_logger::*;
+        init_from_env(Env::default().filter_or(DEFAULT_FILTER_ENV, "info"));
+        None
+    }
+    #[cfg(not(debug_assertions))]
+    {
+        // https://docs.rs/flexi_logger/latest/flexi_logger/error_info/index.html#write
+        // though async logger more efficient, but it also causes more problems, disable it for now
+        let mut logger_holder: Option<flexi_logger::LoggerHandle> = None;
+        let mut path = config::Config::log_path();
+        if !_name.is_empty() {
+            path.push(_name);
+        }
+        use flexi_logger::*;
+        if let Ok(x) = Logger::try_with_env_or_str("debug") {
+            logger_holder = x
+                .log_to_file(FileSpec::default().directory(path))
+                .write_mode(if _is_async {
+                    WriteMode::Async
+                } else {
+                    WriteMode::Direct
+                })
+                .format(opt_format)
+                .rotate(
+                    Criterion::Age(Age::Day),
+                    Naming::Timestamps,
+                    Cleanup::KeepLogFiles(6),
+                )
+                .start()
+                .ok();
+        }
+        logger_holder
+    }
+}
+
 #[cfg(test)]
 mod test {
     use super::*;