feat: set govhook

cosmos · Sep 19, 2023 · fbb74fc · fbb74fc
1 parent b0a3721
commit fbb74fc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/app/provider/app.go b/app/provider/app.go
@@ -442,7 +442,7 @@ func New(
 		AddRoute(ibcclienttypes.RouterKey, ibcclient.NewClientProposalHandler(app.IBCKeeper.ClientKeeper))
 	govConfig := govtypes.DefaultConfig()
 
-	app.GovKeeper = *govkeeper.NewKeeper(
+	govKeeper := govkeeper.NewKeeper(
 		appCodec,
 		keys[govtypes.StoreKey],
 		app.AccountKeeper,
@@ -454,7 +454,12 @@ func New(
 	)
 
 	// Set legacy router for backwards compatibility with gov v1beta1
-	app.GovKeeper.SetLegacyRouter(govRouter)
+	govKeeper.SetLegacyRouter(govRouter)
+
+	govHook := app.ProviderKeeper.GovHooks(govKeeper)
+	app.GovKeeper = *govKeeper.SetHooks(
+		govtypes.NewMultiGovHooks(govHook),
+		)
 
 	app.TransferKeeper = ibctransferkeeper.NewKeeper(
 		appCodec,

diff --git a/x/ccv/provider/keeper/gov_hook.go b/x/ccv/provider/keeper/gov_hook.go
@@ -12,15 +12,15 @@ import (
 )
 
 type GovHooks struct {
-	gk govkeeper.Keeper
+	gk *govkeeper.Keeper
 	k  *Keeper
 }
 
 // Implements GovHooks interface
 // GovHooks exist in cosmos-sdk/x/gov/keeper/hooks.go of v0.45.16-lsm-ics and on
 var _ sdkgov.GovHooks = GovHooks{}
 
-func (k *Keeper) GovHooks(gk govkeeper.Keeper) GovHooks {
+func (k *Keeper) GovHooks(gk *govkeeper.Keeper) GovHooks {
 	return GovHooks{
 		gk: gk,
 		k:  k,