nologies/cycles-quartz
4
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use entry API for HashMap

Browse source
This commit is contained in:
hu55a1n1 2023-11-15 02:01:39 -08:00
parent c58ec93f08
commit 4bb6dfbeff
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
bisenzone-cw-mvp/src/contract.rs
View file

@ -66,10 +66,10 @@ pub mod execute {
*state *state
.utilization .utilization
.get_mut(&creditor) .entry(creditor)
.unwrap() .or_default()
.get_mut(&info.sender) .entry(info.sender)
.unwrap() += amount; .or_default() += amount;
Ok(state) Ok(state)
})?; })?;
@ -131,13 +131,18 @@ pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
pub mod query { pub mod query {
use super::*; use super::*;
use std::collections::HashMap;
use crate::msg::GetObligationsResponse; use crate::msg::GetObligationsResponse;
pub fn get_obligations(deps: Deps, creditor: Addr) -> StdResult<GetObligationsResponse> { pub fn get_obligations(deps: Deps, creditor: Addr) -> StdResult<GetObligationsResponse> {
let state = STATE.load(deps.storage)?; let state = STATE.load(deps.storage)?;
Ok(GetObligationsResponse { Ok(GetObligationsResponse {
obligations: state.utilization[&creditor].clone(), obligations: state
.utilization
.get(&creditor)
.unwrap_or(&HashMap::new())
.clone(),
}) })
} }
} }