1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
use std::sync::Arc;
use crate::filters::{CreateFilterArgs, Error, FilterInstance, FilterMap, FilterSet};
#[derive(Clone, Default)]
pub struct FilterRegistry {
registry: Arc<FilterMap>,
}
impl FilterRegistry {
pub fn new(factories: FilterSet) -> Self {
Self {
registry: Arc::new(
factories
.into_iter()
.map(|factory| (factory.name(), factory))
.collect(),
),
}
}
pub fn get(&self, key: &str, args: CreateFilterArgs) -> Result<FilterInstance, Error> {
match self.registry.get(key).map(|p| p.create_filter(args)) {
None => Err(Error::NotFound(key.to_owned())),
Some(filter) => filter,
}
}
}
#[cfg(test)]
mod tests {
use std::net::{IpAddr, Ipv4Addr, SocketAddr};
use crate::test_utils::{logger, new_registry};
use super::*;
use crate::endpoint::{Endpoint, Endpoints};
use crate::filters::{Filter, ReadContext, ReadResponse, WriteContext, WriteResponse};
use prometheus::Registry;
struct TestFilter {}
impl Filter for TestFilter {
fn read(&self, _: ReadContext) -> Option<ReadResponse> {
None
}
fn write(&self, _: WriteContext) -> Option<WriteResponse> {
None
}
}
#[test]
fn insert_and_get() {
let reg = new_registry(&logger());
match reg.get(
&String::from("not.found"),
CreateFilterArgs::fixed(Registry::default(), None),
) {
Ok(_) => unreachable!("should not be filter"),
Err(err) => assert_eq!(Error::NotFound("not.found".to_string()), err),
};
assert!(reg
.get(
&String::from("TestFilter"),
CreateFilterArgs::fixed(Registry::default(), None)
)
.is_ok());
let filter = reg
.get(
&String::from("TestFilter"),
CreateFilterArgs::fixed(Registry::default(), None),
)
.unwrap()
.filter;
let addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)), 8080);
let endpoint = Endpoint::new(addr);
assert!(filter
.read(ReadContext::new(
Endpoints::new(vec![Endpoint::new("127.0.0.1:8080".parse().unwrap(),)])
.unwrap()
.into(),
addr,
vec![]
))
.is_some());
assert!(filter
.write(WriteContext::new(&endpoint, addr, addr, vec![],))
.is_some());
}
}