continuwuity/src/core/log/suppress.rs
Jason Volk 5df7443437 add log suppression tool
Signed-off-by: Jason Volk <jason@zemos.net>
2024-06-13 02:16:46 +00:00

38 lines
709 B
Rust

use std::sync::Arc;
use super::EnvFilter;
use crate::Server;
pub struct Suppress {
server: Arc<Server>,
restore: EnvFilter,
}
impl Suppress {
pub fn new(server: &Arc<Server>) -> Self {
let config = &server.config.log;
Self::from_filters(server, EnvFilter::try_new(config).unwrap_or_default(), &EnvFilter::default())
}
fn from_filters(server: &Arc<Server>, restore: EnvFilter, suppress: &EnvFilter) -> Self {
server
.log
.reload
.reload(suppress)
.expect("log filter reloaded");
Self {
server: server.clone(),
restore,
}
}
}
impl Drop for Suppress {
fn drop(&mut self) {
self.server
.log
.reload
.reload(&self.restore)
.expect("log filter reloaded");
}
}