use crate::traits::{Api, Querier, Storage};
use crate::QuerierWrapper;
pub struct OwnedDeps<S: Storage, A: Api, Q: Querier> {
pub storage: S,
pub api: A,
pub querier: Q,
}
pub struct DepsMut<'a> {
pub storage: &'a mut dyn Storage,
pub api: &'a dyn Api,
pub querier: QuerierWrapper<'a>,
}
#[derive(Copy, Clone)]
pub struct Deps<'a> {
pub storage: &'a dyn Storage,
pub api: &'a dyn Api,
pub querier: QuerierWrapper<'a>,
}
impl<S: Storage, A: Api, Q: Querier> OwnedDeps<S, A, Q> {
pub fn as_ref(&'_ self) -> Deps<'_> {
Deps {
storage: &self.storage,
api: &self.api,
querier: QuerierWrapper::new(&self.querier),
}
}
pub fn as_mut(&'_ mut self) -> DepsMut<'_> {
DepsMut {
storage: &mut self.storage,
api: &self.api,
querier: QuerierWrapper::new(&self.querier),
}
}
}
impl<'a> DepsMut<'a> {
pub fn as_ref(&'_ self) -> Deps<'_> {
Deps {
storage: self.storage,
api: self.api,
querier: self.querier,
}
}
pub fn branch(&'_ mut self) -> DepsMut<'_> {
DepsMut {
storage: self.storage,
api: self.api,
querier: self.querier,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::mock::mock_dependencies;
fn execute(mut deps: DepsMut) {
execute2(deps.branch());
query(deps.as_ref());
execute2(deps.branch());
}
fn execute2(_deps: DepsMut) {}
fn query(deps: Deps) {
query2(deps);
query2(deps);
}
fn query2(_deps: Deps) {}
#[test]
fn ensure_easy_reuse() {
let mut deps = mock_dependencies(&[]);
execute(deps.as_mut());
query(deps.as_ref())
}
}