Use Deque for REQUESTS storage

This commit is contained in:
hu55a1n1 2024-01-01 12:37:08 -08:00
parent 1533bb8cd2
commit de5314122a
2 changed files with 10 additions and 10 deletions

View file

@ -108,10 +108,12 @@ pub mod execute {
let _ = deps.api.addr_validate(&address)?; let _ = deps.api.addr_validate(&address)?;
let _ = hex::decode(&nonce); let _ = hex::decode(&nonce);
REQUESTS.save( REQUESTS.push_back(
deps.storage, deps.storage,
&nonce, &(
&Request::JoinComputeNode((io_exchange_key.clone(), address)), nonce,
Request::JoinComputeNode((io_exchange_key.clone(), address)),
),
)?; )?;
Ok(Response::new() Ok(Response::new()
@ -129,10 +131,10 @@ pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
} }
pub mod query { pub mod query {
use cosmwasm_std::{Deps, Order, StdResult}; use cosmwasm_std::{Deps, StdResult};
use crate::msg::query::{GetRequestsResponse, GetSgxStateResponse}; use crate::msg::query::{GetRequestsResponse, GetSgxStateResponse};
use crate::state::{RawNonce, Request, SgxState, REQUESTS, SGX_STATE}; use crate::state::{SgxState, REQUESTS, SGX_STATE};
pub fn get_sgx_state(deps: Deps) -> StdResult<GetSgxStateResponse> { pub fn get_sgx_state(deps: Deps) -> StdResult<GetSgxStateResponse> {
let SgxState { let SgxState {
@ -148,9 +150,7 @@ pub mod query {
pub fn get_requests(deps: Deps) -> StdResult<GetRequestsResponse> { pub fn get_requests(deps: Deps) -> StdResult<GetRequestsResponse> {
Ok(GetRequestsResponse { Ok(GetRequestsResponse {
requests: REQUESTS requests: REQUESTS.iter(deps.storage)?.flat_map(|r| r.ok()).collect(),
.range(deps.storage, None, None, Order::Ascending)
.collect::<StdResult<Vec<(RawNonce, Request)>>>()?,
}) })
} }
} }

View file

@ -1,5 +1,5 @@
use cosmwasm_schema::cw_serde; use cosmwasm_schema::cw_serde;
use cw_storage_plus::{Item, Map}; use cw_storage_plus::{Deque, Item};
pub type RawNonce = String; pub type RawNonce = String;
pub type RawPublicKey = String; pub type RawPublicKey = String;
@ -27,5 +27,5 @@ pub struct SgxState {
} }
pub const STATE: Item<State> = Item::new("state"); pub const STATE: Item<State> = Item::new("state");
pub const REQUESTS: Map<&RawNonce, Request> = Map::new("requests"); pub const REQUESTS: Deque<(RawNonce, Request)> = Deque::new("requests");
pub const SGX_STATE: Item<SgxState> = Item::new("sgxstate"); pub const SGX_STATE: Item<SgxState> = Item::new("sgxstate");