Add simple WriteGenesis for stake & cool modules

KKnight99 · Apr 28, 2018 · f71191e · f71191e
1 parent ff66629
commit f71191e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/cmd/gaia/app/app.go b/cmd/gaia/app/app.go
@@ -113,7 +113,7 @@ func (app *GaiaApp) initChainer(ctx sdk.Context, req abci.RequestInitChain) abci
 	}
 
 	// load the initial stake information
-	stake.InitGenesis(ctx, app.stakeKeeper, genesisState.StakeData)
+	app.stakeKeeper.InitGenesis(ctx, genesisState.StakeData)
 
 	return abci.ResponseInitChain{}
 }
diff --git a/examples/democoin/x/cool/keeper.go b/examples/democoin/x/cool/keeper.go
@@ -48,3 +48,9 @@ func (k Keeper) InitGenesis(ctx sdk.Context, data Genesis) error {
 	k.setTrend(ctx, data.Trend)
 	return nil
 }
+
+// WriteGenesis - output the genesis trend
+func (k Keeper) WriteGenesis(ctx sdk.Context) Genesis {
+	trend := k.GetTrend(ctx)
+	return Genesis{trend}
+}
diff --git a/x/stake/handler.go b/x/stake/handler.go
@@ -49,14 +49,21 @@ func NewEndBlocker(k Keeper) sdk.EndBlocker {
 //_____________________________________________________________________
 
 // InitGenesis - store genesis parameters
-func InitGenesis(ctx sdk.Context, k Keeper, data GenesisState) {
+func (k Keeper) InitGenesis(ctx sdk.Context, data GenesisState) {
 	k.setPool(ctx, data.Pool)
 	k.setParams(ctx, data.Params)
 	for _, candidate := range data.Candidates {
 		k.setCandidate(ctx, candidate)
 	}
 }
 
+// WriteGenesis - output genesis parameters
+func (k Keeper) WriteGenesis(ctx sdk.Context) GenesisState {
+	pool := k.GetPool(ctx)
+	params := k.GetParams(ctx)
+	return GenesisState{pool, params}
+}
+
 //_____________________________________________________________________
 
 // These functions assume everything has been authenticated,