chore(storage-scrubber): allow disable file logging

Signed-off-by: Alex Chi Z <chi@neon.tech>
neondatabase · Jul 5, 2024 · e3dd5ae · e3dd5ae
1 parent 13522fb
commit e3dd5ae
Showing 1 changed file with 33 additions and 18 deletions.
diff --git a/storage_scrubber/src/lib.rs b/storage_scrubber/src/lib.rs
@@ -242,24 +242,39 @@ impl ConsoleConfig {
     }
 }
 
-pub fn init_logging(file_name: &str) -> WorkerGuard {
-    let (file_writer, guard) =
-        tracing_appender::non_blocking(tracing_appender::rolling::never("./logs/", file_name));
-
-    let file_logs = fmt::Layer::new()
-        .with_target(false)
-        .with_ansi(false)
-        .with_writer(file_writer);
-    let stderr_logs = fmt::Layer::new()
-        .with_target(false)
-        .with_writer(std::io::stderr);
-    tracing_subscriber::registry()
-        .with(EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info")))
-        .with(file_logs)
-        .with(stderr_logs)
-        .init();
-
-    guard
+pub fn init_logging(file_name: &str) -> Option<WorkerGuard> {
+    let disable_file_logging = match std::env::var("PAGESERVER_DISABLE_FILE_LOGGING") {
+        Ok(s) => s == "1" || s.to_lowercase() == "true",
+        Err(_) => false,
+    };
+
+    if disable_file_logging {
+        let stderr_logs = fmt::Layer::new()
+            .with_target(false)
+            .with_writer(std::io::stderr);
+        tracing_subscriber::registry()
+            .with(EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info")))
+            .with(stderr_logs)
+            .init();
+        None
+    } else {
+        let (file_writer, guard) =
+            tracing_appender::non_blocking(tracing_appender::rolling::never("./logs/", file_name));
+
+        let file_logs = fmt::Layer::new()
+            .with_target(false)
+            .with_ansi(false)
+            .with_writer(file_writer);
+        let stderr_logs = fmt::Layer::new()
+            .with_target(false)
+            .with_writer(std::io::stderr);
+        tracing_subscriber::registry()
+            .with(EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info")))
+            .with(file_logs)
+            .with(stderr_logs)
+            .init();
+        Some(guard)
+    }
 }
 
 pub fn init_s3_client(bucket_region: Region) -> Client {