raka 0.4.1 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6a91b1fab962b4bd99bd333aacc88b52267750458e9430584668f13389aec117
-  data.tar.gz: a58613eddc635b0440e6cb0e061311abc137e7b893dd87b5c66f8f97c6f952c2
+  metadata.gz: f8d1e99a862ad8d34b330ee50d92afdbf3c6aa11c8111621213e5794dd18830c
+  data.tar.gz: ac8a414a0bb4e0830aea306d80901370efdf77a4a9a9ddc159d79869f0a95410
 SHA512:
-  metadata.gz: cf28cfc8c9e3c3be6f4a1d12edb525f8353dd225efd64e7e6d5904ed068173f036af496bcf13a3d5febb731249e3e57c1bee87eea1994a00b34334efe8a5a5c2
-  data.tar.gz: bdd4d58a420a46cdb4fd23a714ea60c14b11c7a0fe8b5bb6cfc343f860208e60ae163d28abab6839cf7abb2548c4c7a849d513026d9c4e091167e2d102f99161
+  metadata.gz: a9163e4fb93266b5ba646fd21f88c75d85a321e20aa0cb6e55ccb331aa41f4722d3695b8af126c77f3857ffd7240966a55e7b4c16c908bd5c8471b30b9da7265
+  data.tar.gz: 4221dc18f6a1b03799d519df61c80828fd9f4b25353a22eee076f97959a25865224145f7403d5f05988394170b0d69be15720b9bcb6df1b7c3039b51b4d19d19

data/VERSION

@@ -1 +1 @@
-0.4.1
+0.5.0

data/lib/raka/lang/duckdb/impl.rb

@@ -17,10 +17,12 @@ end
 # 1. Persistent mode: operations on .db file with CREATE TABLE
 # 2. Ad-hoc mode: parquet in/out using COPY operations
 class Duckdb
-  def initialize(database: nil, params: {})
+  def initialize(database: nil, params: {}, before: nil, after: nil)
     @params = params
     @database = database
     @mode = @database ? :persistent : :adhoc
+    @before = before
+    @after = after
   end
 
   def duckdb_cmd
@@ -32,26 +34,50 @@ class Duckdb
     end
   end
 
-  def build(code, _task)
-    # Replace parameter placeholders
+  def process_params(code)
+    return code if code.nil?
+    
     processed_code = code
     (@params || {}).each do |key, value|
       processed_code = processed_code.gsub("$#{key}", "'#{value}'")
     end
+    processed_code
+  end
+
+  def build(code, _task)
+    # Process parameter placeholders for all parts
+    main_sql = process_params(code)
+    before_sql = process_params(@before)
+    after_sql = process_params(@after)
 
+    # Build SQL parts as separate statements
+    sql_parts = []
+    
+    # Add before hook if present
+    sql_parts << before_sql if before_sql
+    
+    # Add main query based on mode
     case @mode
     when :persistent
-      "DROP TABLE IF EXISTS :_name_; CREATE TABLE :_name_ AS (#{processed_code});"
+      sql_parts << "DROP TABLE IF EXISTS :_name_;"
+      sql_parts << "CREATE TABLE :_name_ AS (#{main_sql});"
     when :adhoc
-      "COPY (#{processed_code}) TO ':output:' (FORMAT PARQUET);"
+      sql_parts << "COPY (#{main_sql}) TO ':output:' (FORMAT PARQUET);"
     end
+    
+    # Add after hook if present
+    sql_parts << after_sql if after_sql
+    
+    sql_parts.join("\n")
   end
 
   def run_script(env, fname, task)
     case @mode
     when :persistent
+      # Split the SQL into separate statements and execute them individually
       bash env, %(
-      #{duckdb_cmd} -c "$(cat #{fname} | sed 's|:_name_|#{task.output_stem}|g')" | tee #{fname}.log
+      # Execute the combined SQL script with proper variable replacement
+      cat #{fname} | sed 's|:_name_|#{task.output_stem}|g' | #{duckdb_cmd} | tee #{fname}.log
       echo "#{@database}" > #{task.name}
       )
     when :adhoc

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: raka
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
 platform: ruby
 authors:
 - yarray