mk-dl-bot_legacy/src/bot/log.rs

46 lines
1.4 KiB
Rust

use std::io;
use tracing::{level_filters::LevelFilter, Subscriber};
use tracing_appender::rolling::{RollingFileAppender, Rotation};
use tracing_subscriber::{fmt, layer::SubscriberExt, layer::Filter, prelude::*};
use super::util::VAR_LOG;
// A layer filter to prevent polling timeout errors from clogging logs
struct TeloxideNoiseFilter {}
impl<S: Subscriber> Filter<S> for TeloxideNoiseFilter {
fn enabled(&self, meta: &tracing::Metadata<'_>, _: &tracing_subscriber::layer::Context<'_,S>) -> bool {
if let Some(module_path) = meta.module_path() {
if module_path == "teloxide::error_handlers" {
false
} else {
true
}
} else {
true
}
}
}
pub fn log_init() {
let log_appender = RollingFileAppender::builder()
.rotation(Rotation::DAILY)
.filename_prefix("mk-dl-bot.log")
.max_log_files(7)
.build(VAR_LOG)
.unwrap();
let stderr_layer = fmt::layer()
.with_writer(io::stderr)
.with_filter(TeloxideNoiseFilter {})
.with_filter(LevelFilter::ERROR);
let file_layer = fmt::layer()
.with_writer(log_appender)
.with_filter(TeloxideNoiseFilter {})
.with_filter(LevelFilter::INFO);
tracing_subscriber::registry()
.with(stderr_layer)
.with(file_layer)
.init();
}