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
mod tests;
pub mod embedded;
pub mod network;
use std::collections::{BTreeMap, HashMap};
use std::hash::Hash;
use std::sync::{Arc, Mutex};
use std::thread;
use std::time::Duration;
use tokio::net::TcpStream;
use crate::embedded::{ThreadSafeCache, ThreadSafeCacheImpl};
use crate::network::NetworkCache;
pub struct BuilderNetwork<K, V> {
address: String,
rt: tokio::runtime::Runtime,
phantom_data: std::marker::PhantomData<(K, V)>,
}
impl <K: std::marker::Send + 'static + Clone + Eq + Hash + serde::Serialize + serde::de::DeserializeOwned,
V: std::marker::Send + Clone + serde::de::DeserializeOwned + serde::Serialize + 'static> BuilderNetwork<K, V> {
pub fn init() -> BuilderNetwork<K, V> {
let rt = tokio::runtime::Runtime::new().unwrap();
BuilderNetwork {
address: "".to_string(),
rt: rt,
phantom_data: Default::default(),
}
}
pub fn address(&mut self, address: String) -> &mut Self {
self.address = address;
self
}
pub fn connect(self) -> NetworkCache<K, V> {
let stream = self.rt.block_on(async {
let stream = TcpStream::connect(self.address.as_str()).await.unwrap();
stream
});
let ret = NetworkCache {
tcp_stream: stream,
rt: self.rt,
phantom_data: Default::default(),
};
ret
}
}
pub struct BuilderEmbedded<K, V> {
max_size: i32,
phantom_data: std::marker::PhantomData<(K, V)>,
}
impl <K: std::marker::Send + 'static + Clone + Eq + Hash + serde::Serialize + serde::de::DeserializeOwned,
V: std::marker::Send + Clone + serde::de::DeserializeOwned + serde::Serialize + 'static> BuilderEmbedded<K, V> {
pub fn init() -> BuilderEmbedded<K, V> {
BuilderEmbedded {
max_size: 1000,
phantom_data: Default::default(),
}
}
pub fn build(self) -> ThreadSafeCache<K, V> {
let im = Arc::new(Mutex::new(ThreadSafeCacheImpl {
cache: HashMap::new(),
expiration_set: BTreeMap::new(),
max_size: self.max_size,
current_size: 0,
}));
let ret = ThreadSafeCache {
implementation: im,
};
let mut ret_clone = ret.clone();
thread::spawn(move || {
loop {
if !ret_clone.clean() {
break;
}
thread::sleep(Duration::from_millis(1000));
}
});
ret
}
pub fn max_size(&mut self, max_size: i32) -> &mut Self {
self.max_size = max_size;
self
}
}
pub trait ThreadSafeCacheTrait<K: 'static + Clone + Eq + Hash + serde::Serialize + serde::de::DeserializeOwned,
V: Clone + serde::Serialize + serde::de::DeserializeOwned +'static> {
fn put(&mut self, key: K, val: V)
where K: Eq + Hash;
fn put_exp(&mut self, key: K, val: V, expiration: i32)
where K: Eq + Hash + Clone;
fn get(&mut self, key: K) -> Option<V>
where K: Eq + Hash, V: Clone;
fn exists(&mut self, key: K) -> bool;
fn rm(&mut self, key: K);
}
pub trait ThreadSafeCachePersistTrait<K: 'static + Clone + Eq + Hash + serde::Serialize + serde::de::DeserializeOwned,
V: Clone + serde::Serialize + serde::de::DeserializeOwned +'static>: ThreadSafeCacheTrait<K,V> {
fn save(&mut self, file_name: &str);
fn load(&mut self, file_name: &str);
}