-
Notifications
You must be signed in to change notification settings - Fork 0
/
genesis.go
executable file
·113 lines (89 loc) · 4.02 KB
/
genesis.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package simulation
import (
"encoding/json"
"fmt"
"math/rand"
"time"
sdkmath "cosmossdk.io/math"
"github.com/axiome-pro/axm-node/x/staking/types"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/types/module"
"github.com/cosmos/cosmos-sdk/types/simulation"
)
// Simulation parameter constants
const (
unbondingTime = "unbonding_time"
maxValidators = "max_validators"
historicalEntries = "historical_entries"
maxMonthlyPoints = "max_monthly_points"
minSelfDelegation = "min_self_delegation"
)
// genUnbondingTime returns randomized UnbondingTime
func genUnbondingTime(r *rand.Rand) (ubdTime time.Duration) {
return time.Duration(simulation.RandIntBetween(r, 60, 60*60*24*3*2)) * time.Second
}
// genMaxValidators returns randomized MaxValidators
func genMaxValidators(r *rand.Rand) (maxValidators uint32) {
return uint32(r.Intn(250) + 1)
}
// getHistEntries returns randomized HistoricalEntries between 0-100.
func getHistEntries(r *rand.Rand) uint32 {
return uint32(r.Intn(int(types.DefaultHistoricalEntries + 1)))
}
// getMaxMonthlyPoints returns randomized HistoricalEntries between 0-100.
func getMaxMonthlyPoints(r *rand.Rand) uint64 {
return uint64(r.Intn(int(types.DefaultMaximumMonthlyPoints + 1)))
}
// getHistEntries returns randomized HistoricalEntries between 0-100.
func getMinSelfDelegation(r *rand.Rand) sdkmath.Int {
return sdkmath.NewInt(r.Int63n(types.DefaultMinSelfDelegation.AddRaw(1).Int64()))
}
// RandomizedGenState generates a random GenesisState for staking
func RandomizedGenState(simState *module.SimulationState) {
// params
var (
unbondTime time.Duration
maxVals uint32
histEntries uint32
validatorEmissionRate sdkmath.LegacyDec
maximumMonthlyPoints uint64
minimumSelfDelegation sdkmath.Int
)
simState.AppParams.GetOrGenerate(unbondingTime, &unbondTime, simState.Rand, func(r *rand.Rand) { unbondTime = genUnbondingTime(r) })
simState.AppParams.GetOrGenerate(maxValidators, &maxVals, simState.Rand, func(r *rand.Rand) { maxVals = genMaxValidators(r) })
simState.AppParams.GetOrGenerate(historicalEntries, &histEntries, simState.Rand, func(r *rand.Rand) { histEntries = getHistEntries(r) })
simState.AppParams.GetOrGenerate(maxMonthlyPoints, &maximumMonthlyPoints, simState.Rand, func(r *rand.Rand) { maximumMonthlyPoints = getMaxMonthlyPoints(r) })
simState.AppParams.GetOrGenerate(maxMonthlyPoints, &minimumSelfDelegation, simState.Rand, func(r *rand.Rand) { minimumSelfDelegation = getMinSelfDelegation(r) })
// NOTE: the slashing module need to be defined after the staking module on the
// NewSimulationManager constructor for this to work
simState.UnbondTime = unbondTime
params := types.NewParams(simState.UnbondTime, simState.UnbondTime, maxVals, 7, histEntries,
simState.BondDenom, maximumMonthlyPoints, validatorEmissionRate, []*types.EmissionRange{&types.DefaultEmissionRange},
minimumSelfDelegation)
// validators & delegations
var (
validators []types.Validator
delegations []types.Delegation
)
valAddrs := make([]sdk.ValAddress, simState.NumBonded)
for i := 0; i < int(simState.NumBonded); i++ {
valAddr := sdk.ValAddress(simState.Accounts[i].Address)
valAddrs[i] = valAddr
validator, err := types.NewValidator(valAddr.String(), simState.Accounts[i].ConsKey.PubKey(), types.Description{})
if err != nil {
panic(err)
}
validator.Tokens = simState.InitialStake
validator.DelegatorShares = sdkmath.LegacyNewDecFromInt(simState.InitialStake)
delegation := types.NewDelegation(simState.Accounts[i].Address.String(), valAddr.String(), sdkmath.LegacyNewDecFromInt(simState.InitialStake))
validators = append(validators, validator)
delegations = append(delegations, delegation)
}
stakingGenesis := types.NewGenesisState(params, validators, delegations)
bz, err := json.MarshalIndent(&stakingGenesis.Params, "", " ")
if err != nil {
panic(err)
}
fmt.Printf("Selected randomly generated staking parameters:\n%s\n", bz)
simState.GenState[types.ModuleName] = simState.Cdc.MustMarshalJSON(stakingGenesis)
}