run migration to fix up storage-used at the end

onflow · Apr 15, 2024 · d65c151 · d65c151
1 parent 2f9ffa7
commit d65c151
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/cmd/util/cmd/execution-state-extract/execution_state_extract.go b/cmd/util/cmd/execution-state-extract/execution_state_extract.go
@@ -395,6 +395,18 @@ func newMigrations(
 		migrations = append(migrations, migration.Migrate)
 	}
 
+	// At the end, fix up storage-used discrepancies
+	migrations = append(
+		migrations,
+		migrators.NewAccountBasedMigration(
+			log,
+			opts.NWorker,
+			[]migrators.AccountBasedMigration{
+				&migrators.AccountUsageMigrator{},
+			},
+		),
+	)
+
 	log.Info().Msgf("initialized migrations")
 
 	return migrations