Fix epoch smeshers stats

spacemeshos · Jul 16, 2024 · bed1287 · bed1287
1 parent 8a96b27
commit bed1287
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/internal/storage/storagereader/epochs.go b/internal/storage/storagereader/epochs.go
@@ -41,6 +41,9 @@ func (s *Reader) GetEpochs(ctx context.Context, query *bson.D, opts ...*options.
 		epoch.Stats.Current.RewardsNumber = count
 		epoch.Stats.Cumulative.Rewards = total
 		epoch.Stats.Cumulative.RewardsNumber = count
+
+		atxCount, err := s.CountActivations(context.Background(), &bson.D{{Key: "targetEpoch", Value: epoch.Number}})
+		epoch.Stats.Current.Smeshers = atxCount
 	}
 
 	return epochs, nil
@@ -69,5 +72,9 @@ func (s *Reader) GetEpoch(ctx context.Context, epochNumber int) (*model.Epoch, e
 	epoch.Stats.Cumulative.Rewards = total
 	epoch.Stats.Cumulative.RewardsNumber = count
 
+	atxCount, err := s.CountActivations(context.Background(), &bson.D{{Key: "targetEpoch", Value: epoch.Number}})
+	epoch.Stats.Current.Smeshers = atxCount
+	epoch.Stats.Cumulative.Smeshers = atxCount
+
 	return epoch, nil
 }