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
use crate::api::stable;
use std::any::{Any, TypeId};
use std::collections::BTreeMap;
type StorageTree = BTreeMap<TypeId, Box<dyn Any>>;
static mut STORAGE: Option<StorageTree> = None;
fn storage() -> &'static mut StorageTree {
unsafe {
if let Some(s) = &mut STORAGE {
s
} else {
STORAGE = Some(BTreeMap::new());
storage()
}
}
}
pub fn delete<T: Sized + Default + 'static>() -> bool {
let type_id = std::any::TypeId::of::<T>();
storage().remove(&type_id).is_some()
}
pub fn get_mut<T: Sized + Default + 'static>() -> &'static mut T {
let type_id = std::any::TypeId::of::<T>();
let store = storage();
store
.entry(type_id)
.or_insert_with(|| Box::new(T::default()))
.downcast_mut()
.expect("Unexpected value of invalid type.")
}
pub fn get<T: Sized + Default + 'static>() -> &'static T {
get_mut::<T>()
}
pub fn stable_save<T>(t: T) -> Result<(), candid::Error>
where
T: candid::ser::ArgumentEncoder,
{
candid::ser::write_args(&mut stable::StableWriter::default(), t)
}
pub fn stable_restore<T>() -> Result<T, String>
where
T: for<'de> candid::de::ArgumentDecoder<'de>,
{
let bytes = stable::stable_bytes();
let mut de =
candid::de::IDLDeserialize::new(bytes.as_slice()).map_err(|e| format!("{:?}", e))?;
let res = candid::de::ArgumentDecoder::decode(&mut de).map_err(|e| format!("{:?}", e))?;
let _ = de.done();
Ok(res)
}