fix(upgrade): fix upgrade npe (datahub-project#10436)

sleeperdeep · Jun 25, 2024 · 8073ad7 · 8073ad7
1 parent 4c75ac2
commit 8073ad7
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/system/AbstractMCLStep.java b/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/system/AbstractMCLStep.java
@@ -13,7 +13,6 @@
 import com.linkedin.metadata.entity.EntityService;
 import com.linkedin.metadata.entity.EntityUtils;
 import com.linkedin.metadata.entity.restoreindices.RestoreIndicesArgs;
-import com.linkedin.metadata.models.AspectSpec;
 import com.linkedin.metadata.utils.AuditStampUtils;
 import com.linkedin.util.Pair;
 import io.datahubproject.metadata.context.OperationContext;
@@ -77,9 +76,6 @@ public Function<UpgradeContext, UpgradeStepResult> executable() {
         args = args.urnLike(getUrnLike());
       }
 
-      final AspectSpec aspectSpec =
-          opContext.getEntityRegistry().getAspectSpecs().get(getAspectName());
-
       aspectDao
           .streamAspectBatches(args)
           .forEach(
@@ -98,7 +94,7 @@ public Function<UpgradeContext, UpgradeStepResult> executable() {
                                     systemAspect.getUrn(),
                                     systemAspect.getUrn().getEntityType(),
                                     getAspectName(),
-                                    aspectSpec,
+                                    systemAspect.getAspectSpec(),
                                     null,
                                     systemAspect.getRecordTemplate(),
                                     null,