data_drain 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0e55b7e816c9f186bb8bbef3bcc63c49bf0574df3a23eec6272f923f39e6eab6
-  data.tar.gz: 93cb8b665c2b2fa39cb279dd4f5ee5bb381a6bf837b53cb75a28e4db04281ff0
+  metadata.gz: 05fca5dc293bebcb71247b8659d52933d8a5baca6b73e38ba9d9b21236c02643
+  data.tar.gz: c7d89c3fb45dc4deb3b871bd403a2e1b39172977893882df2658deb8eafcb22f
 SHA512:
-  metadata.gz: 8760d19a38e5f41ddbed8555e7a9454ff938cd8a00113afa6e14dace79b910683b266cfcd49871a249c78f3e24b8a283ce1cb0f92e831bf7613640e20c8139b5
-  data.tar.gz: ccc40f5aab2f6ec9f369e0b59049c7f78d8d587e860073403c5b643334d577131aa481c6118e8fe1180d4200feb5644cfad361f3f1dc6f11daed7a8a2d760d90
+  metadata.gz: fce9cc7ecc7008136c456ed3442bdd4b6c8171e0b28c37a1ba6bc897866b3e4469dfd7af8285a9b1022c70f00cf8e14a0e16a8fd280038f2a0b604418471be4b
+  data.tar.gz: 766e0ea20985a90e39fd84727318b6b1a753bad695af44f78bbf94da088f3aef06c1b428811af27fdb46466c9c694a70974ef113f9ee9bdb90b3a616fe353f9c

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 ## [Unreleased]
 
+## [0.1.5] - 2026-03-16
+
+- Se agrego el attacha para duckdb.
+
+## [0.1.4] - 2026-03-16
+
+- Corrección de error por comilla simple
+
 ## [0.1.3] - 2026-03-16
 
 - Corrección de la sintaxis para postres_query

data/lib/data_drain/engine.rb

@@ -81,6 +81,7 @@ module DataDrain
     def setup_duckdb
       @duckdb.query("INSTALL postgres; LOAD postgres;")
       @duckdb.query("SET max_memory='#{@config.limit_ram}';") if @config.limit_ram.present?
+      @duckdb.query("ATTACH '#{@config.duckdb_connection_string}' AS pg_source (TYPE POSTGRES, READ_ONLY)")
 
       # 💡 Magia del Adapter: Él sabe si cargar httpfs y setear credenciales o no hacer nada
       @adapter.setup_duckdb(@duckdb)
@@ -90,7 +91,8 @@ module DataDrain
     # @return [Integer]
     def get_postgres_count
       pg_sql = "SELECT COUNT(*) AS row_count FROM public.#{@table_name} WHERE #{base_where_sql}"
-      query = "SELECT row_count FROM postgres_query('#{@config.duckdb_connection_string}', '#{pg_sql}')"
+      pg_sql = pg_sql.gsub("'", "''")
+      query = "SELECT row_count FROM postgres_query('pg_source', '#{pg_sql}')"
       @duckdb.query(query).first.first
     end
 
@@ -103,11 +105,12 @@ module DataDrain
       dest_path = @config.storage_mode.to_sym == :s3 ? "s3://#{@bucket}/#{@folder_name}/" : File.join(@bucket, @folder_name, "")
 
       pg_sql = "SELECT #{@select_sql} FROM public.#{@table_name} WHERE #{base_where_sql}"
+      pg_sql = pg_sql.gsub("'", "''")
 
       query = <<~SQL
         COPY (
           SELECT #{@select_sql}
-          FROM postgres_query('#{@config.duckdb_connection_string}', '#{pg_sql}')
+          FROM postgres_query('pg_source', '#{pg_sql}')
         ) TO '#{dest_path}'
         (
           FORMAT PARQUET,

data/lib/data_drain/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DataDrain
-  VERSION = "0.1.3"
+  VERSION = "0.1.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: data_drain
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Gabriel