duckdb 1.5.2.0 → 1.5.2.1

data/lib/duckdb/appender.rb

@@ -93,6 +93,80 @@ module DuckDB
       raise_appender_error('failed to close')
     end
 
+    # :call-seq:
+    #   appender.add_column(column_name) -> self
+    #
+    # Specifies a column to append to, allowing selective column insertion.
+    # Columns not added will use their default values or be computed from
+    # generated column expressions.
+    # Raises DuckDB::Error if the column does not exist in the table.
+    #
+    #   require 'duckdb'
+    #   db = DuckDB::Database.open
+    #   con = db.connect
+    #   con.query('CREATE TABLE t (id UUID PRIMARY KEY DEFAULT uuidv4(), name VARCHAR)')
+    #   appender = con.appender('t')
+    #   appender.add_column('name')
+    #   appender
+    #     .append_varchar('Alice')
+    #     .end_row
+    #     .flush
+    def add_column(column)
+      return self if _add_column(column)
+
+      raise_appender_error('failed to add_column')
+    end
+
+    # :call-seq:
+    #   appender.clear_columns -> self
+    #
+    # Clears the list of columns previously set by #add_column, so that all
+    # columns of the table become active again. Any previously appended rows
+    # are flushed before the column list is reset; if the flush fails (e.g.
+    # a constraint violation), this method raises DuckDB::Error.
+    #
+    #   require 'duckdb'
+    #   db = DuckDB::Database.open
+    #   con = db.connect
+    #   con.query('CREATE TABLE t (id UUID PRIMARY KEY DEFAULT uuidv4(), name VARCHAR)')
+    #   appender = con.appender('t')
+    #   appender.add_column('name')
+    #   appender
+    #     .append_varchar('Alice')
+    #     .end_row
+    #     .flush
+    #   appender.clear_columns
+    #   # all table columns are active again
+    def clear_columns
+      return self if _clear_columns
+
+      raise_appender_error('failed to clear_columns')
+    end
+
+    if DuckDB::Appender.private_method_defined?(:_clear)
+      # :call-seq:
+      #   appender.clear -> self
+      #
+      # Clears all unflushed data from the appender, discarding any appended rows
+      # that have not yet been flushed to the table.
+      #
+      #   require 'duckdb'
+      #   db = DuckDB::Database.open
+      #   con = db.connect
+      #   con.query('CREATE TABLE users (id INTEGER, name VARCHAR)')
+      #   appender = con.appender('users')
+      #   appender
+      #     .append_int32(1)
+      #     .append_varchar('Alice')
+      #     .end_row
+      #     .clear # discards the row above without flushing to the table
+      def clear
+        return self if _clear
+
+        raise_appender_error('failed to clear')
+      end
+    end
+
     # call-seq:
     #   appender.append_bool(val) -> self
     #

data/lib/duckdb/connection.rb

@@ -34,7 +34,7 @@ module DuckDB
 
     def query_multi_sql(sql)
       stmts = ExtractedStatements.new(self, sql)
-      return query_sql(sql) if stmts.size == 1
+      return _query_sql(sql) if stmts.size == 1
 
       result = nil
       stmts.each do |stmt|
@@ -276,7 +276,6 @@ module DuckDB
     def register_table_function(table_function)
       raise ArgumentError, 'table_function must be a TableFunction' unless table_function.is_a?(TableFunction)
 
-      check_threads
       _register_table_function(table_function)
     end
 
@@ -291,12 +290,10 @@ module DuckDB
     # @param columns [Hash{String => DuckDB::LogicalType}, nil] optional column schema override;
     #   if omitted, the adapter determines the columns (e.g. from headers or inference)
     # @raise [ArgumentError] if no adapter is registered for the object's class
-    # @raise [DuckDB::Error] if threads setting is not 1
     # @return [void]
     #
     # @example Expose a CSV as a table
     #   require 'csv'
-    #   con.execute('SET threads=1')
     #   DuckDB::TableFunction.add_table_adapter(CSV, CSVTableAdapter.new)
     #   csv = CSV.new(File.read('data.csv'), headers: true)
     #   con.expose_as_table(csv, 'csv_table')
@@ -318,18 +315,6 @@ module DuckDB
 
     private
 
-    def check_threads
-      result = execute("SELECT current_setting('threads')")
-      thread_count = result.first.first.to_i
-
-      return unless thread_count > 1
-
-      raise DuckDB::Error,
-            'Functions with Ruby callbacks require single-threaded execution. ' \
-            "Current threads setting: #{thread_count}. " \
-            "Execute 'SET threads=1' before registering functions."
-    end
-
     def run_appender_block(appender, &)
       return appender unless block_given?
 

data/lib/duckdb/converter.rb

