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
use serde::de::DeserializeOwned;
use serde::Serialize;
use std::marker::PhantomData;
use crate::helpers::{may_deserialize, must_deserialize, nested_namespaces_with_key};
use cosmwasm_std::{to_vec, StdError, StdResult, Storage};
use std::ops::Deref;
#[derive(Debug, Clone)]
pub struct Path<T>
where
T: Serialize + DeserializeOwned,
{
storage_key: Vec<u8>,
data: PhantomData<T>,
}
impl<T> Deref for Path<T>
where
T: Serialize + DeserializeOwned,
{
type Target = [u8];
fn deref(&self) -> &[u8] {
&self.storage_key
}
}
impl<T> Path<T>
where
T: Serialize + DeserializeOwned,
{
pub fn new(namespace: &[u8], keys: &[&[u8]]) -> Self {
let l = keys.len();
let storage_key = nested_namespaces_with_key(&[namespace], &keys[0..l - 1], keys[l - 1]);
Path {
storage_key,
data: PhantomData,
}
}
pub fn save<S: Storage>(&self, store: &mut S, data: &T) -> StdResult<()> {
store.set(&self.storage_key, &to_vec(data)?);
Ok(())
}
pub fn remove<S: Storage>(&self, store: &mut S) {
store.remove(&self.storage_key);
}
pub fn load<S: Storage>(&self, store: &S) -> StdResult<T> {
let value = store.get(&self.storage_key);
must_deserialize(&value)
}
pub fn may_load<S: Storage>(&self, store: &S) -> StdResult<Option<T>> {
let value = store.get(&self.storage_key);
may_deserialize(&value)
}
pub fn update<A, E, S>(&self, store: &mut S, action: A) -> Result<T, E>
where
A: FnOnce(Option<T>) -> Result<T, E>,
E: From<StdError>,
S: Storage,
{
let input = self.may_load(store)?;
let output = action(input)?;
self.save(store, &output)?;
Ok(output)
}
}