Merge pull request #294 from scorpioborn/feature/upgrade-handler-params

Fix / Upgrade handler for max withdraw count

app/upgrades/v2/upgrades.go

@@ -5,7 +5,9 @@ import (
 	"github.com/cosmos/cosmos-sdk/types/module"
 	upgradetypes "github.com/cosmos/cosmos-sdk/x/upgrade/types"
 
+	sdkmath "cosmossdk.io/math"
 	"github.com/sge-network/sge/app/keepers"
+	housetypes "github.com/sge-network/sge/x/house/types"
 )
 
 func CreateUpgradeHandler(
@@ -14,6 +16,24 @@ func CreateUpgradeHandler(
 	keepers *keepers.AppKeepers,
 ) upgradetypes.UpgradeHandler {
 	return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
+		housePS := keepers.GetSubspace(housetypes.ModuleName)
+
+		if !housePS.Has(ctx, []byte("MaxWithdrawalCount")) {
+			var minDeposit sdkmath.Int
+			housePS.Get(ctx, []byte("MinDeposit"), &minDeposit)
+
+			var houseParticipationFee sdk.Dec
+			housePS.Get(ctx, []byte("HouseParticipationFee"), &houseParticipationFee)
+
+			p := housetypes.NewParams(
+				minDeposit,
+				houseParticipationFee,
+				housetypes.DefaultMaxWithdrawalCount,
+			)
+
+			keepers.HouseKeeper.SetParams(ctx, p)
+		}
+
 		return mm.RunMigrations(ctx, configurator, fromVM)
 	}
 }