fix: Snapshot generators should only apply when executing in a BigQue…

…ry database (#254) * fix: Snapshot generators for bigquery should only apply when we execute in a BigQuery database.
liquibase · Feb 27, 2024 · 02640e2 · 02640e2
1 parent d946498
commit 02640e2
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 1 deletion.
diff --git a/src/main/java/liquibase/ext/bigquery/snapshot/jvm/BigQueryDatasetSnapshotGenerator.java b/src/main/java/liquibase/ext/bigquery/snapshot/jvm/BigQueryDatasetSnapshotGenerator.java
@@ -24,6 +24,9 @@ public class BigQueryDatasetSnapshotGenerator extends SchemaSnapshotGenerator {
 
     @Override
     public int getPriority(Class<? extends DatabaseObject> objectType, Database database) {
+        if (!(database instanceof BigqueryDatabase)) {
+            return PRIORITY_NONE;
+        }
         int priority = super.getPriority(objectType, database);
         if (priority > PRIORITY_NONE && database instanceof BigqueryDatabase) {
             priority += PRIORITY_DATABASE;
@@ -123,4 +126,4 @@ protected DatabaseObject snapshotObject(DatabaseObject example, DatabaseSnapshot
         return match;
     }
 
-}
+}
diff --git a/src/main/java/liquibase/ext/bigquery/snapshot/jvm/BigQuerySequenceSnapshotGenerator.java b/src/main/java/liquibase/ext/bigquery/snapshot/jvm/BigQuerySequenceSnapshotGenerator.java
@@ -18,6 +18,9 @@ public class BigQuerySequenceSnapshotGenerator extends SequenceSnapshotGenerator
 
         @Override
         public int getPriority(Class<? extends DatabaseObject> objectType, Database database) {
+            if (!(database instanceof BigqueryDatabase)) {
+                return PRIORITY_NONE;
+            }
             int priority = super.getPriority(objectType, database);
             if (priority > PRIORITY_NONE && database instanceof BigqueryDatabase) {
                 priority += PRIORITY_DATABASE;

diff --git a/...n/java/liquibase/ext/bigquery/snapshot/jvm/BigQueryUniqueConstraintSnapshotGenerator.java b/...n/java/liquibase/ext/bigquery/snapshot/jvm/BigQueryUniqueConstraintSnapshotGenerator.java
@@ -22,6 +22,9 @@ public class BigQueryUniqueConstraintSnapshotGenerator extends UniqueConstraintS
 
     @Override
     public int getPriority(Class<? extends DatabaseObject> objectType, Database database) {
+        if (!(database instanceof BigqueryDatabase)) {
+            return PRIORITY_NONE;
+        }
         int priority = super.getPriority(objectType, database);
         if (priority > PRIORITY_NONE && database instanceof BigqueryDatabase) {
             priority += PRIORITY_DATABASE;

diff --git a/src/main/java/liquibase/ext/bigquery/snapshot/jvm/BigQueryViewSnapshotGenerator.java b/src/main/java/liquibase/ext/bigquery/snapshot/jvm/BigQueryViewSnapshotGenerator.java
@@ -23,13 +23,18 @@ public class BigQueryViewSnapshotGenerator extends ViewSnapshotGenerator {
 
     @Override
     public int getPriority(Class<? extends DatabaseObject> objectType, Database database) {
+        if (!(database instanceof BigqueryDatabase)) {
+            return PRIORITY_NONE;
+        }
         int priority = super.getPriority(objectType, database);
         if (priority > PRIORITY_NONE && database instanceof BigqueryDatabase) {
             priority += PRIORITY_DATABASE;
         }
         return priority;
     }
 
+
+
     @Override
     protected DatabaseObject snapshotObject(DatabaseObject example, DatabaseSnapshot snapshot) throws DatabaseException {
         if (((View) example).getDefinition() != null) {