handle missing staged contracts

onflow · Apr 26, 2024 · e46d9a2 · e46d9a2
1 parent e5febd5
commit e46d9a2
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/cmd/util/ledger/migrations/staged_contracts_migration.go b/cmd/util/ledger/migrations/staged_contracts_migration.go
@@ -239,7 +239,17 @@ func (m *StagedContractsMigration) collectAndRegisterStagedContractsFromPayloads
 	inter := mr.Interpreter
 	locationRange := interpreter.EmptyLocationRange
 
-	storageMap := mr.Storage.GetStorageMap(stagingAccountAddress, common.PathDomainStorage.Identifier(), false)
+	storageMap := mr.Storage.GetStorageMap(
+		stagingAccountAddress,
+		common.PathDomainStorage.Identifier(),
+		false,
+	)
+	if storageMap == nil {
+		m.log.Error().
+			Msgf("failed to get staged contracts from account %s", stagingAccount)
+		return nil
+	}
+
 	iterator := storageMap.Iterator(inter)
 
 	stagedContractCapsuleStaticType := interpreter.NewCompositeStaticTypeComputeTypeID(