diff --git a/src/main/java/liquibase/ext/redshift/snapshot/ForeignKeySnapshotGenerator.java b/src/main/java/liquibase/ext/redshift/snapshot/ForeignKeySnapshotGenerator.java index 839e670..969b19b 100644 --- a/src/main/java/liquibase/ext/redshift/snapshot/ForeignKeySnapshotGenerator.java +++ b/src/main/java/liquibase/ext/redshift/snapshot/ForeignKeySnapshotGenerator.java @@ -6,11 +6,15 @@ import liquibase.snapshot.InvalidExampleException; import liquibase.snapshot.SnapshotGenerator; import liquibase.structure.DatabaseObject; +import liquibase.ext.redshift.database.RedshiftDatabase; public class ForeignKeySnapshotGenerator extends liquibase.snapshot.jvm.ForeignKeySnapshotGenerator { @Override public int getPriority(Class objectType, Database database) { - return PRIORITY_DATABASE; + if (database instanceof RedshiftDatabase) { + return PRIORITY_DATABASE; + } + return PRIORITY_NONE; } @Override