Add tableLocation

liquibase · Jul 18, 2024 · 3b12076 · 3b12076
1 parent 25b00d8
commit 3b12076
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();