captains_log/
console_impl.rs

1use crate::{
2    config::{LogFormat, SinkConfigTrait},
3    env::EnvVarDefault,
4    log_impl::{LogSink, LogSinkTrait},
5    time::Timer,
6};
7use log::{Level, Record};
8use std::hash::{Hash, Hasher};
9use std::path::Path;
10use std::str::FromStr;
11
12/// Log config for output to console
13///
14/// # Example
15///
16/// source of [crate::recipe::console_logger()]
17///
18/// ``` rust
19/// use captains_log::*;
20///
21/// pub fn console_logger(target: ConsoleTarget, max_level: Level) -> Builder {
22///     let console_config = LogConsole::new(target, max_level, recipe::LOG_FORMAT_DEBUG);
23///     return Builder::default().console(console_config);
24/// }
25/// ```
26#[derive(Hash)]
27pub struct LogConsole {
28    pub target: ConsoleTarget,
29
30    /// max log level in this file
31    pub level: Level,
32
33    pub format: LogFormat,
34}
35
36impl LogConsole {
37    pub fn new(target: ConsoleTarget, level: Level, format: LogFormat) -> Self {
38        Self { target, level, format }
39    }
40}
41
42#[derive(Copy, Clone, Debug, Hash, PartialEq)]
43#[repr(u8)]
44pub enum ConsoleTarget {
45    Stdout = 1,
46    Stderr = 2,
47}
48
49impl FromStr for ConsoleTarget {
50    type Err = ();
51
52    /// accepts case-insensitive: stdout, stderr, out, err, 1, 2
53    fn from_str(s: &str) -> Result<Self, ()> {
54        let v = s.to_lowercase();
55        match v.as_str() {
56            "stdout" => Ok(ConsoleTarget::Stdout),
57            "stderr" => Ok(ConsoleTarget::Stderr),
58            "out" => Ok(ConsoleTarget::Stdout),
59            "err" => Ok(ConsoleTarget::Stderr),
60            "1" => Ok(ConsoleTarget::Stdout),
61            "2" => Ok(ConsoleTarget::Stderr),
62            _ => Err(()),
63        }
64    }
65}
66
67// Tried to impl blanket trait T: FromStr, rust reports conflict with
68// - impl<T, U> Into<U> for T where U: From<T>;
69crate::impl_from_env!(ConsoleTarget);
70
71impl SinkConfigTrait for LogConsole {
72    fn get_level(&self) -> Level {
73        self.level
74    }
75
76    fn get_file_path(&self) -> Option<Box<Path>> {
77        None
78    }
79
80    fn write_hash(&self, hasher: &mut Box<dyn Hasher>) {
81        self.hash(hasher);
82        hasher.write(b"LogConsole");
83    }
84
85    fn build(&self) -> LogSink {
86        LogSink::Console(LogSinkConsole::new(self))
87    }
88}
89
90pub(crate) struct LogSinkConsole {
91    target_fd: libc::c_int,
92    max_level: Level,
93    formatter: LogFormat,
94}
95
96impl LogSinkConsole {
97    fn new(config: &LogConsole) -> Self {
98        Self {
99            target_fd: config.target as i32,
100            max_level: config.level,
101            formatter: config.format.clone(),
102        }
103    }
104}
105
106impl LogSinkTrait for LogSinkConsole {
107    fn reopen(&self) -> std::io::Result<()> {
108        Ok(())
109    }
110
111    #[inline(always)]
112    fn log(&self, now: &Timer, r: &Record) {
113        if r.level() <= self.max_level {
114            let buf = self.formatter.process(now, r);
115            unsafe {
116                let _ = libc::write(self.target_fd, buf.as_ptr() as *const libc::c_void, buf.len());
117            }
118        }
119    }
120
121    #[inline(always)]
122    fn flush(&self) {}
123}
captains_log/console_impl.rs

captains_log/
console_impl.rs
