153 lines
4.2 KiB
Go
153 lines
4.2 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"cosmossdk.io/simapp"
|
|
storetypes "cosmossdk.io/store/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
banktestutil "github.com/cosmos/cosmos-sdk/x/bank/testutil"
|
|
"github.com/cosmos/cosmos-sdk/x/staking/types"
|
|
)
|
|
|
|
func BenchmarkGetValidator(b *testing.B) {
|
|
// 900 is the max number we are allowed to use in order to avoid simtestutil.CreateTestPubKeys
|
|
// panic: encoding/hex: odd length hex string
|
|
powersNumber := 900
|
|
|
|
var totalPower int64
|
|
powers := make([]int64, powersNumber)
|
|
for i := range powers {
|
|
powers[i] = int64(i)
|
|
totalPower += int64(i)
|
|
}
|
|
|
|
app, ctx, _, valAddrs, vals := initValidators(b, totalPower, len(powers), powers)
|
|
|
|
for _, validator := range vals {
|
|
app.StakingKeeper.SetValidator(ctx, validator)
|
|
}
|
|
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
for _, addr := range valAddrs {
|
|
_, _ = app.StakingKeeper.GetValidator(ctx, addr)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkGetValidatorDelegations(b *testing.B) {
|
|
var totalPower int64
|
|
powersNumber := 10
|
|
|
|
powers := make([]int64, powersNumber)
|
|
for i := range powers {
|
|
powers[i] = int64(i)
|
|
totalPower += int64(i)
|
|
}
|
|
|
|
app, ctx, _, valAddrs, vals := initValidators(b, totalPower, len(powers), powers)
|
|
for _, validator := range vals {
|
|
app.StakingKeeper.SetValidator(ctx, validator)
|
|
}
|
|
|
|
delegationsNum := 1000
|
|
for _, val := range valAddrs {
|
|
for i := 0; i < delegationsNum; i++ {
|
|
delegator := sdk.AccAddress(fmt.Sprintf("address%d", i))
|
|
banktestutil.FundAccount(ctx, app.BankKeeper, delegator,
|
|
sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(int64(i)))))
|
|
NewDel := types.NewDelegation(delegator, val, sdk.NewDec(int64(i)))
|
|
app.StakingKeeper.SetDelegation(ctx, NewDel)
|
|
}
|
|
}
|
|
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
updateValidatorDelegations(ctx, app, valAddrs[0], sdk.ValAddress("val"))
|
|
}
|
|
}
|
|
|
|
func BenchmarkGetValidatorDelegationsLegacy(b *testing.B) {
|
|
var totalPower int64
|
|
powersNumber := 10
|
|
|
|
powers := make([]int64, powersNumber)
|
|
for i := range powers {
|
|
powers[i] = int64(i)
|
|
totalPower += int64(i)
|
|
}
|
|
|
|
app, ctx, _, valAddrs, vals := initValidators(b, totalPower, len(powers), powers)
|
|
|
|
for _, validator := range vals {
|
|
app.StakingKeeper.SetValidator(ctx, validator)
|
|
}
|
|
|
|
delegationsNum := 1000
|
|
for _, val := range valAddrs {
|
|
for i := 0; i < delegationsNum; i++ {
|
|
delegator := sdk.AccAddress(fmt.Sprintf("address%d", i))
|
|
banktestutil.FundAccount(ctx, app.BankKeeper, delegator,
|
|
sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(int64(i)))))
|
|
NewDel := types.NewDelegation(delegator, val, sdk.NewDec(int64(i)))
|
|
app.StakingKeeper.SetDelegation(ctx, NewDel)
|
|
}
|
|
}
|
|
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
updateValidatorDelegationsLegacy(ctx, app, valAddrs[0], sdk.ValAddress("val"))
|
|
}
|
|
}
|
|
|
|
func updateValidatorDelegationsLegacy(ctx sdk.Context, app *simapp.SimApp, existingValAddr, newValAddr sdk.ValAddress) {
|
|
storeKey := app.GetKey(types.StoreKey)
|
|
cdc, k := app.AppCodec(), app.StakingKeeper
|
|
|
|
store := ctx.KVStore(storeKey)
|
|
|
|
iterator := storetypes.KVStorePrefixIterator(store, types.DelegationKey)
|
|
defer iterator.Close()
|
|
|
|
for ; iterator.Valid(); iterator.Next() {
|
|
delegation := types.MustUnmarshalDelegation(cdc, iterator.Value())
|
|
if delegation.GetValidatorAddr().Equals(existingValAddr) {
|
|
k.RemoveDelegation(ctx, delegation)
|
|
delegation.ValidatorAddress = newValAddr.String()
|
|
k.SetDelegation(ctx, delegation)
|
|
}
|
|
}
|
|
}
|
|
|
|
func updateValidatorDelegations(ctx sdk.Context, app *simapp.SimApp, existingValAddr, newValAddr sdk.ValAddress) {
|
|
storeKey := app.GetKey(types.StoreKey)
|
|
cdc, k := app.AppCodec(), app.StakingKeeper
|
|
|
|
store := ctx.KVStore(storeKey)
|
|
|
|
itr := storetypes.KVStorePrefixIterator(store, types.GetDelegationsByValPrefixKey(existingValAddr))
|
|
defer itr.Close()
|
|
|
|
for ; itr.Valid(); itr.Next() {
|
|
key := itr.Key()
|
|
valAddr, delAddr, err := types.ParseDelegationsByValKey(key)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
bz := store.Get(types.GetDelegationKey(delAddr, valAddr))
|
|
delegation := types.MustUnmarshalDelegation(cdc, bz)
|
|
|
|
// remove old operator addr from delegation
|
|
if err := k.RemoveDelegation(ctx, delegation); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
delegation.ValidatorAddress = newValAddr.String()
|
|
// add with new operator addr
|
|
k.SetDelegation(ctx, delegation)
|
|
}
|
|
}
|