Noratrieb/haesli
1
0
Fork 0
mirror of https://github.com/Noratrieb/haesli.git synced 2026-01-14 19:55:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

add tokio-console support

Browse source
This commit is contained in:
nora 2022-02-20 22:03:39 +01:00
parent f2195133fb
commit ab736984cc
6 changed files with 456 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

33
src/main.rs
View file

@ -7,35 +7,44 @@ use tracing::{info_span, Instrument};
#[tokio::main]
async fn main() -> Result<()> {
let mut dashboard = false;
let mut console = false;
for arg in env::args().skip(1) {
match arg.as_str() {
"--dashboard" => dashboard = true,
"--console" => console = true,
"ignore-this-clippy" => eprintln!("yes please"),
_ => {}
}
}
setup_tracing();
setup_tracing(console);
let global_data = amqp_core::GlobalData::default();
if dashboard {
let dashboard_span = info_span!("dashboard");
tokio::spawn(amqp_dashboard::dashboard(global_data.clone()).instrument(dashboard_span));
tokio::task::Builder::new()
.name("dashboard")
.spawn(amqp_dashboard::dashboard(global_data.clone()).instrument(dashboard_span));
}
amqp_transport::do_thing_i_guess(global_data).await
}
fn setup_tracing() {
tracing_subscriber::fmt()
.with_level(true)
.with_timer(tracing_subscriber::fmt::time::time())
.with_ansi(true)
.with_thread_names(true)
.with_env_filter(
std::env::var("RUST_LOG").unwrap_or_else(|_| "hyper=info,debug".to_string()),
)
.init()
fn setup_tracing(console: bool) {
if console {
console_subscriber::init();
} else {
tracing_subscriber::fmt()
.with_level(true)
.with_timer(tracing_subscriber::fmt::time::time())
.with_ansi(true)
.with_thread_names(true)
.with_env_filter(
std::env::var("RUST_LOG")
.unwrap_or_else(|_| "hyper=info,tokio=trace,runtime=trace,debug".to_string()),
)
.init();
}
}