Merge pull request #362 from liquibase/DAT-18635

DAT-18635: check of datasets on connect command added.
liquibase · Nov 26, 2024 · 2b64fff · 2b64fff
2 parents 3d8acf2 + 3da6617
commit 2b64fff
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/main/java/liquibase/ext/bigquery/database/BigQueryDatabase.java b/src/main/java/liquibase/ext/bigquery/database/BigQueryDatabase.java
@@ -13,6 +13,8 @@
 import liquibase.structure.DatabaseObject;
 import liquibase.structure.core.*;
 
+import java.sql.ResultSet;
+import java.sql.SQLException;
 import java.util.HashSet;
 import java.util.Set;
 import java.util.regex.Pattern;
@@ -358,4 +360,23 @@ private Set<String> getDefaultReservedWords() {
 
         return reservedWords;
     }
+
+    @Override
+    public void checkDatabaseConnection() throws DatabaseException {
+        BigQueryConnection connection = (BigQueryConnection) getConnection();
+        try {
+            String catalogName = getConnectionCatalogName();
+            String schemaName = getConnectionSchemaName();
+            ResultSet schemasAlikeUsed = connection.getMetaData().getSchemas(catalogName, schemaName);
+            while (schemasAlikeUsed.next()) {
+                if (schemasAlikeUsed.getString(1).equals(schemaName)) {
+                    return;
+                }
+            }
+            throw new DatabaseException(String.format("Please specify existing dataset in connection url. " +
+                    "Current connection points to '%s.%s'", catalogName, schemaName));
+        } catch (SQLException e) {
+            Scope.getCurrentScope().getLog(getClass()).info("Error checking database connection", e);
+        }
+    }
 }