use log::{LevelFilter};
use std::time::{Duration, SystemTime};
use std::ops::{Add, Sub};
use crossbeam_utils::sync::WaitGroup;
use crate::appender::Command::CommandRecord;
use crate::date;
pub trait LogAppender: Send {
fn do_logs(&self, records: &[FastLogRecord]);
fn flush(&self) {}
}
#[derive(Clone, Debug)]
pub enum Command {
CommandRecord,
CommandExit,
CommandFlush(WaitGroup),
}
impl Command {
pub fn to_i32(&self) -> i32 {
match self {
Command::CommandRecord => { 1 }
Command::CommandExit => { 2 }
Command::CommandFlush(_) => { 3 }
}
}
}
impl PartialEq for Command {
fn eq(&self, other: &Self) -> bool {
self.to_i32().eq(&other.to_i32())
}
}
impl Eq for Command {}
#[derive(Clone, Debug)]
pub struct FastLogRecord {
pub command: Command,
pub level: log::Level,
pub target: String,
pub args: String,
pub module_path: String,
pub file: String,
pub line: Option<u32>,
pub now: SystemTime,
pub formated: String,
}
pub trait RecordFormat: Send + Sync {
fn do_format(&self, arg: &mut FastLogRecord) -> String;
}
pub struct FastLogFormat {
pub duration_zone: Duration,
pub display_line_level: log::LevelFilter,
}
impl RecordFormat for FastLogFormat {
fn do_format(&self, arg: &mut FastLogRecord) -> String {
match arg.command {
CommandRecord => {
let data;
let now = date::LogDate::from(arg.now.add(self.duration_zone));
if arg.level.to_level_filter() <= self.display_line_level {
data = format!(
"{:26} {} {} - {} {}:{}\n",
&now,
arg.level,
arg.module_path,
arg.args,
arg.file,
arg.line.unwrap_or_default()
);
} else {
data = format!(
"{:26} {} {} - {}\n",
&now, arg.level, arg.module_path, arg.args
);
}
return data;
}
Command::CommandExit => {}
Command::CommandFlush(_) => {}
}
return String::new();
}
}
impl FastLogFormat {
pub fn local_duration() -> Duration {
let utc = chrono::Utc::now().naive_utc();
let tz = chrono::Local::now().naive_local();
tz.sub(utc).to_std().unwrap_or_default()
}
pub fn new() -> FastLogFormat {
Self {
duration_zone: Self::local_duration(),
display_line_level: LevelFilter::Warn,
}
}
pub fn set_display_line_level(mut self, level: LevelFilter) -> Self {
self.display_line_level = level;
self
}
pub fn set_duration(mut self, duration: Duration) -> Self {
self.duration_zone = duration;
self
}
}