log

Author: chenjunliang

Cargo.toml

@@ -13,4 +13,6 @@ rcron = "1.1.3"
 clap = { version = "3.2.16", features = ["derive"] }
 anyhow = "1.0.60"
 chrono = "0.4.21"
-rust_decimal = "1.25.0"
+rust_decimal = "1.25.0"
+log = "0.4.14"
+simple_logger = "2.1.0"

src/main.rs

@@ -29,9 +29,15 @@ fn parse_path(s: &str) -> Result<String> {
     Ok(s.into())
 }
 
+fn init_log() {
+    simple_logger::init_with_level(log::Level::Info).unwrap();
+}
+
 fn main() -> Result<()> {
     let options = Options::parse();
-    println!("Received parameters:{:?}", options);
+    log::info!("App environments:{:?}", options);
+
+    init_log();
 
     let username = libc_util::get_current_user();
     let pkg_name = env!("CARGO_PKG_NAME");

src/tasks.rs

@@ -10,13 +10,6 @@ use std::time::{Duration, SystemTime};
 fn mysql_audit_log_rotate(sched: &mut JobScheduler, path: String, max_size: u32, max_file: u32) {
     // utc time
     sched.add(Job::new("1/10 * * * * *".parse().unwrap(), move || {
-        println!(
-            "执行日志轮转任务! path:{}, max_size:{}, max_file:{}, time:{:?}",
-            path,
-            max_size,
-            max_file,
-            SystemTime::now()
-        );
         let r = fs::File::options().write(true).open(path.as_str());
         match r {
             Ok(file) => {
@@ -26,17 +19,17 @@ fn mysql_audit_log_rotate(sched: &mut JobScheduler, path: String, max_size: u32,
                 let file_size = (file_len / cf / cf).round_dp(0);
                 let file_max_size = Decimal::from(max_size);
                 if file_size >= file_max_size {
-                    println!("The file size reaches the split standard:{:?}M", file_size);
+                    log::info!("The file size reaches the split standard:{:?}M", file_size);
                     let file_path = Path::new(path.as_str());
                     let origin_file_name =
                         file_path.file_name().unwrap().to_str().unwrap().to_string();
                     let sv: Vec<&str> = origin_file_name.split(".").collect();
                     let origin_name = sv[0];
                     let origin_file_type = sv[1];
-                    println!("origin_name:{}", origin_name);
+                    log::info!("origin_name:{}", origin_name);
 
                     let parent_path = file_path.parent().unwrap();
-                    println!("parent_path:{}", parent_path.to_str().unwrap());
+                    log::info!("parent_path:{}", parent_path.to_str().unwrap());
 
                     let local: DateTime<Local> = Local::now();
                     let time_str = local.format("%Y%m%d%H%M%S").to_string();
@@ -45,7 +38,7 @@ fn mysql_audit_log_rotate(sched: &mut JobScheduler, path: String, max_size: u32,
                     let new_file_path =
                         parent_path.to_str().unwrap().to_owned() + "/" + new_file_name.as_str();
                     fs::copy(file_path, new_file_path.as_str()).unwrap();
-                    println!("Log file copied to:{}", new_file_path);
+                    log::info!("Log file copied to:{}", new_file_path);
 
                     file.set_len(0).unwrap();
 
@@ -57,20 +50,20 @@ fn mysql_audit_log_rotate(sched: &mut JobScheduler, path: String, max_size: u32,
                         .collect::<Vec<String>>();
                     files.sort();
                     if files.len() > max_file as usize {
-                        println!("The number of files exceeds the limit,start cleaning...");
+                        log::info!("The number of files exceeds the limit,start cleaning...");
                         for i in 0..(files.len() - max_file as usize) {
                             let item_file_name = files.get(i).unwrap();
                             let item_path =
                                 parent_path.to_str().unwrap().to_owned() + "/" + item_file_name;
                             fs::remove_file(item_path.as_str())
                                 .expect("Failed to clean up redundant files");
-                            println!("Delete file {}", item_path);
+                            log::info!("Delete file {}", item_path);
                         }
                     }
                 }
             }
             Err(e) => {
-                eprintln!("Read file failed[{}]：{}", path.as_str(), e.to_string());
+                log::error!("Read file failed[{}]：{}", path.as_str(), e.to_string());
             }
         }
     }));