Added pagination query for delegations

DA0-DA0 · Dec 28, 2022 · 5c1bc28 · 5c1bc28
1 parent b0e54d0
commit 5c1bc28
Show file tree

Hide file tree

Showing 4 changed files with 58 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/contracts/proposal/cwd-proposal-delegate/Cargo.toml b/contracts/proposal/cwd-proposal-delegate/Cargo.toml
@@ -37,6 +37,7 @@ dao-pre-propose-base = { workspace = true }
 dao-interface = { workspace = true }
 dao-voting = { workspace = true }
 cw-hooks = { workspace = true }
+cw-paginate = { workspace = true }
 dao-proposal-hooks = { workspace = true }
 dao-vote-hooks = { workspace = true }
 dao-pre-propose-multiple = { workspace = true }

diff --git a/contracts/proposal/cwd-proposal-delegate/src/contract.rs b/contracts/proposal/cwd-proposal-delegate/src/contract.rs
@@ -4,6 +4,7 @@ use cosmwasm_std::{
     to_binary, Binary, BlockInfo, Deps, DepsMut, Empty, Env, MessageInfo, OverflowError, Reply,
     Response, StdError, StdResult, Storage, SubMsg, WasmMsg,
 };
+use cw_paginate::paginate_map_values;
 use cw_utils::Expiration;
 // use cw2::set_contract_version;
 
@@ -105,6 +106,16 @@ pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
         QueryMsg::Delegation { delegation_id } => {
             Ok(to_binary(&query_delegation(deps, delegation_id)?)?)
         }
+        QueryMsg::Delegations { start_after, limit } => {
+            let delegations = paginate_map_values(
+                deps,
+                &DELEGATIONS,
+                start_after,
+                limit,
+                cosmwasm_std::Order::Ascending,
+            )?;
+            Ok(to_binary(&delegations)?)
+        }
     }
 }
 

diff --git a/contracts/proposal/cwd-proposal-delegate/src/msg.rs b/contracts/proposal/cwd-proposal-delegate/src/msg.rs
@@ -29,8 +29,13 @@ pub enum ExecuteMsg {
 #[cw_serde]
 #[derive(QueryResponses)]
 pub enum QueryMsg {
-    #[returns(DelegationResponse)]
+    #[returns(Delegation)]
     Delegation { delegation_id: u64 },
+    #[returns(Vec<Delegation>)]
+    Delegations {
+        start_after: Option<u64>,
+        limit: Option<u32>,
+    },
 }
 
 pub type DelegationResponse = Delegation;