fix for case tested by it_gracefully_handles_connect_fail_reconnect

tari-project · Feb 9, 2022 · db6c088 · db6c088
1 parent b1a6b5f
commit db6c088
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/base_layer/wallet/src/connectivity_service/service.rs b/base_layer/wallet/src/connectivity_service/service.rs
@@ -116,12 +116,21 @@ impl WalletConnectivityService {
     }
 
     async fn check_connection(&mut self) {
-        if let Some(pool) = self.pools.as_ref() {
-            if !pool.base_node_wallet_rpc_client.is_connected().await {
-                debug!(target: LOG_TARGET, "Peer connection lost. Attempting to reconnect...");
+        debug!(target: LOG_TARGET, "HERE1");
+        match self.pools.as_ref() {
+            Some(pool) => {
+                debug!(target: LOG_TARGET, "HERE2");
+                if !pool.base_node_wallet_rpc_client.is_connected().await {
+                    debug!(target: LOG_TARGET, "Peer connection lost. Attempting to reconnect...");
+                    self.set_online_status(OnlineStatus::Offline);
+                    self.setup_base_node_connection().await;
+                }
+            },
+            None => {
+                debug!(target: LOG_TARGET, "No connection. Attempting to connect...");
                 self.set_online_status(OnlineStatus::Offline);
                 self.setup_base_node_connection().await;
-            }
+            },
         }
     }