387 lines
12 KiB
Rust
387 lines
12 KiB
Rust
|
use std::fmt;
|
|||
|
|
use std::io::{IsTerminal, Write};
|
|||
|
|
use std::sync::atomic::{AtomicU8, Ordering};
|
|||
|
|
use std::sync::OnceLock;
|
|||
|
|
|
|||
|
|
// 极简日志：等级存在一个原子变量里，被关掉的等级只花一次原子读——不求值参数、不做格式化。
|
|||
|
|
// 行格式：[等级 本地日期时间] [来源] 消息
|
|||
|
|
// - ERROR 加粗红、WARN 加粗橙、DEBUG 加粗不变色（只在 stderr 是终端时着色，重定向到文件时是纯文本）
|
|||
|
|
// - 来源区分 core（服务自身：启动、配置）和 pipe（由管道请求触发）
|
|||
|
|
// 等级由配置文件的 log.level 或环境变量 CHUSQL_LOG 决定。
|
|||
|
|
|
|||
|
|
/// 转义序列收尾。
|
|||
|
|
const RESET: &str = "\u{1b}[0m";
|
|||
|
|
/// 加粗红（ERROR）。
|
|||
|
|
const BOLD_RED: &str = "\u{1b}[1m\u{1b}[31m";
|
|||
|
|
/// 加粗橙（WARN）：256 色里的 208 号。
|
|||
|
|
const BOLD_ORANGE: &str = "\u{1b}[1m\u{1b}[38;5;208m";
|
|||
|
|
/// 只加粗（DEBUG）。
|
|||
|
|
const BOLD: &str = "\u{1b}[1m";
|
|||
|
|
|
|||
|
|
/// 日志等级；Off 表示全关。
|
|||
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
|
|||
|
|
pub enum Level {
|
|||
|
|
Off,
|
|||
|
|
Error,
|
|||
|
|
Warn,
|
|||
|
|
Info,
|
|||
|
|
Debug,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 日志来源：core = 服务自身，pipe = 由管道请求触发。
|
|||
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|||
|
|
pub enum Channel {
|
|||
|
|
Core,
|
|||
|
|
Pipe,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
impl Channel {
|
|||
|
|
/// 日志行里的来源标签。
|
|||
|
|
fn tag(self) -> &'static str {
|
|||
|
|
match self {
|
|||
|
|
Channel::Core => "core",
|
|||
|
|
Channel::Pipe => "pipe",
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
impl Level {
|
|||
|
|
/// 解析等级名（大小写不敏感）。
|
|||
|
|
pub fn parse(s: &str) -> Option<Level> {
|
|||
|
|
match s.trim().to_ascii_lowercase().as_str() {
|
|||
|
|
"off" => Some(Level::Off),
|
|||
|
|
"error" => Some(Level::Error),
|
|||
|
|
"warn" | "warning" => Some(Level::Warn),
|
|||
|
|
"info" => Some(Level::Info),
|
|||
|
|
"debug" => Some(Level::Debug),
|
|||
|
|
_ => None,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 等级名（写日志和写配置说明用）。
|
|||
|
|
pub fn name(self) -> &'static str {
|
|||
|
|
match self {
|
|||
|
|
Level::Off => "off",
|
|||
|
|
Level::Error => "error",
|
|||
|
|
Level::Warn => "warn",
|
|||
|
|
Level::Info => "info",
|
|||
|
|
Level::Debug => "debug",
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 日志行里的标签，固定 5 字符宽，方便对齐。
|
|||
|
|
fn tag(self) -> &'static str {
|
|||
|
|
match self {
|
|||
|
|
Level::Error => "ERROR",
|
|||
|
|
Level::Warn => "WARN ",
|
|||
|
|
Level::Info => "INFO ",
|
|||
|
|
_ => "DEBUG",
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 这个等级要不要着色（INFO 不着色）。
|
|||
|
|
fn color(self) -> &'static str {
|
|||
|
|
match self {
|
|||
|
|
Level::Error => BOLD_RED,
|
|||
|
|
Level::Warn => BOLD_ORANGE,
|
|||
|
|
Level::Debug => BOLD,
|
|||
|
|
_ => "",
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 当前等级，默认 info。
|
|||
|
|
static LEVEL: AtomicU8 = AtomicU8::new(Level::Info as u8);
|
|||
|
|
|
|||
|
|
/// stderr 是不是终端（决定要不要着色），只问一次。
|
|||
|
|
static COLOR: OnceLock<bool> = OnceLock::new();
|
|||
|
|
|
|||
|
|
/// 设置全局等级。
|
|||
|
|
pub fn set_level(level: Level) {
|
|||
|
|
LEVEL.store(level as u8, Ordering::Relaxed);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 这个等级要不要输出——热路径上只走这一步（一次原子读 + 比较）。
|
|||
|
|
#[inline]
|
|||
|
|
pub fn enabled(level: Level) -> bool {
|
|||
|
|
level as u8 <= LEVEL.load(Ordering::Relaxed)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 启动时调一次：Windows 上给 stderr 打开 VT 处理，让 ANSI 颜色能被认识。
|
|||
|
|
pub fn init() {
|
|||
|
|
enable_vt();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 写出一行；只在 enabled 为真时调用。
|
|||
|
|
pub fn write(channel: Channel, level: Level, args: fmt::Arguments) {
|
|||
|
|
let msg = args.to_string();
|
|||
|
|
let line = render(level, channel, color_enabled(), &now(), &msg);
|
|||
|
|
let stderr = std::io::stderr();
|
|||
|
|
let mut out = stderr.lock();
|
|||
|
|
let _ = writeln!(out, "{}", line);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// stderr 是终端才着色；CHUSQL_LOG_COLOR=always / never 可以强制开关。
|
|||
|
|
fn color_enabled() -> bool {
|
|||
|
|
*COLOR.get_or_init(|| match std::env::var("CHUSQL_LOG_COLOR").as_deref().map(str::trim) {
|
|||
|
|
Ok("always") | Ok("yes") | Ok("1") => true,
|
|||
|
|
Ok("never") | Ok("no") | Ok("0") => false,
|
|||
|
|
_ => std::io::stderr().is_terminal(),
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 拼一行：[等级 时间] [来源] 消息。纯函数，方便测试。
|
|||
|
|
fn render(level: Level, channel: Channel, color: bool, stamp: &Stamp, msg: &str) -> String {
|
|||
|
|
let tag = level.tag();
|
|||
|
|
let paint = level.color();
|
|||
|
|
let level_text = if color && !paint.is_empty() {
|
|||
|
|
format!("{}{}{}", paint, tag, RESET)
|
|||
|
|
} else {
|
|||
|
|
tag.to_string()
|
|||
|
|
};
|
|||
|
|
format!("[{} {}] [{}] {}", level_text, stamp, channel.tag(), msg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 一个本地日期时间。
|
|||
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|||
|
|
pub struct Stamp {
|
|||
|
|
pub year: u16,
|
|||
|
|
pub month: u16,
|
|||
|
|
pub day: u16,
|
|||
|
|
pub hour: u16,
|
|||
|
|
pub minute: u16,
|
|||
|
|
pub second: u16,
|
|||
|
|
pub milli: u16,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
impl fmt::Display for Stamp {
|
|||
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|||
|
|
write!(
|
|||
|
|
f,
|
|||
|
|
"{:04}-{:02}-{:02} {:02}:{:02}:{:02}.{:03}",
|
|||
|
|
self.year, self.month, self.day, self.hour, self.minute, self.second, self.milli
|
|||
|
|
)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 当前本地时间：Windows 上走 kernel32::GetLocalTime。
|
|||
|
|
#[cfg(windows)]
|
|||
|
|
fn now() -> Stamp {
|
|||
|
|
#[repr(C)]
|
|||
|
|
struct SystemTime {
|
|||
|
|
year: u16,
|
|||
|
|
month: u16,
|
|||
|
|
day_of_week: u16,
|
|||
|
|
day: u16,
|
|||
|
|
hour: u16,
|
|||
|
|
minute: u16,
|
|||
|
|
second: u16,
|
|||
|
|
milliseconds: u16,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[link(name = "kernel32")]
|
|||
|
|
unsafe extern "system" {
|
|||
|
|
fn GetLocalTime(out: *mut SystemTime);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let mut t = SystemTime {
|
|||
|
|
year: 0,
|
|||
|
|
month: 0,
|
|||
|
|
day_of_week: 0,
|
|||
|
|
day: 0,
|
|||
|
|
hour: 0,
|
|||
|
|
minute: 0,
|
|||
|
|
second: 0,
|
|||
|
|
milliseconds: 0,
|
|||
|
|
};
|
|||
|
|
unsafe { GetLocalTime(&mut t) };
|
|||
|
|
Stamp {
|
|||
|
|
year: t.year,
|
|||
|
|
month: t.month,
|
|||
|
|
day: t.day,
|
|||
|
|
hour: t.hour,
|
|||
|
|
minute: t.minute,
|
|||
|
|
second: t.second,
|
|||
|
|
milli: t.milliseconds,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 当前时间：其他平台没有取本地时区的零依赖办法，退回 UTC。
|
|||
|
|
#[cfg(not(windows))]
|
|||
|
|
fn now() -> Stamp {
|
|||
|
|
let d = std::time::SystemTime::now()
|
|||
|
|
.duration_since(std::time::UNIX_EPOCH)
|
|||
|
|
.unwrap_or_default();
|
|||
|
|
let secs = d.as_secs();
|
|||
|
|
let (year, month, day) = civil_from_days((secs / 86_400) as i64);
|
|||
|
|
Stamp {
|
|||
|
|
year: year as u16,
|
|||
|
|
month: month as u16,
|
|||
|
|
day: day as u16,
|
|||
|
|
hour: ((secs % 86_400) / 3600) as u16,
|
|||
|
|
minute: ((secs % 3_600) / 60) as u16,
|
|||
|
|
second: (secs % 60) as u16,
|
|||
|
|
milli: d.subsec_millis() as u16,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 把"距 1970-01-01 的天数"换算成 (年, 月, 日)：非 Windows 平台算时间用，单元测试也走这里。
|
|||
|
|
pub fn civil_from_days(days: i64) -> (i64, u16, u16) {
|
|||
|
|
let z = days + 719_468;
|
|||
|
|
let era = if z >= 0 { z } else { z - 146_096 } / 146_097;
|
|||
|
|
let doe = z - era * 146_097;
|
|||
|
|
let yoe = (doe - doe / 1_460 + doe / 36_524 - doe / 146_096) / 365;
|
|||
|
|
let y = yoe + era * 400;
|
|||
|
|
let doy = doe - (365 * yoe + yoe / 4 - yoe / 100);
|
|||
|
|
let mp = (5 * doy + 2) / 153;
|
|||
|
|
let day = (doy - (153 * mp + 2) / 5 + 1) as u16;
|
|||
|
|
let month = if mp < 10 { mp + 3 } else { mp - 9 } as u16;
|
|||
|
|
(y + i64::from(month <= 2), month, day)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// Windows：打开 stderr 的 VT 处理。
|
|||
|
|
#[cfg(windows)]
|
|||
|
|
fn enable_vt() {
|
|||
|
|
#[link(name = "kernel32")]
|
|||
|
|
unsafe extern "system" {
|
|||
|
|
fn GetStdHandle(which: u32) -> *mut core::ffi::c_void;
|
|||
|
|
fn GetConsoleMode(handle: *mut core::ffi::c_void, mode: *mut u32) -> i32;
|
|||
|
|
fn SetConsoleMode(handle: *mut core::ffi::c_void, mode: u32) -> i32;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// STD_ERROR_HANDLE：-12 的无符号写法。
|
|||
|
|
const STD_ERROR_HANDLE: u32 = 0xFFFF_FFF4;
|
|||
|
|
/// 让控制台认识 ANSI 转义序列。
|
|||
|
|
const ENABLE_VIRTUAL_TERMINAL_PROCESSING: u32 = 0x0004;
|
|||
|
|
|
|||
|
|
unsafe {
|
|||
|
|
let handle = GetStdHandle(STD_ERROR_HANDLE);
|
|||
|
|
let mut mode = 0u32;
|
|||
|
|
if GetConsoleMode(handle, &mut mode) != 0 {
|
|||
|
|
let _ = SetConsoleMode(handle, mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 非 Windows：不用做任何事。
|
|||
|
|
#[cfg(not(windows))]
|
|||
|
|
fn enable_vt() {}
|
|||
|
|
|
|||
|
|
/// 打一条 debug 日志（关掉时只花一次原子读）。
|
|||
|
|
#[macro_export]
|
|||
|
|
macro_rules! log_debug {
|
|||
|
|
(core, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Core, $crate::log::Level::Debug, $($arg)*)
|
|||
|
|
};
|
|||
|
|
(pipe, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Pipe, $crate::log::Level::Debug, $($arg)*)
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 打一条 info 日志。
|
|||
|
|
#[macro_export]
|
|||
|
|
macro_rules! log_info {
|
|||
|
|
(core, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Core, $crate::log::Level::Info, $($arg)*)
|
|||
|
|
};
|
|||
|
|
(pipe, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Pipe, $crate::log::Level::Info, $($arg)*)
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 打一条 warn 日志。
|
|||
|
|
#[macro_export]
|
|||
|
|
macro_rules! log_warn {
|
|||
|
|
(core, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Core, $crate::log::Level::Warn, $($arg)*)
|
|||
|
|
};
|
|||
|
|
(pipe, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Pipe, $crate::log::Level::Warn, $($arg)*)
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 打一条 error 日志。
|
|||
|
|
#[macro_export]
|
|||
|
|
macro_rules! log_error {
|
|||
|
|
(core, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Core, $crate::log::Level::Error, $($arg)*)
|
|||
|
|
};
|
|||
|
|
(pipe, $($arg:tt)*) => {
|
|||
|
|
$crate::log_line!($crate::log::Channel::Pipe, $crate::log::Level::Error, $($arg)*)
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 上面四个宏的公共部分：先看等级，再看要不要真的写。内部用，别直接调。
|
|||
|
|
#[doc(hidden)]
|
|||
|
|
#[macro_export]
|
|||
|
|
macro_rules! log_line {
|
|||
|
|
($chan:expr, $lvl:expr, $($arg:tt)*) => {
|
|||
|
|
if $crate::log::enabled($lvl) {
|
|||
|
|
$crate::log::write($chan, $lvl, format_args!($($arg)*))
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#[cfg(test)]
|
|||
|
|
mod tests {
|
|||
|
|
use super::*;
|
|||
|
|
|
|||
|
|
/// 天数换算日期：几个已知点 + 两次闰年的 2 月 29 日。
|
|||
|
|
#[test]
|
|||
|
|
fn civil_from_days_known_dates() {
|
|||
|
|
assert_eq!(civil_from_days(0), (1970, 1, 1));
|
|||
|
|
assert_eq!(civil_from_days(10_957), (2000, 1, 1));
|
|||
|
|
assert_eq!(civil_from_days(11_016), (2000, 2, 29));
|
|||
|
|
assert_eq!(civil_from_days(19_723), (2024, 1, 1));
|
|||
|
|
assert_eq!(civil_from_days(19_782), (2024, 2, 29));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 行格式：等级 + 本地日期时间 + 来源 + 消息，等级该粗的粗、该色的色。
|
|||
|
|
#[test]
|
|||
|
|
fn render_has_time_channel_and_colors() {
|
|||
|
|
let stamp = Stamp {
|
|||
|
|
year: 2026,
|
|||
|
|
month: 9,
|
|||
|
|
day: 24,
|
|||
|
|
hour: 13,
|
|||
|
|
minute: 45,
|
|||
|
|
second: 12,
|
|||
|
|
milli: 130,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
assert_eq!(
|
|||
|
|
render(Level::Error, Channel::Pipe, true, &stamp, "boom"),
|
|||
|
|
"[\u{1b}[1m\u{1b}[31mERROR\u{1b}[0m 2026-09-24 13:45:12.130] [pipe] boom"
|
|||
|
|
);
|
|||
|
|
assert_eq!(
|
|||
|
|
render(Level::Warn, Channel::Core, true, &stamp, "careful"),
|
|||
|
|
"[\u{1b}[1m\u{1b}[38;5;208mWARN \u{1b}[0m 2026-09-24 13:45:12.130] [core] careful"
|
|||
|
|
);
|
|||
|
|
assert_eq!(
|
|||
|
|
render(Level::Debug, Channel::Pipe, true, &stamp, "detail"),
|
|||
|
|
"[\u{1b}[1mDEBUG\u{1b}[0m 2026-09-24 13:45:12.130] [pipe] detail"
|
|||
|
|
);
|
|||
|
|
assert_eq!(
|
|||
|
|
render(Level::Info, Channel::Core, true, &stamp, "hello"),
|
|||
|
|
"[INFO 2026-09-24 13:45:12.130] [core] hello"
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// 关掉颜色时整行必须是纯文本（重定向到文件时用）。
|
|||
|
|
#[test]
|
|||
|
|
fn render_without_color_is_plain() {
|
|||
|
|
let stamp = Stamp {
|
|||
|
|
year: 2026,
|
|||
|
|
month: 9,
|
|||
|
|
day: 24,
|
|||
|
|
hour: 1,
|
|||
|
|
minute: 2,
|
|||
|
|
second: 3,
|
|||
|
|
milli: 4,
|
|||
|
|
};
|
|||
|
|
assert_eq!(
|
|||
|
|
render(Level::Error, Channel::Core, false, &stamp, "boom"),
|
|||
|
|
"[ERROR 2026-09-24 01:02:03.004] [core] boom"
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
}
|