fix itc endblocker

OmniFlix · Nov 7, 2024 · 7ddf9e4 · 7ddf9e4
1 parent 1bc33eb
commit 7ddf9e4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/x/itc/abci.go b/x/itc/abci.go
@@ -6,12 +6,12 @@ import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
-func EndBlock(ctx sdk.Context, k keeper.Keeper) []abcitypes.ValidatorUpdate {
+func EndBlock(ctx sdk.Context, k keeper.Keeper) ([]abcitypes.ValidatorUpdate, error) {
 	log := k.Logger(ctx)
 	err := k.FinalizeAndEndCampaigns(ctx)
 	if err != nil {
-		panic(err)
+		return nil, err
 	}
 	log.Info("Updated and processed campaigns .. ")
-	return []abcitypes.ValidatorUpdate{}
+	return []abcitypes.ValidatorUpdate{}, nil
 }
diff --git a/x/itc/module.go b/x/itc/module.go
@@ -164,3 +164,13 @@ func (am AppModule) ExportGenesis(ctx sdk.Context, cdc codec.JSONCodec) json.Raw
 
 // ConsensusVersion implements ConsensusVersion.
 func (AppModule) ConsensusVersion() uint64 { return ConsensusVersion }
+
+// BeginBlock executes all ABCI BeginBlock logic respective to the marketplace module.
+func (am AppModule) BeginBlock(_ sdk.Context) {}
+
+// EndBlock executes all ABCI EndBlock logic respective to the marketplace module. It
+// returns no validator updates.
+func (am AppModule) EndBlock(goCtx context.Context) ([]abci.ValidatorUpdate, error) {
+	ctx := sdk.UnwrapSDKContext(goCtx)
+	return EndBlock(ctx, am.keeper)
+}