Lucas Schwiderski
fa8764984f
Keep the flexibility for dev, but in prod drop stdout and restrict the log view to stuff useful to the user.
65 lines
1.9 KiB
Rust
65 lines
1.9 KiB
Rust
use tokio::sync::mpsc::UnboundedSender;
|
|
use tracing_error::ErrorLayer;
|
|
use tracing_subscriber::filter::FilterFn;
|
|
use tracing_subscriber::fmt;
|
|
use tracing_subscriber::fmt::format::debug_fn;
|
|
use tracing_subscriber::layer::SubscriberExt;
|
|
use tracing_subscriber::prelude::*;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
pub struct ChannelWriter {
|
|
tx: UnboundedSender<String>,
|
|
}
|
|
|
|
impl ChannelWriter {
|
|
pub fn new(tx: UnboundedSender<String>) -> Self {
|
|
Self { tx }
|
|
}
|
|
}
|
|
|
|
impl std::io::Write for ChannelWriter {
|
|
fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
|
|
let tx = self.tx.clone();
|
|
let string = String::from_utf8_lossy(buf).to_string();
|
|
|
|
// The `send` errors when the receiving end has closed.
|
|
// But there's not much we can do at that point, so we just ignore it.
|
|
let _ = tx.send(string);
|
|
|
|
Ok(buf.len())
|
|
}
|
|
|
|
fn flush(&mut self) -> std::io::Result<()> {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
pub fn create_tracing_subscriber(tx: UnboundedSender<String>) {
|
|
let env_layer = if cfg!(debug_assertions) {
|
|
EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("info"))
|
|
} else {
|
|
EnvFilter::new("error,dtmm=info")
|
|
};
|
|
|
|
let stdout_layer = if cfg!(debug_assertions) {
|
|
let layer = fmt::layer().pretty();
|
|
Some(layer)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let channel_layer = fmt::layer()
|
|
// TODO: Re-enable and implement a formatter for the Druid widget
|
|
.with_ansi(false)
|
|
.event_format(dtmt_shared::Formatter)
|
|
.fmt_fields(debug_fn(dtmt_shared::format_fields))
|
|
.with_writer(move || ChannelWriter::new(tx.clone()))
|
|
.with_filter(FilterFn::new(dtmt_shared::filter_fields));
|
|
|
|
tracing_subscriber::registry()
|
|
.with(env_layer)
|
|
.with(channel_layer)
|
|
.with(stdout_layer)
|
|
.with(ErrorLayer::new(fmt::format::Pretty::default()))
|
|
.init();
|
|
}
|