diff --git a/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/nocode/CleanupStep.java b/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/nocode/CleanupStep.java deleted file mode 100644 index a8a7f40e72..0000000000 --- a/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/nocode/CleanupStep.java +++ /dev/null @@ -1,34 +0,0 @@ -package com.linkedin.datahub.upgrade.nocode; - -import com.linkedin.datahub.upgrade.UpgradeCleanupStep; -import com.linkedin.datahub.upgrade.UpgradeContext; -import com.linkedin.datahub.upgrade.UpgradeResult; -import io.ebean.EbeanServer; -import java.util.function.BiConsumer; - - -/** - * Removes the Aspect Table on Upgrade Failure. - */ -public class CleanupStep implements UpgradeCleanupStep { - - private final EbeanServer _server; - - public CleanupStep(final EbeanServer server) { - _server = server; - } - - @Override - public String id() { - return "CleanupStep"; - } - - @Override - public BiConsumer executable() { - return (context, result) -> { - if (UpgradeResult.Result.FAILED.equals(result.result())) { - _server.execute(_server.createSqlUpdate("DROP TABLE metadata_aspect_v2")); - } - }; - } -} diff --git a/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/nocode/NoCodeUpgrade.java b/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/nocode/NoCodeUpgrade.java index e719bae8f8..bbf85536b8 100644 --- a/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/nocode/NoCodeUpgrade.java +++ b/datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/nocode/NoCodeUpgrade.java @@ -1,6 +1,5 @@ package com.linkedin.datahub.upgrade.nocode; -import com.google.common.collect.ImmutableList; import com.linkedin.datahub.upgrade.Upgrade; import com.linkedin.datahub.upgrade.UpgradeCleanupStep; import com.linkedin.datahub.upgrade.UpgradeStep; @@ -8,6 +7,7 @@ import com.linkedin.metadata.entity.EntityService; import com.linkedin.metadata.models.registry.SnapshotEntityRegistry; import io.ebean.EbeanServer; import java.util.ArrayList; +import java.util.Collections; import java.util.List; public class NoCodeUpgrade implements Upgrade { @@ -48,7 +48,7 @@ public class NoCodeUpgrade implements Upgrade { } private List buildCleanupSteps(final EbeanServer server) { - return ImmutableList.of(new CleanupStep(server)); + return Collections.emptyList(); } private List buildUpgradeSteps(