Merge pull request #12770 from filecoin-project/kubuxu/state/claim-check

* feat: optimize ForEachClaim to return only eligible claims

Avoids double AMT traversal

Signed-off-by: Jakub Sztandera <oss@kubuxu.com>

* add comment, bump go-state-types

Signed-off-by: Jakub Sztandera <oss@kubuxu.com>

---------

Signed-off-by: Jakub Sztandera <oss@kubuxu.com>

chain/actors/builtin/power/actor.go.template

@@ -99,7 +99,10 @@ type State interface {
 	MinerPower(address.Address) (Claim, bool, error)
 	MinerNominalPowerMeetsConsensusMinimum(address.Address) (bool, error)
 	ListAllMiners() ([]address.Address, error)
-	ForEachClaim(func(miner address.Address, claim Claim) error) error
+	// ForEachClaim iterates over claims in the power actor.
+	// If onlyEligible is true, it applies the MinerNominalPowerMeetsConsensusMinimum check
+	// before returning the actor.
+	ForEachClaim(cb func(miner address.Address, claim Claim) error, onlyEligible bool) error
 	ClaimsChanged(State) (bool, error)
 
 	// Testing or genesis setup only

chain/actors/builtin/power/power.go

@@ -183,7 +183,10 @@ type State interface {
 	MinerPower(address.Address) (Claim, bool, error)
 	MinerNominalPowerMeetsConsensusMinimum(address.Address) (bool, error)
 	ListAllMiners() ([]address.Address, error)
-	ForEachClaim(func(miner address.Address, claim Claim) error) error
+	// ForEachClaim iterates over claims in the power actor.
+	// If onlyEligible is true, it applies the MinerNominalPowerMeetsConsensusMinimum check
+	// before returning the actor.
+	ForEachClaim(cb func(miner address.Address, claim Claim) error, onlyEligible bool) error
 	ClaimsChanged(State) (bool, error)
 
 	// Testing or genesis setup only

chain/actors/builtin/power/state.go.template

@@ -147,7 +147,7 @@ func (s *state{{.v}}) ListAllMiners() ([]address.Address, error) {
 	return miners, nil
 }
 
-func (s *state{{.v}}) ForEachClaim(cb func(miner address.Address, claim Claim) error) error {
+func (s *state{{.v}}) ForEachClaim(cb func(miner address.Address, claim Claim) error, onlyEligible bool) error {
 	claims, err := s.claims()
 	if err != nil {
 		return err
@@ -159,10 +159,28 @@ func (s *state{{.v}}) ForEachClaim(cb func(miner address.Address, claim Claim) e
 		if err != nil {
 			return err
 		}
-		return cb(a, Claim{
-			RawBytePower:    claim.RawBytePower,
-			QualityAdjPower: claim.QualityAdjPower,
-		})
+		if !onlyEligible {
+			return cb(a, Claim{
+				RawBytePower:    claim.RawBytePower,
+				QualityAdjPower: claim.QualityAdjPower,
+			})
+		}
+		{{if (ge .v 8)}}
+		eligible, err := s.State.ClaimMeetsConsensusMinimums(&claim)
+		{{else}}
+		//slow path
+		eligible, err := s.State.MinerNominalPowerMeetsConsensusMinimum(s.store, a)
+		{{end}}
+		if err != nil {
+			return fmt.Errorf("checking consensus minimums: %w", err)
+		}
+		if eligible {
+			return cb(a, Claim{
+				RawBytePower:    claim.RawBytePower,
+				QualityAdjPower: claim.QualityAdjPower,
+			})
+		}
+		return nil
 	})
 }