raka 0.8.3 → 0.9.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/VERSION +1 -1
- data/lib/raka/lang/duckdb/impl.rb +11 -3
- metadata +1 -1
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: 8b04ff22d6ec2fd1a4c986b4a7969596ea9fbe155669558eb69dd469b947aebc
|
|
4
|
+
data.tar.gz: 680f48420d9d1f3bb0e758b21df43b97f7de9af489385b19fa465cc76bb3f816
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: 41de921265fc6b0cd178405054dfed5b8c65acb99819f9fe1dfd1808d1a45e08dc19ef871e01566047f4014d1d0cae1235a9c2f0b71e8d67646b7a69d8e58826
|
|
7
|
+
data.tar.gz: f5c274b5a85941fd347822ce67e9aff14969c5e9816a7472c77005bc38c655e9ccb7e4d865319646e04b140fa14c3b2d72117d668066ef727084c2818a5c94f4
|
data/VERSION
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
0.
|
|
1
|
+
0.9.0
|
|
@@ -17,12 +17,13 @@ end
|
|
|
17
17
|
# 1. Persistent mode: operations on .db file with CREATE TABLE
|
|
18
18
|
# 2. Ad-hoc mode: parquet in/out using COPY operations
|
|
19
19
|
class Duckdb
|
|
20
|
-
def initialize(database: nil, params: {}, before: nil, after: nil)
|
|
20
|
+
def initialize(database: nil, params: {}, before: nil, after: nil, format: nil)
|
|
21
21
|
@params = params
|
|
22
22
|
@database = database
|
|
23
23
|
@mode = @database ? :persistent : :adhoc
|
|
24
24
|
@before = before
|
|
25
25
|
@after = after
|
|
26
|
+
@format = format&.upcase
|
|
26
27
|
end
|
|
27
28
|
|
|
28
29
|
def duckdb_cmd
|
|
@@ -44,7 +45,12 @@ class Duckdb
|
|
|
44
45
|
processed_code
|
|
45
46
|
end
|
|
46
47
|
|
|
47
|
-
def
|
|
48
|
+
def detect_format_from_extension(filename)
|
|
49
|
+
ext = File.extname(filename)[1..]&.upcase # Remove dot and convert to uppercase
|
|
50
|
+
ext || 'PARQUET' # Default fallback
|
|
51
|
+
end
|
|
52
|
+
|
|
53
|
+
def build(code, task)
|
|
48
54
|
# Process parameter placeholders for all parts
|
|
49
55
|
main_sql = process_params(code)
|
|
50
56
|
before_sql = process_params(@before)
|
|
@@ -62,7 +68,9 @@ class Duckdb
|
|
|
62
68
|
sql_parts << 'DROP TABLE IF EXISTS :_name_;'
|
|
63
69
|
sql_parts << "CREATE TABLE :_name_ AS (#{main_sql});"
|
|
64
70
|
when :adhoc
|
|
65
|
-
|
|
71
|
+
# Determine format: use explicit format if provided, otherwise detect from output filename
|
|
72
|
+
format = @format || detect_format_from_extension(task.name)
|
|
73
|
+
sql_parts << "COPY (#{main_sql}) TO ':output:' (FORMAT #{format});"
|
|
66
74
|
end
|
|
67
75
|
|
|
68
76
|
# Add after hook if present
|