write logs to file

Author: GyulyVGC

Cargo.lock

@@ -10,3 +10,5 @@ pingora-proxy = "0.8.0"
 nullnet-grpc-lib = { git = "https://github.com/GyulyVGC/nullnet-grpc.git" }
 nullnet-liberror = { git = "https://github.com/NullNet-ai/libguard.git", branch = "nullnet-liberror-print" }
 tokio = { version = "1.51.1", features = ["rt-multi-thread"] }
+gag = "1.0.0"
+chrono = "0.4.44"

Cargo.toml

@@ -8,6 +8,7 @@ use pingora_core::server::Server;
 use pingora_core::upstreams::peer::HttpPeer;
 use pingora_core::{Error, ErrorType, Result};
 use pingora_proxy::{ProxyHttp, Session};
+use std::fs::create_dir_all;
 use std::thread;
 use std::time::Instant;
 
@@ -75,6 +76,15 @@ impl ProxyHttp for NullnetProxy {
 
 #[tokio::main]
 async fn main() -> Result<(), nullnet_liberror::Error> {
+    let _gag1: gag::Redirect<std::fs::File>;
+    let _gag2: gag::Redirect<std::fs::File>;
+    if let Some((gag1, gag2)) = redirect_stdout_stderr_to_file() {
+        _gag1 = gag1;
+        _gag2 = gag2;
+    } else {
+        println!("Failed to redirect stdout and stderr to file, logs will be printed to console");
+    }
+
     let proxy_address = format!("0.0.0.0:{PROXY_PORT}");
     println!("Running Nullnet proxy at {proxy_address}\n");
 
@@ -92,3 +102,23 @@ async fn main() -> Result<(), nullnet_liberror::Error> {
     handle.join().unwrap();
     Ok(())
 }
+
+fn redirect_stdout_stderr_to_file()
+-> Option<(gag::Redirect<std::fs::File>, gag::Redirect<std::fs::File>)> {
+    let dir = "/var/log/nullnet";
+    create_dir_all(dir).handle_err(location!()).ok()?;
+    let timestamp = chrono::Utc::now().format("%Y-%m-%d_%H-%M-%S");
+    let file_path = format!("{dir}/proxy_{timestamp}.txt");
+    if let Ok(logs_file) = std::fs::OpenOptions::new()
+        .create(true)
+        .append(true)
+        .open(&file_path)
+    {
+        println!("Writing logs to '{file_path}'");
+        return Some((
+            gag::Redirect::stdout(logs_file.try_clone().ok()?).ok()?,
+            gag::Redirect::stderr(logs_file).ok()?,
+        ));
+    }
+    None
+}