Added migration to fix entities

x/millions/keeper/keeper_deposit.go

@@ -388,3 +388,14 @@ func (k Keeper) ListDeposits(ctx sdk.Context) (deposits []types.Deposit) {
 	}
 	return
 }
+
+func (k Keeper) UnsafeSetUnpersistedDeposits(ctx sdk.Context) int {
+	i := 0
+	for _, deposit := range k.ListDeposits(ctx) {
+		k.setPoolDeposit(ctx, &deposit)
+		k.setAccountDeposit(ctx, &deposit)
+		i++
+	}
+
+	return i
+}

x/millions/keeper/keeper_withdrawal.go

@@ -416,3 +416,14 @@ func (k Keeper) ListPoolWithdrawals(ctx sdk.Context, poolID uint64) (withdrawals
 	}
 	return
 }
+
+func (k Keeper) UnsafeSetUnpersistedWithdrawals(ctx sdk.Context) int {
+	i := 0
+	for _, withdrawal := range k.ListWithdrawals(ctx) {
+		k.setPoolWithdrawal(ctx, withdrawal)
+		k.setAccountWithdrawal(ctx, withdrawal)
+		i++
+	}
+
+	return i
+}

x/millions/migrations/migrator.go

@@ -2,6 +2,7 @@ package migrations
 
 import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	v164 "github.com/lum-network/chain/x/millions/migrations/v164"
 
 	millionskeeper "github.com/lum-network/chain/x/millions/keeper"
 	v150 "github.com/lum-network/chain/x/millions/migrations/v150"
@@ -37,3 +38,8 @@ func (m Migrator) Migrate3To4(ctx sdk.Context) error {
 func (m Migrator) Migrate4To5(ctx sdk.Context) error {
 	return v162.MigratePendingWithdrawalsToNewEpochUnbonding(ctx, m.keeper)
 }
+
+// Migrate5To6 migrates from version 5 to 6
+func (m Migrator) Migrate5To6(ctx sdk.Context) error {
+	return v164.SaveEntitiesOnAccountLevel(ctx, m.keeper)
+}

x/millions/migrations/v164/store.go

@@ -0,0 +1,17 @@
+package v164
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	millionskeeper "github.com/lum-network/chain/x/millions/keeper"
+)
+
+func SaveEntitiesOnAccountLevel(ctx sdk.Context, k millionskeeper.Keeper) error {
+	ctx.Logger().Info("Saving entities on account level")
+
+	updatedDeposits := k.UnsafeSetUnpersistedDeposits(ctx)
+	updatedWithdrawals := k.UnsafeSetUnpersistedWithdrawals(ctx)
+
+	ctx.Logger().Info("Saving entities on account level: deposits", "count", updatedDeposits)
+	ctx.Logger().Info("Saving entities on account level: withdrawals", "count", updatedWithdrawals)
+	return nil
+}

x/millions/module.go

@@ -145,6 +145,11 @@ func (a AppModule) RegisterServices(cfg module.Configurator) {
 	if err != nil {
 		panic(err)
 	}
+
+	err = cfg.RegisterMigration(types.ModuleName, 5, migrator.Migrate5To6)
+	if err != nil {
+		panic(err)
+	}
 }
 
 func (a AppModule) BeginBlock(ctx sdk.Context, _ abci.RequestBeginBlock) {

x/millions/types/keys.go

@@ -14,7 +14,7 @@ import (
 const (
 	ModuleName = "millions"
 
-	ModuleVersion = 5
+	ModuleVersion = 6
 
 	StoreKey = ModuleName