Expose the AccountRoot and RewardsRoot and print them on engine start

Author: alexsporn

pkg/protocol/engine/engine.go

@@ -253,6 +253,18 @@ func New(
 			e.InitializedEvent().Trigger()
 
 			e.LogTrace("initialized", "settings", e.Storage.Settings().String())
+
+			latestCommitment := e.Storage.Settings().LatestCommitment()
+			e.LogInfo("LatestCommitment", "slot", latestCommitment.Slot(), "ID", latestCommitment.ID())
+			e.LogInfo("Ledger state", "AccountRoot", e.Ledger.AccountRoot(), "latestCommitment.Slot", latestCommitment.Slot())
+
+			currentEpoch := e.CommittedAPI().TimeProvider().EpochFromSlot(latestCommitment.Slot())
+			e.LogInfo("Rewards state", "RewardsRoot", lo.PanicOnErr(e.SybilProtection.RewardsRoot(currentEpoch)), "epoch", currentEpoch, "latestCommitment.Slot", latestCommitment.Slot())
+
+			if currentEpoch > 0 {
+				prevEpoch := currentEpoch - 1
+				e.LogInfo("Rewards state", "RewardsRoot", lo.PanicOnErr(e.SybilProtection.RewardsRoot(prevEpoch)), "epoch", prevEpoch, "lastSlot", e.CommittedAPI().TimeProvider().EpochEnd(prevEpoch))
+			}
 		},
 	)
 }

pkg/protocol/engine/ledger/ledger.go

@@ -43,6 +43,8 @@ type Ledger interface {
 	Export(writer io.WriteSeeker, targetSlot iotago.SlotIndex) error
 	TrackBlock(block *blocks.Block)
 
+	AccountRoot() iotago.Identifier
+
 	// Reset resets the component to a clean state as if it was created at the last commitment.
 	Reset()
 

pkg/protocol/engine/ledger/ledger/ledger.go

@@ -221,6 +221,10 @@ func (l *Ledger) CommitSlot(slot iotago.SlotIndex) (stateRoot iotago.Identifier,
 	return stateTreeRoot, stateDiff.Mutations().Root(), l.accountsLedger.AccountsTreeRoot(), outputs, spenders, mutations, nil
 }
 
+func (l *Ledger) AccountRoot() iotago.Identifier {
+	return l.accountsLedger.AccountsTreeRoot()
+}
+
 func (l *Ledger) AddAccount(output *utxoledger.Output, blockIssuanceCredits iotago.BlockIssuanceCredits) error {
 	return l.accountsLedger.AddAccount(output, blockIssuanceCredits)
 }

pkg/protocol/sybilprotection/sybilprotection.go

@@ -39,5 +39,7 @@ type SybilProtection interface {
 	// Reset resets the component to a clean state as if it was created at the last commitment.
 	Reset()
 
+	RewardsRoot(epoch iotago.EpochIndex) (rewardsRoot iotago.Identifier, err error)
+
 	module.Module
 }

pkg/protocol/sybilprotection/sybilprotectionv1/performance/rewards.go

@@ -18,7 +18,17 @@ func (t *Tracker) RewardsRoot(epoch iotago.EpochIndex) (iotago.Identifier, error
 		return iotago.Identifier{}, err
 	}
 
-	return m.Root(), nil
+	root := m.Root()
+
+	t.LogDebug("RewardsRoot", "epoch", epoch, "root", root, "WasRestoredFromStorage", m.WasRestoredFromStorage())
+	if err := m.Stream(func(accountID iotago.AccountID, poolRewards *model.PoolRewards) error {
+		t.LogDebug("RewardsRoot", "accountID", accountID, "poolRewards", poolRewards)
+		return nil
+	}); err != nil {
+		panic(err)
+	}
+
+	return root, nil
 }
 
 func (t *Tracker) ValidatorReward(validatorID iotago.AccountID, stakingFeature *iotago.StakingFeature, claimingEpoch iotago.EpochIndex) (validatorReward iotago.Mana, firstRewardEpoch iotago.EpochIndex, lastRewardEpoch iotago.EpochIndex, err error) {

pkg/protocol/sybilprotection/sybilprotectionv1/sybilprotection.go

@@ -225,6 +225,10 @@ func (o *SybilProtection) CommitSlot(slot iotago.SlotIndex) (committeeRoot iotag
 	return committeeRoot, rewardsRoot, nil
 }
 
+func (o *SybilProtection) RewardsRoot(epoch iotago.EpochIndex) (rewardsRoot iotago.Identifier, err error) {
+	return o.performanceTracker.RewardsRoot(epoch)
+}
+
 func (o *SybilProtection) committeeRoot(targetCommitteeEpoch iotago.EpochIndex) (committeeRoot iotago.Identifier, err error) {
 	committee, exists := o.performanceTracker.LoadCommitteeForEpoch(targetCommitteeEpoch)
 	if !exists {