tdfischer/renderbug-bike
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

logging: implement a way to set logging levels and categories by env variables

Browse Source
This commit is contained in:
Victoria Fischer
2026-01-03 14:50:18 +01:00
parent fa57c0ce81
commit 23043a42a2
1 changed files with 28 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/logging.rs
View File

@@ -5,19 +5,19 @@ use static_cell::StaticCell;
// Provides a threadsafe serial logger
pub struct RenderbugLogger {
lock: Mutex<CriticalSectionRawMutex, ()>
lock: Mutex<CriticalSectionRawMutex, ()>,
default_level: LevelFilter
}
impl Default for RenderbugLogger {
fn default() -> Self {
Self {
lock: Mutex::new(())
lock: Mutex::new(()),
default_level: LevelFilter::Info
}
}
}
static MAX_LEVEL: LevelFilter = LevelFilter::Debug;
static LOGGER: StaticCell<RenderbugLogger> = StaticCell::new();
impl RenderbugLogger {
@@ -25,17 +25,27 @@ impl RenderbugLogger {
#[cfg(feature = "rtt")]
rtt_target::rtt_init_print!(rtt_target::ChannelMode::BlockIfFull);
let logger = LOGGER.init(Default::default());
let default_level = match option_env!("LOG_LEVEL").unwrap_or("info").to_lowercase().as_str() {
"debug" => LevelFilter::Debug,
"warn" => LevelFilter::Warn,
"trace" => LevelFilter::Trace,
"error" => LevelFilter::Error,
_ => LevelFilter::Info
};
let logger = LOGGER.init(RenderbugLogger { default_level, ..Default::default() });
unsafe {
critical_section::with(|_| {
log::set_logger_racy(logger).ok();
log::set_max_level_racy(MAX_LEVEL);
log::set_max_level_racy(default_level);
});
};
}
}
impl log::Log for RenderbugLogger {
fn enabled(&self, metadata: &Metadata) -> bool {
metadata.level() >= MAX_LEVEL
metadata.level() >= self.default_level
}
fn flush(&self) {}
@@ -58,21 +68,17 @@ impl log::Log for RenderbugLogger {
};
let filename = record.file().map_or("???", |f| {f});
let prefix = if filename.chars().nth(0).unwrap() == '/' {
if record.level() < LevelFilter::Info {
record.module_path_static().unwrap()
} else {
return;
}
} else {
filename.split('/').next().unwrap()
};
let suffix = filename.split('/').next_back().unwrap();
let crate_name = record.module_path_static().unwrap();
for enabled in option_env!("LOG_CRATES").unwrap_or("").split(",") {
if crate_name.starts_with(enabled) {
self.lock.lock(|_| {
#[cfg(feature = "rtt")]
rtt_target::rprintln!("{}{}\t{}{}/{}:{}{}\t{}{}", color, record.level(), GREY, prefix, suffix, record.line().map_or(0, |f| {f}), RESET, record.args(), RESET);
rtt_target::rprintln!("{}{}\t{}{}:{}{}\t{}{}", color, record.level(), GREY, filename, record.line().map_or(0, |f| {f}), RESET, record.args(), RESET);
println!("{color}{}\t{GREY}{} {prefix}/{suffix}:{}{RESET}\t{}{RESET}", record.level(), record.module_path_static().unwrap(), record.line().map_or(0, |f| {f}), record.args());
println!("{color}{}\t{GREY}{} {filename}:{}{RESET}\t{}{RESET}", record.level(), record.module_path_static().unwrap(), record.line().map_or(0, |f| {f}), record.args());
});
return;
}
}
}
}