migrator

Author: oren-lava

x/epochstorage/keeper/migrations.go

@@ -124,3 +124,25 @@ func (m Migrator) SetEpochHashForMigrator(ctx sdk.Context, epoch uint64, hash []
 		panic(err)
 	}
 }
+
+// fix provider metadata to be consistent with the stake entries
+func (m Migrator) MigrateVersion8To9(ctx sdk.Context) error {
+	allMetadata, err := m.keeper.GetAllMetadata(ctx)
+	if err != nil {
+		return err
+	}
+
+	for _, metadata := range allMetadata {
+		chains := []string{}
+		for _, chainID := range metadata.Chains {
+			stakeEntry, found := m.keeper.GetStakeEntryCurrent(ctx, chainID, metadata.Provider)
+			if found {
+				chains = append(chains, stakeEntry.Chain)
+			}
+		}
+		metadata.Chains = chains
+		m.keeper.SetMetadata(ctx, metadata)
+	}
+
+	return nil
+}

x/epochstorage/module.go

@@ -135,6 +135,11 @@ func (am AppModule) RegisterServices(cfg module.Configurator) {
 		// panic:ok: at start up, migration cannot proceed anyhow
 		panic(fmt.Errorf("%s: failed to register migration to v8: %w", types.ModuleName, err))
 	}
+
+	// register v8 -> v9 migration
+	if err := cfg.RegisterMigration(types.ModuleName, 8, migrator.MigrateVersion8To9); err != nil {
+		panic(fmt.Errorf("%s: failed to register migration to v9: %w", types.ModuleName, err))
+	}
 }
 
 // RegisterInvariants registers the capability module's invariants.