feat: improve balances query

contracts/neutron_interchain_queries/src/contract.rs

@@ -21,7 +21,7 @@ use neutron_sdk::interchain_queries::v045::queries::{
 use neutron_sdk::interchain_queries::{
     check_query_type, get_registered_query, query_kv_result,
     v045::{
-        new_register_balance_query_msg, new_register_bank_total_supply_query_msg,
+        new_register_balances_query_msg, new_register_bank_total_supply_query_msg,
         new_register_delegator_delegations_query_msg, new_register_distribution_fee_pool_query_msg,
         new_register_gov_proposal_query_msg, new_register_staking_validators_query_msg,
         new_register_transfers_query_msg,
@@ -128,7 +128,7 @@ pub fn register_balance_query(
     denom: String,
     update_period: u64,
 ) -> NeutronResult<Response<NeutronMsg>> {
-    let msg = new_register_balance_query_msg(connection_id, addr, denom, update_period)?;
+    let msg = new_register_balances_query_msg(connection_id, addr, vec![denom], update_period)?;
 
     Ok(Response::new().add_message(msg))
 }

packages/neutron-sdk/src/interchain_queries/v045/mod.rs

@@ -4,7 +4,7 @@ pub mod register_queries;
 pub mod types;
 
 pub use register_queries::{
-    new_register_balance_query_msg, new_register_bank_total_supply_query_msg,
+    new_register_balances_query_msg, new_register_bank_total_supply_query_msg,
     new_register_delegator_delegations_query_msg, new_register_distribution_fee_pool_query_msg,
     new_register_gov_proposal_query_msg, new_register_staking_validators_query_msg,
     new_register_transfers_query_msg,

packages/neutron-sdk/src/interchain_queries/v045/register_queries.rs

@@ -17,32 +17,34 @@ use crate::{
 };
 use cosmwasm_std::Binary;
 
-/// Creates a message to register an Interchain Query to get balance of account on remote chain for particular denom
+/// Creates a message to register an Interchain Query to get balance of account on remote chain for list of denoms
 ///
 /// * **connection_id** is an IBC connection identifier between Neutron and remote chain;
 /// * **addr** address of an account on remote chain for which you want to get balances;
-/// * **denom** denomination of the coin for which you want to get balance;
+/// * **denoms** denominations of the coins for which you want to get balance;
 /// * **update_period** is used to say how often the query must be updated.
-pub fn new_register_balance_query_msg(
+pub fn new_register_balances_query_msg(
     connection_id: String,
     addr: String,
-    denom: String,
+    denoms: Vec<String>,
     update_period: u64,
 ) -> NeutronResult<NeutronMsg> {
     let converted_addr_bytes = decode_and_convert(addr.as_str())?;
 
-    let balance_key = create_account_denom_balance_key(converted_addr_bytes, denom)?;
+    let mut kv_keys: Vec<KVKey> = Vec::with_capacity(denoms.len());
 
-    let kv_key = KVKey {
-        path: BANK_STORE_KEY.to_string(),
-        key: Binary(balance_key),
-    };
+    for denom in denoms {
+        let balance_key = create_account_denom_balance_key(converted_addr_bytes.clone(), denom)?;
 
-    NeutronMsg::register_interchain_query(
-        QueryPayload::KV(vec![kv_key]),
-        connection_id,
-        update_period,
-    )
+        let kv_key = KVKey {
+            path: BANK_STORE_KEY.to_string(),
+            key: Binary(balance_key),
+        };
+
+        kv_keys.push(kv_key)
+    }
+
+    NeutronMsg::register_interchain_query(QueryPayload::KV(kv_keys), connection_id, update_period)
 }
 
 /// Creates a message to register an Interchain Query to get total supply on remote chain for particular denom

packages/neutron-sdk/src/interchain_queries/v045/types.rs

@@ -104,9 +104,11 @@ impl KVReconstruct for Balances {
         let mut coins: Vec<Coin> = Vec::with_capacity(storage_values.len());
 
         for kv in storage_values {
-            let balance: CosmosCoin = CosmosCoin::decode(kv.value.as_slice())?;
-            let amount = Uint128::from_str(balance.amount.as_str())?;
-            coins.push(Coin::new(amount.u128(), balance.denom));
+            if kv.value.len() > 0 {
+                let balance: CosmosCoin = CosmosCoin::decode(kv.value.as_slice())?;
+                let amount = Uint128::from_str(balance.amount.as_str())?;
+                coins.push(Coin::new(amount.u128(), balance.denom));
+            }
         }
 
         Ok(Balances { coins })