create shortcut 7970

treeverse · Jul 21, 2024 · d9924c0 · d9924c0
1 parent 16fec21
commit d9924c0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/actions/lua/lakefs/catalogexport/glue_exporter.lua b/pkg/actions/lua/lakefs/catalogexport/glue_exporter.lua
@@ -103,6 +103,7 @@ end
     - table_name(string): override default glue table name
     - debug(boolean)
     - export_base_uri(string): override the default prefix in S3 for symlink location i.e s3://other-bucket/path/
+    - create_nonexistant_db(boolean): if the db doesn't exist, creates it
 ]]
 local function export_glue(glue, db, table_src_path, create_table_input, action_info, options)
     local opts = options or {}
@@ -127,6 +128,11 @@ local function export_glue(glue, db, table_src_path, create_table_input, action_
         error("table " .. descriptor.type .. " in path " .. table_src_path .. " not supported")
     end
 
+    local create_db = opts.create_nonexistant_db or false
+    if create_db then
+        glue.create_database(db) -- may fail if db doesn't exist
+    end
+
     -- finallize create glue table input
     local table_input = build_glue_create_table_input(create_table_input, descriptor, symlink_location, columns,
         partitions, action_info, opts)