fix: logger issues (#524)

Signed-off-by: usamoi <[email protected]>

Cargo.toml

@@ -21,8 +21,6 @@ arrayvec.workspace = true
 bincode.workspace = true
 byteorder.workspace = true
 chrono = "0.4.38"
-csv = "1.3.0"
-env_logger = "0.11.2"
 libc.workspace = true
 log.workspace = true
 memmap2.workspace = true
@@ -72,7 +70,7 @@ bincode = "1.3.3"
 byteorder = "1.5.0"
 half = { version = "2.4.0", features = ["num-traits", "rand_distr", "serde"] }
 libc = "0.2.153"
-log = "0.4.21"
+log = { version = "0.4.21", features = ["std"] }
 memmap2 = "0.9.4"
 num-traits = "0.2.19"
 parking_lot = "0.12.1"

crates/detect/Cargo.toml

@@ -4,8 +4,6 @@ version.workspace = true
 edition.workspace = true
 
 [dependencies]
-rustix.workspace = true
-
 detect_macros = { path = "../detect_macros" }
 
 [lints]

crates/memfd/Cargo.toml

@@ -7,7 +7,5 @@ edition.workspace = true
 rand.workspace = true
 rustix.workspace = true
 
-detect = { path = "../detect" }
-
 [lints]
 workspace = true

src/bgworker/mod.rs

@@ -30,28 +30,23 @@ pub fn is_started() -> bool {
 #[pgrx::pg_guard]
 #[no_mangle]
 extern "C" fn _vectors_main(_arg: pgrx::pg_sys::Datum) {
-    {
-        let mut builder = crate::logger::VectorLogger::build();
-        #[cfg(not(debug_assertions))]
-        {
-            builder.filter_level(log::LevelFilter::Info);
-        }
-        // #[cfg(debug_assertions)]
-        {
-            builder.filter_level(log::LevelFilter::Trace);
-        }
-        builder.init();
-    }
+    // for debugging, set `RUST_LOG=trace`
+    crate::logger::Logger::new(
+        match std::env::var("RUST_LOG").as_ref().map(|x| x.as_str()) {
+            Ok("off" | "Off" | "OFF") => log::LevelFilter::Off,
+            Ok("error" | "Error" | "ERROR") => log::LevelFilter::Error,
+            Ok("warn" | "Warn" | "WARN") => log::LevelFilter::Warn,
+            Ok("info" | "Info" | "INFO") => log::LevelFilter::Info,
+            Ok("debug" | "Debug" | "DEBUG") => log::LevelFilter::Debug,
+            Ok("trace" | "Trace" | "TRACE") => log::LevelFilter::Trace,
+            _ => log::LevelFilter::Info, // default level
+        },
+    )
+    .init()
+    .expect("failed to set logger");
     std::panic::set_hook(Box::new(|info| {
-        let backtrace;
-        #[cfg(not(debug_assertions))]
-        {
-            backtrace = std::backtrace::Backtrace::capture();
-        }
-        #[cfg(debug_assertions)]
-        {
-            backtrace = std::backtrace::Backtrace::force_capture();
-        }
+        // for debugging, set `RUST_BACKTRACE=1`
+        let backtrace = std::backtrace::Backtrace::capture();
         log::error!("Panickied. Info: {:?}. Backtrace: {}.", info, backtrace);
     }));
     use service::Version;