tui_logger/logger/
api.rs

1use std::thread;
2
3use log::LevelFilter;
4use log::SetLoggerError;
5use log::Record;
6use crate::TuiLoggerFile;
7use crate::CircularBuffer;
8
9use crate::TUI_LOGGER;
10
11// Lots of boilerplate code, so that init_logger can return two error types...
12#[derive(Debug)]
13pub enum TuiLoggerError {
14    SetLoggerError(SetLoggerError),
15    ThreadError(std::io::Error),
16}
17impl std::error::Error for TuiLoggerError {
18    fn description(&self) -> &str {
19        match self {
20            TuiLoggerError::SetLoggerError(_) => "SetLoggerError",
21            TuiLoggerError::ThreadError(_) => "ThreadError",
22        }
23    }
24    fn cause(&self) -> Option<&dyn std::error::Error> {
25        match self {
26            TuiLoggerError::SetLoggerError(_) => None,
27            TuiLoggerError::ThreadError(err) => Some(err),
28        }
29    }
30}
31impl std::fmt::Display for TuiLoggerError {
32    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
33        match self {
34            TuiLoggerError::SetLoggerError(err) => write!(f, "SetLoggerError({})", err),
35            TuiLoggerError::ThreadError(err) => write!(f, "ThreadError({})", err),
36        }
37    }
38}
39
40/// Init the logger.
41pub fn init_logger(max_level: LevelFilter) -> Result<(), TuiLoggerError> {
42    let join_handle = thread::Builder::new()
43        .name("tui-logger::move_events".into())
44        .spawn(|| {
45            let duration = std::time::Duration::from_millis(10);
46            loop {
47                thread::park_timeout(duration);
48                TUI_LOGGER.move_events();
49            }
50        })
51        .map_err(|err| TuiLoggerError::ThreadError(err))?;
52    TUI_LOGGER.hot_log.lock().mover_thread = Some(join_handle);
53    if cfg!(feature = "tracing-support") {
54        set_default_level(max_level);
55        Ok(())
56    } else {
57        log::set_max_level(max_level);
58        log::set_logger(&*TUI_LOGGER).map_err(|err| TuiLoggerError::SetLoggerError(err))
59    }
60}
61
62/// Set the depth of the hot buffer in order to avoid message loss.
63/// This is effective only after a call to move_events()
64pub fn set_hot_buffer_depth(depth: usize) {
65    TUI_LOGGER.inner.lock().hot_depth = depth;
66}
67
68/// Set the depth of the circular buffer in order to avoid message loss.
69/// This will delete all existing messages in the circular buffer.
70pub fn set_buffer_depth(depth: usize) {
71    TUI_LOGGER.inner.lock().events = CircularBuffer::new(depth);
72}
73
74/// Define filename and log formmating options for file dumping.
75pub fn set_log_file(file_options: TuiLoggerFile) {
76    TUI_LOGGER.inner.lock().dump = Some(file_options);
77}
78
79/// Set default levelfilter for unknown targets of the logger
80pub fn set_default_level(levelfilter: LevelFilter) {
81    TUI_LOGGER.hot_select.lock().default = levelfilter;
82    TUI_LOGGER.inner.lock().default = levelfilter;
83}
84
85/// Set levelfilter for a specific target in the logger
86pub fn set_level_for_target(target: &str, levelfilter: LevelFilter) {
87    let h = fxhash::hash64(&target);
88    TUI_LOGGER.inner.lock().targets.set(target, levelfilter);
89    let mut hs = TUI_LOGGER.hot_select.lock();
90    hs.hashtable.insert(h, levelfilter);
91}
92
93// Move events from the hot log to the main log
94pub fn move_events() {
95    TUI_LOGGER.move_events();
96}
97
98/// A simple `Drain` to log any event directly.
99#[derive(Default)]
100pub struct Drain;
101
102impl Drain {
103    /// Create a new Drain
104    pub fn new() -> Self {
105        Drain
106    }
107    /// Log the given record to the main tui-logger
108    pub fn log(&self, record: &Record) {
109        TUI_LOGGER.raw_log(record)
110    }
111}
tui_logger/logger/api.rs

tui_logger/logger/
api.rs
