parent
6aa503c745
commit
bb42d3b49e
3 changed files with 51 additions and 35 deletions
25
src/log.rs
25
src/log.rs
|
|
@ -1,6 +1,6 @@
|
|||
use crate::utils::ToCanonical;
|
||||
pub use log::{debug, error, info, warn, Level};
|
||||
use std::net::SocketAddr;
|
||||
pub use tracing::{debug, error, info, warn};
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct MessageLog {
|
||||
|
|
@ -95,3 +95,26 @@ impl MessageLog {
|
|||
);
|
||||
}
|
||||
}
|
||||
|
||||
pub fn init_logger() {
|
||||
use tracing_subscriber::{fmt, prelude::*, EnvFilter};
|
||||
|
||||
let format_layer = fmt::layer()
|
||||
.with_line_number(false)
|
||||
.with_thread_ids(false)
|
||||
.with_target(false)
|
||||
.with_thread_names(true)
|
||||
.with_target(true)
|
||||
.with_level(true)
|
||||
.compact();
|
||||
|
||||
// This limits the logger to emits only rpxy crate
|
||||
let level_string = std::env::var(EnvFilter::DEFAULT_ENV).unwrap_or_else(|_| "info".to_string());
|
||||
let filter_layer = EnvFilter::new(format!("{}={}", env!("CARGO_PKG_NAME"), level_string));
|
||||
// let filter_layer = EnvFilter::from_default_env();
|
||||
|
||||
tracing_subscriber::registry()
|
||||
.with(format_layer)
|
||||
.with(filter_layer)
|
||||
.init();
|
||||
}
|
||||
|
|
|
|||
17
src/main.rs
17
src/main.rs
|
|
@ -30,24 +30,11 @@ use futures::future::select_all;
|
|||
use hyper::Client;
|
||||
// use hyper_trust_dns::TrustDnsResolver;
|
||||
use rustc_hash::FxHashMap as HashMap;
|
||||
use std::{io::Write, sync::Arc};
|
||||
use std::sync::Arc;
|
||||
use tokio::time::Duration;
|
||||
|
||||
fn main() {
|
||||
// env::set_var("RUST_LOG", "info");
|
||||
env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info"))
|
||||
.format(|buf, rec| {
|
||||
let ts = buf.timestamp();
|
||||
match rec.level() {
|
||||
log::Level::Debug => {
|
||||
writeln!(buf, "{} [{}] {} ({})", ts, rec.level(), rec.args(), rec.target(),)
|
||||
}
|
||||
_ => {
|
||||
writeln!(buf, "{} [{}] {}", ts, rec.level(), rec.args(),)
|
||||
}
|
||||
}
|
||||
})
|
||||
.init();
|
||||
init_logger();
|
||||
|
||||
let mut runtime_builder = tokio::runtime::Builder::new_multi_thread();
|
||||
runtime_builder.enable_all();
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue