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
use std::cell::RefCell;
use std::rc::Rc;
use std::collections::HashMap;
use futures::Future;
use ::client::ClientHandle;
use ::client::rc_future::{rc_future, RcFuture};
use ::error::*;
use ::op::{Message, Query};
#[derive(Clone)]
#[must_use = "queries can only be sent through a ClientHandle"]
pub struct MemoizeClientHandle<H: ClientHandle> {
client: H,
active_queries: Rc<RefCell<HashMap<Query, RcFuture<Box<Future<Item=Message, Error=ClientError>>>>>>,
}
impl<H> MemoizeClientHandle<H> where H: ClientHandle {
pub fn new(client: H) -> MemoizeClientHandle<H> {
MemoizeClientHandle { client: client, active_queries: Rc::new(RefCell::new(HashMap::new())) }
}
}
impl<H> ClientHandle for MemoizeClientHandle<H> where H: ClientHandle {
fn send(&self, message: Message) -> Box<Future<Item=Message, Error=ClientError>> {
let query = message.get_queries().first().expect("no query!").clone();
if let Some(rc_future) = self.active_queries.borrow().get(&query) {
return Box::new(rc_future.clone());
}
let mut map = self.active_queries.borrow_mut();
let rc_future = map.entry(query).or_insert_with(move ||{
rc_future(self.client.send(message))
});
return Box::new(rc_future.clone());
}
}
#[cfg(test)]
mod test {
use std::cell::Cell;
use ::client::*;
use ::error::*;
use ::op::*;
use ::rr::*;
use futures::*;
#[derive(Clone)]
struct TestClient { i: Cell<u16> }
impl ClientHandle for TestClient {
fn send(&self, _: Message) -> Box<Future<Item=Message, Error=ClientError>> {
let mut message = Message::new();
let i = self.i.get();
message.id(i);
self.i.set(i + 1);
Box::new(finished(message))
}
}
#[test]
fn test_memoized() {
let client = MemoizeClientHandle::new(TestClient{i: Cell::new(0)});
let mut test1 = Message::new();
test1.add_query(Query::new().query_type(RecordType::A).clone());
let mut test2 = Message::new();
test2.add_query(Query::new().query_type(RecordType::AAAA).clone());
let result = client.send(test1.clone()).wait().ok().unwrap();
assert_eq!(result.get_id(), 0);
let result = client.send(test2.clone()).wait().ok().unwrap();
assert_eq!(result.get_id(), 1);
let result = client.send(test1).wait().ok().unwrap();
assert_eq!(result.get_id(), 0);
let result = client.send(test2).wait().ok().unwrap();
assert_eq!(result.get_id(), 1);
}
}