Optimize reward sync

spacemeshos · Jan 22, 2024 · a5ce4b7 · a5ce4b7
1 parent 131d1dd
commit a5ce4b7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/collector/collector.go b/collector/collector.go
@@ -44,6 +44,7 @@ type Listener interface {
 	GetEpochNumLayers() uint32
 	GetTransactions(parent context.Context, query *bson.D, opts ...*options.FindOptions) ([]model.Transaction, error)
 	UpdateTransactionState(parent context.Context, id string, state int32) error
+	UpdateEpochStats(layer uint32)
 }
 
 type Collector struct {

diff --git a/collector/mesh.go b/collector/mesh.go
@@ -175,6 +175,8 @@ func (c *Collector) syncLayer(lid types.LayerID) error {
 		c.listener.OnReward(r)
 	}
 
+	c.listener.UpdateEpochStats(layer.Number.Number)
+
 	return nil
 }
 

diff --git a/storage/storage.go b/storage/storage.go
@@ -252,8 +252,11 @@ func (s *Storage) OnReward(in *pb.Reward) {
 	}
 
 	s.requestBalanceUpdate(reward.Layer, reward.Coinbase)
-	s.setChangedEpoch(reward.Layer)
-	s.updateEpochs() // trigger epoch stat recalculation todo: optimize this
+}
+
+func (s *Storage) UpdateEpochStats(layer uint32) {
+	s.setChangedEpoch(layer)
+	s.updateEpochs()
 }
 
 func (s *Storage) OnTransactionResult(res *pb.TransactionResult, state *pb.TransactionState) {