Merge pull request #141 from TidierOrg/temptable

add temp table arg

Author: drizk1

NEWS.md

@@ -1,4 +1,8 @@
 # TidierDB.jl updates
+## v.8.7 - 2025-07-07
+- AWS Athena backend bug fixes 
+- add `temp` option to `@create_table`, default is `true`
+
 ## v.8.6 - 2025-05-05
 - add `@pivot_longer`
 - resolve issue when calling `@summarize` consecutively 

Project.toml

@@ -1,7 +1,7 @@
 name = "TidierDB"
 uuid = "86993f9b-bbba-4084-97c5-ee15961ad48b"
 authors = ["Daniel Rizk <rizk.daniel.12@gmail.com> and contributors"]
-version = "0.8.6"
+version = "0.8.7"
 
 [deps]
 Arrow = "69666777-d1a9-59fb-9406-91d4454c9d45"

src/TidierDB.jl

@@ -23,7 +23,7 @@ using GZip
  export db_table, set_sql_mode, connect, from_query, update_con,  
  clickhouse, duckdb, sqlite, mysql, mssql, postgres, athena, snowflake, gbq, 
  oracle, databricks, SQLQuery, show_tables, 
- t, @create_view, drop_view, @compute, warnings, copy_to, dt,
+ t, @create_view, drop_view, @create_table, warnings, copy_to, dt,
  @show_query, @collect, @window_order, @window_frame, write_file
 
  abstract type SQLBackend end

src/docstrings.jl

@@ -1768,9 +1768,9 @@ julia> drop_view(db, "viewer");
 ```
 """
 
-const docstring_compute =
+const docstring_create_table =
 """
-    @compute(sql_query, name, replace = false)
+    @compute(sql_query, name, replace = false, temp = true)
 
 Creates a remote table on database memory from a SQL query. Currently supports DuckDB, MySQL, GBQ, Postgres
 
@@ -1779,14 +1779,15 @@ Creates a remote table on database memory from a SQL query. Currently supports D
 - `sql_query`: The SQL query
 - `name`: The name of the table to create.
 - `replace`: defaults to false if table should be replaced if it already exists.
+- `temp`: defaults to true if table should be created as a temporary table.
 
 # Examples
 ```jldoctest
 julia> db = connect(duckdb());
 
 julia> df = DataFrame(id = [1, 2, 3], value = [10, 20, 30]);
 
-julia> @chain dt(db, df, "df1") @compute(table2, true);
+julia> @chain dt(db, df, "df1") @create_table(table2, replace = true, temp = true);
 
 julia> dt(db, "table2")
 SQLQuery("", "table", "", "", "", "", "", "", false, false, 2×4 DataFrame

src/view_compute.jl

@@ -42,32 +42,54 @@ function drop_view(db, name)
 end
 
 
-macro compute(sqlquery, name, replace = false)
-    
-    if replace == true 
-        sql_cr_or_replace = "CREATE OR REPLACE Table $name AS "
-    elseif replace == false
-         sql_cr_or_replace = "CREATE Table $name AS "
+macro create_table(sqlquery, name, args...)
+    replace_flag = false
+    temp_flag    = false          # new flag
+
+    for arg in args
+        if arg === true || arg === false          # legacy positional arg
+            replace_flag = arg
+        elseif isa(arg, Expr) && arg.head === :(=)
+            lhs, rhs = arg.args
+            if lhs === :replace
+                replace_flag = rhs
+            elseif lhs === :temp
+                temp_flag = rhs
+            else
+                error("@create_table: unknown keyword $(lhs)")
+            end
+        else
+            error("@create_table: unsupported argument $(arg)")
+        end
     end
-    return quote
-       # prin
+
+    replace_clause = replace_flag ? " OR REPLACE" : ""
+    temp_clause    = temp_flag    ? " TEMP"       : ""
+    sql_prefix     = "CREATE$(replace_clause)$(temp_clause) TABLE $name AS "
+
+    quote
         backend = current_sql_mode[]
         sq = $(esc(sqlquery))
+
         if backend == duckdb()
             final_query = finalize_query(sq)
-            final_query = $sql_cr_or_replace  * final_query
+            final_query = $sql_prefix * final_query
+            #println(final_query)
             DBInterface.execute(sq.db, final_query)
-        elseif  backend == postgres()
-            final_compute($(esc(sqlquery)), postgres, $sql_cr_or_replace)
+
+        elseif backend == postgres()
+            final_compute($(esc(sqlquery)), postgres, $sql_prefix)
+
         elseif backend == gbq()
-            final_compute(sq, gbq, $sql_cr_or_replace)
-        elseif current_sql_mode[] == mysql()
-            final_compute($(esc(sqlquery)), mysql, $sql_cr_or_replace)
-         else
-             backend = current_sql_mode[]
-             print("$backend not yet supported") # COV_EXCL_LINE
+            final_compute(sq, gbq, $sql_prefix)
+
+        elseif backend == mysql()
+            final_compute($(esc(sqlquery)), mysql, $sql_prefix)
+
+        else
+            backend = current_sql_mode[]
+            print("$(backend) not yet supported")  # COV_EXCL_LINE
         end
-        
     end
 end