added total amount of txs to transfer history

Author: reuvenpo

src/contract.rs

@@ -261,9 +261,9 @@ pub fn query_transactions<S: Storage, A: Api, Q: Querier>(
     page_size: u32,
 ) -> StdResult<Binary> {
     let address = deps.api.canonical_address(account).unwrap();
-    let txs = get_transfers(&deps.api, &deps.storage, &address, page, page_size)?;
+    let (txs, total) = get_transfers(&deps.api, &deps.storage, &address, page, page_size)?;
 
-    let result = QueryAnswer::TransferHistory { txs };
+    let result = QueryAnswer::TransferHistory { txs, total };
     to_binary(&result)
 }
 
@@ -2707,7 +2707,7 @@ mod tests {
         // let a: QueryAnswer = from_binary(&query_result.unwrap()).unwrap();
         // println!("{:?}", a);
         let transfers = match from_binary(&query_result.unwrap()).unwrap() {
-            QueryAnswer::TransferHistory { txs } => txs,
+            QueryAnswer::TransferHistory { txs, total } => txs,
             _ => panic!("Unexpected"),
         };
         assert!(transfers.is_empty());
@@ -2720,7 +2720,7 @@ mod tests {
         };
         let query_result = query(&deps, query_msg);
         let transfers = match from_binary(&query_result.unwrap()).unwrap() {
-            QueryAnswer::TransferHistory { txs } => txs,
+            QueryAnswer::TransferHistory { txs, total } => txs,
             _ => panic!("Unexpected"),
         };
         assert_eq!(transfers.len(), 3);
@@ -2733,7 +2733,7 @@ mod tests {
         };
         let query_result = query(&deps, query_msg);
         let transfers = match from_binary(&query_result.unwrap()).unwrap() {
-            QueryAnswer::TransferHistory { txs } => txs,
+            QueryAnswer::TransferHistory { txs, total } => txs,
             _ => panic!("Unexpected"),
         };
         assert_eq!(transfers.len(), 2);

src/msg.rs

@@ -288,6 +288,7 @@ pub enum QueryAnswer {
     },
     TransferHistory {
         txs: Vec<Tx>,
+        total: u64,
     },
 
     ViewingKeyError {

src/state.rs

@@ -130,15 +130,15 @@ pub fn get_transfers<A: Api, S: ReadonlyStorage>(
     for_address: &CanonicalAddr,
     page: u32,
     page_size: u32,
-) -> StdResult<Vec<Tx>> {
+) -> StdResult<(Vec<Tx>, u64)> {
     let store = ReadonlyPrefixedStorage::multilevel(&[PREFIX_TXS, for_address.as_slice()], storage);
 
     // Try to access the storage of txs for the account.
     // If it doesn't exist yet, return an empty list of transfers.
     let store = if let Some(result) = AppendStore::<StoredTx, _>::attach(&store) {
         result?
     } else {
-        return Ok(vec![]);
+        return Ok((vec![], 0));
     };
 
     // Take `page_size` txs starting from the latest tx, potentially skipping `page * page_size`
@@ -152,7 +152,7 @@ pub fn get_transfers<A: Api, S: ReadonlyStorage>(
     let txs: StdResult<Vec<Tx>> = tx_iter
         .map(|tx| tx.map(|tx| tx.into_humanized(api)).and_then(|x| x))
         .collect();
-    txs
+    txs.map(|txs| (txs, store.len() as u64))
 }
 
 // Config