mirror of https://github.com/sigoden/aichat
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.2 KiB
Rust
41 lines
1.2 KiB
Rust
use crate::config::WorkingMode;
|
|
|
|
use anyhow::Result;
|
|
use log::LevelFilter;
|
|
use simplelog::{format_description, Config as LogConfig, ConfigBuilder};
|
|
|
|
#[cfg(debug_assertions)]
|
|
pub fn setup_logger(working_mode: WorkingMode) -> Result<()> {
|
|
let config = build_config();
|
|
if working_mode == WorkingMode::Serve {
|
|
simplelog::SimpleLogger::init(LevelFilter::Debug, config)?;
|
|
} else {
|
|
let file = std::fs::File::create(crate::config::Config::local_path("debug.log")?)?;
|
|
simplelog::WriteLogger::init(LevelFilter::Debug, config, file)?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(debug_assertions))]
|
|
pub fn setup_logger(working_mode: WorkingMode) -> Result<()> {
|
|
let config = build_config();
|
|
if working_mode == WorkingMode::Serve {
|
|
simplelog::SimpleLogger::init(log::LevelFilter::Info, config)?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn build_config() -> LogConfig {
|
|
let log_filter = match std::env::var("AICHAT_LOG_FILTER") {
|
|
Ok(v) => v,
|
|
Err(_) => "aichat".into(),
|
|
};
|
|
ConfigBuilder::new()
|
|
.add_filter_allow(log_filter)
|
|
.set_time_format_custom(format_description!(
|
|
"[year]-[month]-[day]T[hour]:[minute]:[second].[subsecond digits:3]Z"
|
|
))
|
|
.set_thread_level(LevelFilter::Off)
|
|
.build()
|
|
}
|