add logic update total balance for overview page

JustinBeBoy · Jul 17, 2024 · cbcd906 · cbcd906
1 parent b891a8e
commit cbcd906
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/ui/page/root/home_page.go b/ui/page/root/home_page.go
@@ -199,7 +199,6 @@ func (hp *HomePage) OnNavigatedTo() {
 	}
 
 	hp.AssetsManager.WatchBalanceChange(func() {
-		fmt.Println("Update Total balance")
 		go hp.CalculateAssetsUSDBalance()
 	})
 }

diff --git a/ui/page/root/overview_page.go b/ui/page/root/overview_page.go
@@ -1205,6 +1205,18 @@ func (pg *OverviewPage) listenForMixerNotifications() {
 		}
 	}
 
+	// add rate listener
+	rateListener := &ext.RateListener{
+		OnRateUpdated: func() {
+			pg.updateAssetsUSDBalance()
+		},
+	}
+	if !pg.AssetsManager.RateSource.IsRateListenerExist(OverviewPageID) {
+		if err := pg.AssetsManager.RateSource.AddRateListener(rateListener, OverviewPageID); err != nil {
+			log.Error("Can't listen rate notification ")
+		}
+	}
+
 	pg.sortedMixerSlideKeys = make([]int, 0)
 	pg.mixerSliderData = make(map[int]*mixerData)
 	for _, wal := range wallets {
-Original file line number
+Diff line change
@@ Expand Up / @@ -199,7 +199,6 @@ func (hp *HomePage) OnNavigatedTo() { @@
     	}
     	hp.AssetsManager.WatchBalanceChange(func() {
-    		fmt.Println("Update Total balance")
     		go hp.CalculateAssetsUSDBalance()
     	})
     }
@@ Expand Down @@