fix(upgrade): apply default epoch staking parameters

Author: hussein-aitlahcen

uniond/app/upgrades.go

@@ -2,9 +2,10 @@ package app
 
 import (
 	"fmt"
-	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
 	"union/app/upgrades"
 	"union/app/upgrades/v0_9_0"
+
+	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
 )
 
 var Upgrades = []upgrades.Upgrade{v0_9_0.Upgrade}
@@ -34,6 +35,9 @@ func (app *UnionApp) setupUpgradeHandlers() {
 			upgrade.CreateUpgradeHandler(
 				app.mm,
 				app.configurator,
+				&upgrades.AppKeepers{
+					StakingKeeper: app.StakingKeeper,
+				},
 			),
 		)
 	}

uniond/app/upgrades/types.go

@@ -3,9 +3,14 @@ package upgrades
 import (
 	store "github.com/cosmos/cosmos-sdk/store/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
+	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
 	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
 )
 
+type AppKeepers struct {
+	StakingKeeper *stakingkeeper.Keeper
+}
+
 // source: https://github.com/osmosis-labs/osmosis/blob/c783ef52af8617d3ec613d9ce9035386ba8d4a49/app/upgrades/types.go#L24
 
 // Upgrade defines a struct containing necessary fields that a SoftwareUpgradeProposal
@@ -17,7 +22,7 @@ type Upgrade struct {
 	UpgradeName string
 
 	// CreateUpgradeHandler defines the function that creates an upgrade handler
-	CreateUpgradeHandler func(*module.Manager, module.Configurator) upgradetypes.UpgradeHandler
+	CreateUpgradeHandler func(*module.Manager, module.Configurator, *AppKeepers) upgradetypes.UpgradeHandler
 
 	// Store upgrades, should be used for any new modules introduced, new modules deleted, or store names renamed.
 	StoreUpgrades store.StoreUpgrades

uniond/app/upgrades/v0_9_0/constants.go

@@ -1,9 +1,8 @@
 package v0_9_0
 
 import (
-	"union/app/upgrades"
-
 	store "github.com/cosmos/cosmos-sdk/store/types"
+	"union/app/upgrades"
 )
 
 const UpgradeName = "v0.9.0"

uniond/app/upgrades/v0_9_0/upgrades.go

@@ -1,13 +1,19 @@
 package v0_9_0
 
 import (
+	"union/app/upgrades"
+
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/types/module"
 	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
 )
 
-func CreateUpgradeHandler(mm *module.Manager, configurator module.Configurator) upgradetypes.UpgradeHandler {
+func CreateUpgradeHandler(mm *module.Manager, configurator module.Configurator, keepers *upgrades.AppKeepers) upgradetypes.UpgradeHandler {
 	return func(ctx sdk.Context, plan upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) {
+		stakingParams := keepers.StakingKeeper.GetParams(ctx)
+		stakingParams.EpochLength = 100
+		stakingParams.JailedValidatorThreshold = 50
+		keepers.StakingKeeper.SetParams(ctx, stakingParams)
 		return mm.RunMigrations(ctx, configurator, vm)
 	}
 }