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
use move_binary_format::CompiledModule;
use move_core_types::language_storage::ModuleId;
use std::{cell::RefCell, collections::hash_map::HashMap, hash::Hash, rc::Rc};
pub struct ModuleCacheImpl<K, V> {
id_map: RefCell<HashMap<K, usize>>,
modules: RefCell<Vec<Rc<V>>>,
}
impl<K, V> ModuleCacheImpl<K, V>
where
K: Eq + Hash,
{
pub fn new() -> Self {
Self {
id_map: RefCell::new(HashMap::new()),
modules: RefCell::new(vec![]),
}
}
pub fn insert(&self, key: K, module: V) -> Rc<V> {
self.modules.borrow_mut().push(Rc::new(module));
let idx = self.modules.borrow().len() - 1;
self.id_map.borrow_mut().insert(key, idx);
self.modules
.borrow()
.last()
.expect("ModuleCache: last() after push() impossible failure")
.clone()
}
pub fn get(&self, key: &K) -> Option<Rc<V>> {
self.id_map
.borrow()
.get(key)
.and_then(|idx| self.modules.borrow().get(*idx).cloned())
}
}
pub type ModuleCache = ModuleCacheImpl<ModuleId, CompiledModule>;