safely migrate

lavanet · Feb 11, 2025 · 4262e5f · 4262e5f
1 parent 3fd90ad
commit 4262e5f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/x/epochstorage/keeper/migrations.go b/x/epochstorage/keeper/migrations.go
@@ -133,6 +133,14 @@ func (m Migrator) MigrateVersion8To9(ctx sdk.Context) error {
 	}
 
 	for _, metadata := range allMetadata {
+		if metadata.Provider != metadata.Vault {
+			_, err := m.keeper.GetMetadata(ctx, metadata.Vault)
+			if err == nil {
+				fmt.Println("exists for vault with different provider: ", metadata.Vault, " and provider: ", metadata.Provider)
+				continue
+			}
+		}
+
 		chains := []string{}
 		for _, chainID := range metadata.Chains {
 			stakeEntry, found := m.keeper.GetStakeEntryCurrent(ctx, chainID, metadata.Provider)
@@ -144,6 +152,7 @@ func (m Migrator) MigrateVersion8To9(ctx sdk.Context) error {
 		}
 		metadata.Chains = chains
 		m.keeper.SetMetadata(ctx, metadata)
+
 	}
 
 	return nil

diff --git a/x/epochstorage/module.go b/x/epochstorage/module.go
@@ -164,7 +164,7 @@ func (am AppModule) ExportGenesis(ctx sdk.Context, cdc codec.JSONCodec) json.Raw
 }
 
 // ConsensusVersion implements ConsensusVersion.
-func (AppModule) ConsensusVersion() uint64 { return 8 }
+func (AppModule) ConsensusVersion() uint64 { return 9 }
 
 // BeginBlock executes all ABCI BeginBlock logic respective to the capability module.
 func (am AppModule) BeginBlock(ctx sdk.Context, _ abci.RequestBeginBlock) {