use log::Record;
#[cfg(feature = "config_parsing")]
use serde::de;
#[cfg(feature = "config_parsing")]
use serde_value::Value;
#[cfg(feature = "config_parsing")]
use std::collections::BTreeMap;
use std::fmt;
#[cfg(feature = "config_parsing")]
use crate::config::Deserializable;
#[cfg(feature = "threshold_filter")]
pub mod threshold;
pub trait Filter: fmt::Debug + Send + Sync + 'static {
fn filter(&self, record: &Record) -> Response;
}
#[cfg(feature = "config_parsing")]
impl Deserializable for dyn Filter {
fn name() -> &'static str {
"filter"
}
}
#[derive(PartialEq, Debug)]
pub enum Response {
Accept,
Neutral,
Reject,
}
#[cfg(feature = "config_parsing")]
#[derive(Clone, Eq, PartialEq, Hash, Debug)]
pub struct FilterConfig {
pub kind: String,
pub config: Value,
}
#[cfg(feature = "config_parsing")]
impl<'de> de::Deserialize<'de> for FilterConfig {
fn deserialize<D>(d: D) -> Result<FilterConfig, D::Error>
where
D: de::Deserializer<'de>,
{
let mut map = BTreeMap::<Value, Value>::deserialize(d)?;
let kind = match map.remove(&Value::String("kind".to_owned())) {
Some(kind) => kind.deserialize_into().map_err(|e| e.to_error())?,
None => return Err(de::Error::missing_field("kind")),
};
Ok(FilterConfig {
kind,
config: Value::Map(map),
})
}
}
#[cfg(test)]
mod test {
#[cfg(feature = "config_parsing")]
use super::*;
#[cfg(feature = "config_parsing")]
use serde_test::{assert_de_tokens, assert_de_tokens_error, Token};
#[test]
#[cfg(feature = "config_parsing")]
fn test_cfg_deserializer() {
let filter = FilterConfig {
kind: "threshold".to_owned(),
config: Value::Map({
let mut map = BTreeMap::new();
map.insert(
Value::String("level".to_owned()),
Value::String("error".to_owned()),
);
map
}),
};
let mut cfg = vec![
Token::Struct {
name: "FilterConfig",
len: 2,
},
Token::Str("kind"),
Token::Str("threshold"),
Token::Str("level"),
Token::Str("error"),
Token::StructEnd,
];
assert_de_tokens(&filter, &cfg);
cfg[1] = Token::Str("knd");
assert_de_tokens_error::<FilterConfig>(&cfg, "missing field `kind`");
}
}