summaryrefslogtreecommitdiffstats
path: root/src/logger.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-07-21 17:24:04 +0800
committerGitHub <noreply@github.com>2024-07-21 17:24:04 +0800
commit341f2d994295d1b88bb849d74893d62df610150c (patch)
tree565a3631fa6fa3e3fa7f8f2d7e8889a69175536b /src/logger.rs
parent17fd7b326097d43a20c7332031bc2f250dd8d4d0 (diff)
downloadaichat-341f2d994295d1b88bb849d74893d62df610150c.tar.gz
feat: enhance logger (#731)
Diffstat (limited to 'src/logger.rs')
-rw-r--r--src/logger.rs45
1 files changed, 0 insertions, 45 deletions
diff --git a/src/logger.rs b/src/logger.rs
deleted file mode 100644
index 25d6382..0000000
--- a/src/logger.rs
+++ /dev/null
@@ -1,45 +0,0 @@
-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 is_serve = working_mode.is_serve();
- let config = build_config(is_serve);
- if is_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<()> {
- if working_mode.is_serve() {
- let config = build_config(true);
- simplelog::SimpleLogger::init(log::LevelFilter::Info, config)?;
- }
- Ok(())
-}
-
-fn build_config(is_serve: bool) -> LogConfig {
- let log_filter = if is_serve {
- "aichat::serve".into()
- } else {
- 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()
-}