2018-05-12 15:22:59 -07:00
|
|
|
package stake
|
|
|
|
|
|
|
|
import sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
|
|
|
|
// GenesisState - all staking state that must be provided at genesis
|
|
|
|
type GenesisState struct {
|
|
|
|
Pool Pool `json:"pool"`
|
|
|
|
Params Params `json:"params"`
|
|
|
|
Validators []Validator `json:"validators"`
|
|
|
|
Bonds []Delegation `json:"bonds"`
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewGenesisState(pool Pool, params Params, validators []Validator, bonds []Delegation) GenesisState {
|
|
|
|
return GenesisState{
|
|
|
|
Pool: pool,
|
|
|
|
Params: params,
|
|
|
|
Validators: validators,
|
|
|
|
Bonds: bonds,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// get raw genesis raw message for testing
|
|
|
|
func DefaultGenesisState() GenesisState {
|
|
|
|
return GenesisState{
|
|
|
|
Pool: initialPool(),
|
|
|
|
Params: defaultParams(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// InitGenesis - store genesis parameters
|
|
|
|
func InitGenesis(ctx sdk.Context, k Keeper, data GenesisState) {
|
|
|
|
k.setPool(ctx, data.Pool)
|
2018-05-17 13:03:35 -07:00
|
|
|
k.setNewParams(ctx, data.Params)
|
2018-05-12 15:22:59 -07:00
|
|
|
for _, validator := range data.Validators {
|
2018-05-28 23:49:37 -07:00
|
|
|
// Staking assumes bonded validators are already stored, need to force update
|
|
|
|
validator.PoolShares.Status = sdk.Unbonded
|
2018-05-22 12:13:03 -07:00
|
|
|
k.updateValidator(ctx, validator)
|
2018-05-12 15:22:59 -07:00
|
|
|
}
|
|
|
|
for _, bond := range data.Bonds {
|
|
|
|
k.setDelegation(ctx, bond)
|
|
|
|
}
|
2018-05-28 23:49:37 -07:00
|
|
|
store := ctx.KVStore(k.storeKey)
|
|
|
|
k.updateBondedValidatorsFull(ctx, store)
|
2018-05-12 15:22:59 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// WriteGenesis - output genesis parameters
|
|
|
|
func WriteGenesis(ctx sdk.Context, k Keeper) GenesisState {
|
|
|
|
pool := k.GetPool(ctx)
|
|
|
|
params := k.GetParams(ctx)
|
2018-05-17 11:09:35 -07:00
|
|
|
validators := k.getAllValidators(ctx)
|
|
|
|
bonds := k.getAllDelegations(ctx)
|
2018-05-12 15:22:59 -07:00
|
|
|
return GenesisState{
|
|
|
|
pool,
|
|
|
|
params,
|
|
|
|
validators,
|
|
|
|
bonds,
|
|
|
|
}
|
|
|
|
}
|