@@ -18,6 +18,7 @@ module DuckDB
     RANGE_UINT64 = 0..18_446_744_073_709_551_615
     RANGE_HUGEINT = (-(1 << 127)..((1 << 127) - 1))
     RANGE_UHUGEINT = (0..((1 << 128) - 1))
+    RANGE_DECIMAL_WIDTH = 1..38
 
     HALF_HUGEINT_BIT = 64
     HALF_HUGEINT = 1 << HALF_HUGEINT_BIT
@@ -122,6 +123,10 @@ module DuckDB
       value >> HALF_HUGEINT_BIT
     end
 
+    def _decimal_width(value)
+      value.to_s('F').gsub(/[^0-9]/, '').length
+    end
+
     def _to_decimal_from_hugeint(width, scale, upper, lower = nil)
       v = lower.nil? ? upper : _to_hugeint_from_vector(lower, upper)
       _to_decimal_from_value(width, scale, v)

data/lib/duckdb/logical_type.rb

@@ -2,8 +2,6 @@
 
 module DuckDB
   class LogicalType # rubocop:disable Metrics/ClassLength
-    RANGE_DECIMAL_WIDTH = 1..38
-
     alias :alias get_alias
     alias :alias= set_alias
 
@@ -171,7 +169,7 @@ module DuckDB
       #   decimal_type.width #=> 18
       #   decimal_type.scale #=> 3
       def create_decimal(width, scale)
-        raise DuckDB::Error, 'width must be between 1 and 38' unless RANGE_DECIMAL_WIDTH.cover?(width)
+        raise DuckDB::Error, 'width must be between 1 and 38' unless Converter::RANGE_DECIMAL_WIDTH.cover?(width)
         raise DuckDB::Error, "scale must be between 0 and width(#{width})" unless (0..width).cover?(scale)
 
         _create_decimal_type(width, scale)

data/lib/duckdb/prepared_statement.rb

@@ -281,7 +281,7 @@ module DuckDB
     def bind_decimal(index, value)
       decimal = _parse_deciaml(value)
       lower, upper = decimal_to_hugeint(decimal)
-      width = decimal.to_s('F').gsub(/[^0-9]/, '').length
+      width = _decimal_width(decimal)
       _bind_decimal(index, lower, upper, width, decimal.scale)
     end
 

data/lib/duckdb/table_function.rb

@@ -163,7 +163,6 @@ module DuckDB
       #
       #   # Register and use:
       #   DuckDB::TableFunction.add_table_adapter(CSV, CSVTableAdapter.new)
-      #   con.execute('SET threads=1')
       #   con.expose_as_table(csv, 'csv_table')
       #   con.query('SELECT * FROM csv_table()').to_a
       #

data/lib/duckdb/value.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'bigdecimal'
+
 module DuckDB
   class Value
     class << self
@@ -219,6 +221,23 @@ module DuckDB
         _create_uhugeint(lower, upper)
       end
 
+      # Creates a DuckDB::Value of DECIMAL type.
+      #
+      #   value = DuckDB::Value.create_decimal(BigDecimal('12345.678'))
+      #
+      # @param value [BigDecimal] the decimal value.
+      # @return [DuckDB::Value] the created Value object.
+      # @raise [ArgumentError] if +value+ is not a BigDecimal or its width is out of range (1..38).
+      def create_decimal(value)
+        check_type!(value, BigDecimal)
+
+        width = _decimal_width(value)
+        check_range!(width, RANGE_DECIMAL_WIDTH, 'DECIMAL width')
+
+        lower, upper = decimal_to_hugeint(value)
+        _create_decimal(lower, upper, width, value.scale)
+      end
+
       private
 
       def check_range!(value, range, type_name)

data/lib/duckdb/version.rb

@@ -3,5 +3,5 @@
 module DuckDB
   # The version string of ruby-duckdb.
   # Currently, ruby-duckdb is NOT semantic versioning.
-  VERSION = '1.5.2.0'
+  VERSION = '1.5.2.1'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: duckdb
 version: !ruby/object:Gem::Version
-  version: 1.5.2.0
+  version: 1.5.2.1
 platform: ruby
 authors:
 - Masaki Suketa
@@ -40,6 +40,7 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
+- duckdb.gemspec
 - ext/duckdb/aggregate_function.c
 - ext/duckdb/aggregate_function.h
 - ext/duckdb/appender.c
@@ -120,7 +121,6 @@ files:
 - lib/duckdb/converter/int_to_sym.rb
 - lib/duckdb/data_chunk.rb
 - lib/duckdb/database.rb
-- lib/duckdb/duckdb_native.so
 - lib/duckdb/expression.rb
 - lib/duckdb/extracted_statements.rb
 - lib/duckdb/function_type_validation.rb