fix(ledger): use mark snapshot for stake distribution (#1434)

Signed-off-by: Chris Gianelloni <wolf31o2@blinklabs.io>

Author: wolf31o2

ledger/view.go

@@ -570,13 +570,14 @@ type StakeDistribution struct {
 }
 
 // GetStakeDistribution returns the stake distribution for leader election.
-// Uses the "go" snapshot which represents stake from 2 epochs ago.
+// Uses the "mark" snapshot at the given epoch. Callers pass currentEpoch-2
+// so the epoch offset already accounts for the Mark→Set→Go rotation.
 func (lv *LedgerView) GetStakeDistribution(
 	epoch uint64,
 ) (*StakeDistribution, error) {
 	snapshots, err := lv.ls.db.Metadata().GetPoolStakeSnapshotsByEpoch(
 		epoch,
-		"go",
+		"mark",
 		(*lv.txn).Metadata(),
 	)
 	if err != nil {
@@ -598,15 +599,16 @@ func (lv *LedgerView) GetStakeDistribution(
 	return dist, nil
 }
 
-// GetPoolStake returns the stake for a specific pool from the "go" snapshot.
-// Returns 0 if the pool has no stake in the snapshot.
+// GetPoolStake returns the stake for a specific pool from the snapshot.
+// Returns 0 if the pool has no stake in the snapshot. Callers pass
+// currentEpoch-2 so the epoch offset accounts for Mark→Set→Go rotation.
 func (lv *LedgerView) GetPoolStake(
 	epoch uint64,
 	poolKeyHash []byte,
 ) (uint64, error) {
 	snapshot, err := lv.ls.db.Metadata().GetPoolStakeSnapshot(
 		epoch,
-		"go",
+		"mark",
 		poolKeyHash,
 		(*lv.txn).Metadata(),
 	)
@@ -619,11 +621,12 @@ func (lv *LedgerView) GetPoolStake(
 	return uint64(snapshot.TotalStake), nil
 }
 
-// GetTotalActiveStake returns the total active stake from the "go" snapshot.
+// GetTotalActiveStake returns the total active stake from the snapshot.
+// Callers pass currentEpoch-2 so the epoch offset accounts for rotation.
 func (lv *LedgerView) GetTotalActiveStake(epoch uint64) (uint64, error) {
 	return lv.ls.db.Metadata().GetTotalActiveStake(
 		epoch,
-		"go",
+		"mark",
 		(*lv.txn).Metadata(),
 	)
 }