Update account balance when new block is received

Author: dyng

internal/service/syncer.go

@@ -18,6 +18,8 @@ const (
 	TopicNewBlock = "service:newBlock"
 	// TopicChainData is the topic about latest chain data (ether price, gas price, etc.)
 	TopicChainData = "service:chainData"
+	// TopicTick is a topic that receives tick event periodically
+	TopicTick = "service:tick"
 
 	// UpdatePeriod is the time duration between two updates
 	UpdatePeriod = 5 * time.Second
@@ -92,11 +94,13 @@ func (s *Syncer) sync() {
 		case tick := <-s.ticker.C:
 			log.Debug("Process periodic synchronization", "tick", tick)
 
+			// update eth price
 			price, err := s.service.GetEthPrice()
 			if err != nil {
 				log.Error("Failed to fetch ether's price", "error", err)
 			}
 
+			// update gas price
 			gasPrice, err := s.service.GetGasPrice()
 			if err != nil {
 				log.Error("Failed to fetch gas price", "error", err)
@@ -107,6 +111,10 @@ func (s *Syncer) sync() {
 				GasPrice: gasPrice,
 			}
 			s.eventBus.Publish(TopicChainData, data)
+
+			go func() {
+				s.eventBus.Publish(TopicTick, tick)
+			}()
 		}
 	}
 }

internal/view/account.go

@@ -163,13 +163,21 @@ func (a *Account) ShowImportABIDialog() {
 }
 
 func (a *Account) onNewBlock(block *common.Block) {
+	if a.account == nil {
+		return
+	}
+
 	txns, err := a.app.service.GetTransactionsByBlock(block)
 	if err != nil {
 		log.Error("cannot extract transactions from block", "blockHash", block.Hash(), "error", err)
 		return
 	}
 
+	// update current account
+	a.account.UpdateBalance()
+
 	a.app.QueueUpdateDraw(func() {
+		a.refresh()
 		a.transactionList.FilterAndPrependTransactions(txns)
 	})
 }

internal/view/signer.go

@@ -1,6 +1,7 @@
 package view
 
 import (
+	"github.com/dyng/ramen/internal/common"
 	"github.com/dyng/ramen/internal/common/conv"
 	"github.com/dyng/ramen/internal/service"
 	"github.com/dyng/ramen/internal/view/style"
@@ -30,6 +31,9 @@ func NewSigner(app *App) *Signer {
 	// setup layout
 	signer.initLayout()
 
+	// subscribe tick event
+	app.eventBus.Subscribe(service.TopicNewBlock, signer.onNewBlock)
+
 	return signer
 }
 
@@ -97,3 +101,10 @@ func (si *Signer) layoutSomeSigner() {
 
 	si.Primitive = flex
 }
+
+func (si *Signer) onNewBlock(block *common.Block) {
+	if si.signer != nil {
+		si.signer.UpdateBalance()
+		si.refresh()
+	}
+}