Keeper & genesis bugfix

lum-network · Apr 28, 2022 · 942b900 · 942b900
1 parent 47a1ce9
commit 942b900
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/x/beam/genesis.go b/x/beam/genesis.go
@@ -11,6 +11,9 @@ import (
 func InitGenesis(ctx sdk.Context, k keeper.Keeper, genState types.GenesisState) (res []abci.ValidatorUpdate) {
 	k.CreateBeamModuleAccount(ctx, genState.ModuleAccountBalance)
 
+	for _, beam := range genState.Beams {
+		k.SetBeam(ctx, beam.GetId(), beam)
+	}
 	return nil
 }
 
@@ -19,7 +22,7 @@ func ExportGenesis(ctx sdk.Context, k keeper.Keeper) *types.GenesisState {
 	beams := k.ListBeams(ctx)
 
 	return &types.GenesisState{
-		Beams: beams,
+		Beams:                beams,
 		ModuleAccountBalance: k.GetBeamAccountBalance(ctx),
 	}
 }
diff --git a/x/beam/keeper/keeper.go b/x/beam/keeper/keeper.go
@@ -7,10 +7,8 @@ import (
 	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
 	bankkeeper "github.com/cosmos/cosmos-sdk/x/bank/keeper"
 	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
-	"strings"
-	"time"
-
 	"github.com/tendermint/tendermint/libs/log"
+	"strings"
 
 	"github.com/cosmos/cosmos-sdk/codec"
 	sdk "github.com/cosmos/cosmos-sdk/types"
@@ -253,7 +251,7 @@ func (k Keeper) OpenBeam(ctx sdk.Context, msg types.MsgOpenBeam) error {
 		CancelReason:   "",
 		Schema:         msg.GetSchema(),
 		Data:           msg.GetData(),
-		CreatedAt:      time.Now(),
+		CreatedAt:      ctx.BlockTime(),
 	}
 
 	if msg.GetAmount() != nil && msg.GetAmount().IsPositive() {