#![crate_name = "logfmt_logger"]
use log::{set_boxed_logger, set_max_level, Log, Metadata, Record, SetLoggerError};
use crate::filter::{create_default_filter, Filter};
use crate::fmt::{create_default_format, Format};
pub mod filter;
pub mod fmt;
pub fn init() {
LogFmtLogger::default().init().unwrap();
}
pub struct LogFmtLogger {
format: Box<dyn Format + Send + Sync>,
filter: Box<dyn Filter + Send + Sync>,
}
impl LogFmtLogger {
pub fn new() -> Self {
LogFmtLogger::default()
}
pub fn with_filter<F: 'static>(mut self, filter: F) -> Self
where
F: Filter + Send + Sync,
{
self.filter = Box::new(filter);
self
}
pub fn with_format<F: 'static>(mut self, format: F) -> Self
where
F: Format + Send + Sync,
{
self.format = Box::new(format);
self
}
pub fn init(self) -> Result<(), SetLoggerError> {
set_max_level(self.filter.filter());
set_boxed_logger(Box::new(self))
}
}
impl Log for LogFmtLogger {
fn enabled(&self, metadata: &Metadata) -> bool {
self.filter.enabled(metadata)
}
fn log(&self, record: &Record) {
if !self.filter.matches(record) {
return;
}
self.format.write(record).unwrap()
}
fn flush(&self) {}
}
impl Default for LogFmtLogger {
fn default() -> Self {
Self {
filter: create_default_filter(),
format: create_default_format(),
}
}
}
#[cfg(test)]
mod tests {
use crate::LogFmtLogger;
#[test]
fn test_thing() {
LogFmtLogger::new()
.with_filter(crate::filter::nop::NopFilter::default())
.init()
.unwrap();
log::info!("test");
log::warn!("test");
log::error!("test");
log::debug!("test");
log::trace!("test");
}
}