feat: trigger time lock balance update when block received (#3567)

Description --- This PR update the console wallet event monitor to trigger a balance refresh when a new block is received IF there is a timelocked balanced that could potentially changed based on the block height. How Has This Been Tested? --- cargo test
tari-project · Nov 15, 2021 · 11b8afa · 11b8afa
1 parent f013e19
commit 11b8afa
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/applications/tari_console_wallet/src/ui/state/app_state.rs b/applications/tari_console_wallet/src/ui/state/app_state.rs
@@ -693,6 +693,15 @@ impl AppStateInner {
         Ok(())
     }
 
+    pub fn has_time_locked_balance(&self) -> bool {
+        if let Some(time_locked_balance) = self.data.balance.time_locked_balance {
+            if time_locked_balance > MicroTari::from(0) {
+                return true;
+            }
+        }
+        false
+    }
+
     pub async fn refresh_balance(&mut self, balance: Balance) -> Result<(), UiError> {
         self.data.balance = balance;
         self.updated = true;

diff --git a/applications/tari_console_wallet/src/ui/state/wallet_event_monitor.rs b/applications/tari_console_wallet/src/ui/state/wallet_event_monitor.rs
@@ -260,6 +260,12 @@ impl WalletEventMonitor {
         if let Err(e) = inner.refresh_base_node_state(state).await {
             warn!(target: LOG_TARGET, "Error refresh app_state: {}", e);
         }
+
+        if inner.has_time_locked_balance() {
+            if let Err(e) = self.balance_enquiry_debounce_tx.send(()) {
+                warn!(target: LOG_TARGET, "Error refresh app_state: {}", e);
+            }
+        }
     }
 
     async fn trigger_base_node_peer_refresh(&mut self, peer: Peer) {