133 lines
3.9 KiB
Go
133 lines
3.9 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cosmossdk.io/collections"
|
|
"cosmossdk.io/x/staking/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// GetDelegatorValidators returns all validators that a delegator is bonded to. If maxRetrieve is supplied, the respective amount will be returned.
|
|
func (k Keeper) GetDelegatorValidators(
|
|
ctx context.Context, delegatorAddr sdk.AccAddress, maxRetrieve uint32,
|
|
) (types.Validators, error) {
|
|
validators := make([]types.Validator, maxRetrieve)
|
|
|
|
var i uint32
|
|
rng := collections.NewPrefixedPairRange[sdk.AccAddress, sdk.ValAddress](delegatorAddr)
|
|
err := k.Delegations.Walk(ctx, rng, func(key collections.Pair[sdk.AccAddress, sdk.ValAddress], del types.Delegation) (stop bool, err error) {
|
|
if i >= maxRetrieve {
|
|
return true, nil
|
|
}
|
|
|
|
valAddr, err := k.validatorAddressCodec.StringToBytes(del.GetValidatorAddr())
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
validator, err := k.GetValidator(ctx, valAddr)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
validators[i] = validator
|
|
i++
|
|
|
|
return false, nil
|
|
})
|
|
if err != nil {
|
|
return types.Validators{}, err
|
|
}
|
|
|
|
return types.Validators{Validators: validators[:i], ValidatorCodec: k.validatorAddressCodec}, nil // trim
|
|
}
|
|
|
|
// GetDelegatorValidator returns a validator that a delegator is bonded to
|
|
func (k Keeper) GetDelegatorValidator(
|
|
ctx context.Context, delegatorAddr sdk.AccAddress, validatorAddr sdk.ValAddress,
|
|
) (validator types.Validator, err error) {
|
|
delegation, err := k.Delegations.Get(ctx, collections.Join(delegatorAddr, validatorAddr))
|
|
if err != nil {
|
|
return validator, err
|
|
}
|
|
|
|
valAddr, err := k.validatorAddressCodec.StringToBytes(delegation.GetValidatorAddr())
|
|
if err != nil {
|
|
return validator, err
|
|
}
|
|
|
|
return k.GetValidator(ctx, valAddr)
|
|
}
|
|
|
|
// GetAllDelegatorDelegations returns all delegations of a delegator
|
|
func (k Keeper) GetAllDelegatorDelegations(ctx context.Context, delegator sdk.AccAddress) ([]types.Delegation, error) {
|
|
delegations := make([]types.Delegation, 0)
|
|
|
|
var i int64
|
|
rng := collections.NewPrefixedPairRange[sdk.AccAddress, sdk.ValAddress](delegator)
|
|
err := k.Delegations.Walk(ctx, rng, func(key collections.Pair[sdk.AccAddress, sdk.ValAddress], del types.Delegation) (stop bool, err error) {
|
|
delegations = append(delegations, del)
|
|
i++
|
|
|
|
return false, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return delegations, nil
|
|
}
|
|
|
|
// GetAllUnbondingDelegations returns all unbonding-delegations of a delegator
|
|
func (k Keeper) GetAllUnbondingDelegations(ctx context.Context, delegator sdk.AccAddress) ([]types.UnbondingDelegation, error) {
|
|
unbondingDelegations := make([]types.UnbondingDelegation, 0)
|
|
|
|
rng := collections.NewPrefixUntilPairRange[[]byte, []byte](delegator)
|
|
err := k.UnbondingDelegations.Walk(
|
|
ctx,
|
|
rng,
|
|
func(key collections.Pair[[]byte, []byte], value types.UnbondingDelegation) (stop bool, err error) {
|
|
unbondingDelegations = append(unbondingDelegations, value)
|
|
return false, nil
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return unbondingDelegations, nil
|
|
}
|
|
|
|
// GetAllRedelegations returns all redelegations of a delegator
|
|
func (k Keeper) GetAllRedelegations(
|
|
ctx context.Context, delegator sdk.AccAddress, srcValAddress, dstValAddress sdk.ValAddress,
|
|
) ([]types.Redelegation, error) {
|
|
srcValFilter := !(srcValAddress.Empty())
|
|
dstValFilter := !(dstValAddress.Empty())
|
|
|
|
redelegations := []types.Redelegation{}
|
|
rng := collections.NewPrefixedTripleRange[[]byte, []byte, []byte](delegator)
|
|
err := k.Redelegations.Walk(ctx, rng,
|
|
func(key collections.Triple[[]byte, []byte, []byte], redelegation types.Redelegation) (stop bool, err error) {
|
|
valSrcAddr, valDstAddr := key.K2(), key.K3()
|
|
|
|
if srcValFilter && !(srcValAddress.Equals(sdk.ValAddress(valSrcAddr))) {
|
|
return false, nil
|
|
}
|
|
|
|
if dstValFilter && !(dstValAddress.Equals(sdk.ValAddress(valDstAddr))) {
|
|
return false, nil
|
|
}
|
|
|
|
redelegations = append(redelegations, redelegation)
|
|
return false, nil
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return redelegations, nil
|
|
}
|