From 6a755179feef72fdf44d21d12b1cf646e8bdb805 Mon Sep 17 00:00:00 2001 From: "Hatter Jiang@Pixelbook" Date: Sun, 13 Sep 2020 10:40:32 +0800 Subject: [PATCH] chore: add logger level checl --- Cargo.lock | 4 ++-- src/main.rs | 9 ++++++--- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index bbee0f6..76fc48b 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -232,9 +232,9 @@ dependencies = [ [[package]] name = "rust_util" -version = "0.6.13" +version = "0.6.14" source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "190763556e2faed2ba3120c04e3051001d84dbe556d397f030def64c1ae27fd2" +checksum = "7222f977acb4264fb55f1aa7cf11e09c735fe961b369aef92eea670949628498" dependencies = [ "lazy_static", "libc", diff --git a/src/main.rs b/src/main.rs index 1bd7046..785b3c7 100644 --- a/src/main.rs +++ b/src/main.rs @@ -11,6 +11,7 @@ use std::sync::mpsc::channel; use std::thread; use std::time::Duration; use clap::{ Arg, App }; +use rust_util::util_msg::{ is_logger_level_enabled, MessageType }; use rust_util::util_net::IpAddressMaskGroup; const LOCAL_ADDR: &str = "127.0.0.1"; @@ -46,9 +47,11 @@ fn main() { .unwrap_or_else(|| vec![]) ); debugging!("Allowed ip address mask list count: {}", allowed_ip_address_mask_list.ip_address_mask_group.len()); - allowed_ip_address_mask_list.ip_address_mask_group.iter().for_each(|ip| { - debugging!("- {}", ip); - }); + if is_logger_level_enabled(MessageType::DEBUG) { + allowed_ip_address_mask_list.ip_address_mask_group.iter().for_each(|ip| { + debugging!("- {}", ip); + }); + } forward(&bind_addr, local_port, &remote_host, remote_port, &allowed_ip_address_mask_list); }