0g-chain/x/committee/genesis.go

59 lines
1.3 KiB
Go
Raw Normal View History

2020-03-11 19:27:36 +00:00
package committee
import (
"fmt"
sdk "github.com/cosmos/cosmos-sdk/types"
2020-03-15 00:00:05 +00:00
"github.com/kava-labs/kava/x/committee/types"
2020-03-11 19:27:36 +00:00
)
// InitGenesis initializes the store state from a genesis state.
func InitGenesis(ctx sdk.Context, keeper Keeper, gs GenesisState) {
if err := gs.Validate(); err != nil {
panic(fmt.Sprintf("failed to validate %s genesis state: %s", ModuleName, err))
}
keeper.SetNextProposalID(ctx, gs.NextProposalID)
2020-03-15 00:00:05 +00:00
for _, com := range gs.Committees {
keeper.SetCommittee(ctx, com)
}
for _, p := range gs.Proposals {
keeper.SetProposal(ctx, p)
}
for _, v := range gs.Votes {
keeper.SetVote(ctx, v)
}
2020-03-11 19:27:36 +00:00
}
// ExportGenesis returns a GenesisState for a given context and keeper.
func ExportGenesis(ctx sdk.Context, keeper Keeper) GenesisState {
2020-03-15 00:00:05 +00:00
nextID, err := keeper.GetNextProposalID(ctx)
if err != nil {
panic(err)
}
committees := []types.Committee{}
keeper.IterateCommittees(ctx, func(com types.Committee) bool {
committees = append(committees, com)
return false
})
proposals := []types.Proposal{}
votes := []types.Vote{}
keeper.IterateProposals(ctx, func(p types.Proposal) bool {
proposals = append(proposals, p)
keeper.IterateVotes(ctx, p.ID, func(v types.Vote) bool {
votes = append(votes, v)
return false
})
return false
})
return types.NewGenesisState(
nextID,
committees,
proposals,
votes,
)
2020-03-11 19:27:36 +00:00
}