Merge pull request #154 from liquibase/DAT-17989

[DAT-17989] Add tableLocation functionality to createTable
liquibase · Aug 21, 2024 · c644c30 · c644c30
2 parents d2ef6f0 + 2887167
commit c644c30
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/java/liquibase/ext/databricks/sqlgenerator/CreateTableGeneratorDatabricks.java b/src/main/java/liquibase/ext/databricks/sqlgenerator/CreateTableGeneratorDatabricks.java
@@ -57,6 +57,8 @@ public Sql[] generateSql(CreateTableStatement statement, Database database, SqlG
             // Databricks can decide to have tables live in a particular location. If null, Databricks will handle the location automatically in DBFS
             if (!StringUtils.isEmpty(thisStatement.getTableLocation())) {
                 finalsql += " LOCATION '" + thisStatement.getTableLocation() + "'";
+            } else if (thisStatement.getExtendedTableProperties() != null && StringUtils.isNotEmpty(thisStatement.getExtendedTableProperties().getTableLocation())) {
+                finalsql += " LOCATION '" + thisStatement.getExtendedTableProperties().getTableLocation() + "'";
             }
 
             ArrayList<String> clusterCols = thisStatement.getClusterColumns();