Check schema when creating tables in BigQuery

trinodb · Apr 6, 2022 · f1a05ee · f1a05ee
1 parent e68bf33
commit f1a05ee
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/BigQueryMetadata.java b/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/BigQueryMetadata.java
@@ -369,6 +369,11 @@ private void createTable(ConnectorSession session, ConnectorTableMetadata tableM
         SchemaTableName schemaTableName = tableMetadata.getTable();
         String schemaName = schemaTableName.getSchemaName();
         String tableName = schemaTableName.getTableName();
+
+        if (!schemaExists(session, schemaName)) {
+            throw new SchemaNotFoundException(schemaName);
+        }
+
         List<Field> fields = tableMetadata.getColumns().stream()
                 .map(column -> toField(column.getName(), column.getType()))
                 .collect(toImmutableList());