mirror of
https://github.com/chatmail/core.git
synced 2026-04-28 19:06:35 +03:00
Very basic logging infrastructure
We can open a logfile and write things to it with a callsite.
This commit is contained in:
121
src/log.rs
121
src/log.rs
@@ -1,4 +1,89 @@
|
||||
//! # Logging macros
|
||||
//! # Logging support
|
||||
|
||||
use std::fmt;
|
||||
use std::io::prelude::*;
|
||||
use std::path::PathBuf;
|
||||
|
||||
/// A logger for a [Context].
|
||||
#[derive(Debug)]
|
||||
pub struct Logger {
|
||||
logdir: PathBuf,
|
||||
logfile: std::ffi::OsString,
|
||||
file_handle: std::fs::File,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||
pub enum LogLevel {
|
||||
Info,
|
||||
Warning,
|
||||
Error,
|
||||
}
|
||||
|
||||
impl fmt::Display for LogLevel {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
match self {
|
||||
LogLevel::Info => write!(f, "I"),
|
||||
LogLevel::Warning => write!(f, "W"),
|
||||
LogLevel::Error => write!(f, "E"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||||
pub struct Callsite<'a> {
|
||||
pub module: &'a str,
|
||||
pub file: &'a str,
|
||||
pub line: u32,
|
||||
}
|
||||
|
||||
impl fmt::Display for Callsite<'_> {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
write!(f, "{}:{} in {}", self.file, self.line, self.module)
|
||||
}
|
||||
}
|
||||
|
||||
impl Logger {
|
||||
pub fn new(logdir: PathBuf) -> Result<Logger, std::io::Error> {
|
||||
let fname = format!(
|
||||
"{}.log",
|
||||
chrono::offset::Utc::now().to_rfc3339_opts(chrono::SecondsFormat::Secs, true)
|
||||
);
|
||||
let file = std::fs::OpenOptions::new()
|
||||
.write(true)
|
||||
.create(true)
|
||||
.truncate(true)
|
||||
.open(logdir.join(&fname))?;
|
||||
Ok(Logger {
|
||||
logdir,
|
||||
logfile: fname.into(),
|
||||
file_handle: file,
|
||||
})
|
||||
}
|
||||
|
||||
pub fn log(
|
||||
&mut self,
|
||||
level: LogLevel,
|
||||
callsite: Callsite,
|
||||
msg: &str,
|
||||
) -> Result<(), std::io::Error> {
|
||||
write!(&mut self.file_handle, "{} [{}]: {}\n", level, callsite, msg)
|
||||
}
|
||||
|
||||
pub fn flush(&mut self) -> Result<(), std::io::Error> {
|
||||
self.file_handle.flush()
|
||||
}
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! callsite {
|
||||
() => {
|
||||
$crate::log::Callsite {
|
||||
module: module_path!(),
|
||||
file: file!(),
|
||||
line: line!(),
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! info {
|
||||
@@ -39,3 +124,37 @@ macro_rules! emit_event {
|
||||
$ctx.call_cb($event);
|
||||
};
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_basic_logging() {
|
||||
let tmp = tempfile::tempdir().unwrap();
|
||||
let dir = tmp.path();
|
||||
let mut logger = Logger::new(dir.to_path_buf()).unwrap();
|
||||
logger.log(LogLevel::Info, callsite!(), "foo").unwrap();
|
||||
logger.log(LogLevel::Warning, callsite!(), "bar").unwrap();
|
||||
logger.log(LogLevel::Error, callsite!(), "baz").unwrap();
|
||||
logger.flush();
|
||||
let log = std::fs::read_to_string(logger.logdir.join(logger.logfile)).unwrap();
|
||||
println!("{}", log);
|
||||
let lines: Vec<&str> = log.lines().collect();
|
||||
|
||||
assert!(lines[0].starts_with("I"));
|
||||
assert!(lines[0].contains("src/log.rs"));
|
||||
assert!(lines[0].contains("deltachat::log::tests"));
|
||||
assert!(lines[0].contains("foo"));
|
||||
|
||||
assert!(lines[1].starts_with("W"));
|
||||
assert!(lines[1].contains("src/log.rs"));
|
||||
assert!(lines[1].contains("deltachat::log::tests"));
|
||||
assert!(lines[1].contains("bar"));
|
||||
|
||||
assert!(lines[2].starts_with("E"));
|
||||
assert!(lines[2].contains("src/log.rs"));
|
||||
assert!(lines[2].contains("deltachat::log::tests"));
|
||||
assert!(lines[2].contains("baz"));
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user