duckdb 1.5.0.1 → 1.5.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 249a190d19c8c4b8ff32f06621a8fd2cba6690008a593e33603761810e3f2263
-  data.tar.gz: f0dd19a5a1b197cf22731b40741771e36b04a59b28388fb68fde6af76233d68e
+  metadata.gz: 35eae50625fd72c7cb1b870ff496d259c0271453404a0646fd05514b4a3f7338
+  data.tar.gz: e96e82fcbffa7427f63cf8a7af0ba9434b5cb7daefbd9951c3eba4866508d67e
 SHA512:
-  metadata.gz: 381f2d2ac0b028e0a8fb2b8ab0f18f22f876f81f868d7921cb6952eaa5059a35f6d87e4bdf39b823de25fa5e0b59f349efdc77507e361aeaac032bf402211b3a
-  data.tar.gz: 8472e39a5f8e1136193f718ad436989fba3ddffe1c5072adc775a59490c08f5a141d2a2826a314413a5a5472aea2c92af467f479e7f2f013fb06bc9630bfd3d0
+  metadata.gz: c7fdc287942e832ee9f33fbbf0056add1a7b7fc8db0e633356b7cf73c3d086f1d7a89ac7cf79673d0ecbbecda15ee33d365182c2874a74ac23aaf95270fe90dd
+  data.tar.gz: 7cc3079783a36442e43b7228223e2ded5cb453184e1eb4344788f5b7c1758e2ba665d8a9d087f6239d84dfb16a389f18fe091bd346a439e7e3c32ed9a333a9ce

data/.github/workflows/test_on_macos.yml

@@ -20,7 +20,7 @@ jobs:
     timeout-minutes: 120
     strategy:
       matrix:
-        ruby: ['3.2.9', '3.3.10', '3.4.9', '4.0.1', 'head']
+        ruby: ['3.2.9', '3.3.10', '3.4.9', '4.0.2', 'head']
         duckdb: ['1.4.4', '1.5.0']
 
     steps:

data/.github/workflows/test_on_ubuntu.yml

@@ -20,7 +20,7 @@ jobs:
     timeout-minutes: 120
     strategy:
       matrix:
-        ruby: ['3.2.9', '3.3.10', '3.4.9', '4.0.1', 'head']
+        ruby: ['3.2.9', '3.3.10', '3.4.9', '4.0.2', 'head']
         duckdb: ['1.4.4', '1.5.0']
 
     services:

data/.github/workflows/test_on_windows.yml

@@ -17,7 +17,7 @@ concurrency:
 jobs:
   test:
     runs-on: windows-latest
-    timeout-minutes: 10
+    timeout-minutes: 15
     strategy:
       matrix:
         ruby: ['3.2.9', '3.3.10', '3.4.8', '4.0.1', 'ucrt', 'mingw', 'mswin', 'head']
@@ -51,8 +51,11 @@ jobs:
         bundle exec rake build -- --with-duckdb-include=../../../.. --with-duckdb-lib=../../../..
 
     - name: rake test
-      run: |
-        bundle exec rake test TESTOPTS="--verbose"
+      uses: nick-fields/retry@v3
+      with:
+        timeout_minutes: 3
+        max_attempts: 3
+        command: bundle exec rake test TESTOPTS="--verbose"
 
   post-test:
     name: All tests passed on Windows

data/CHANGELOG.md

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 
 # Unreleased
 
+# 1.5.0.2 - 2026-03-22
+
+- add `DuckDB.vector_size` to return the DuckDB vector size (number of rows processed per vectorized operation).
+- `DuckDB::ScalarFunction.create`, `DuckDB::ScalarFunction#add_parameter`, `DuckDB::ScalarFunction#return_type=` accept logical type symbol as argument.
+- add `DuckDB.default_timezone` configuration (defaults to `:local`) to control how TIMESTAMP and TIME values without time zone are converted to Ruby `Time` (`:utc` or `:local`).
+
 # 1.5.0.1 - 2026-03-17
 
 - enable `DuckDB::ScalarFunction` work with duckdb multi threads.

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    duckdb (1.5.0.1)
+    duckdb (1.5.0.2)
       bigdecimal (>= 3.1.4)
 
 GEM
@@ -13,13 +13,13 @@ GEM
     bigdecimal (4.0.1)
     csv (3.3.5)
     drb (2.2.3)
-    json (2.19.1)
+    json (2.19.2)
     json-schema (6.2.0)
       addressable (~> 2.8)
       bigdecimal (>= 3.1, < 5)
     language_server-protocol (3.17.0.5)
     lint_roller (1.1.0)
-    mcp (0.8.0)
+    mcp (0.9.0)
       json-schema (>= 4.1)
     minitest (6.0.2)
       drb (~> 2.0)

data/README.md

@@ -263,3 +263,7 @@ con.query('INSERT INTO numbers VALUES (2), (1), (4), (3)')
 res = con.query('SELECT number FROM numbers ORDER BY number')
 res.first.first # => 4
 ```
+
+#### Global configuration
+
+Set `DuckDB.default_timezone` to control how TIMESTAMP and TIME values without time zone are converted to Ruby `Time` objects. The default is `:local`, but you can use `:utc` for UTC conversion.

data/ext/duckdb/duckdb.c

@@ -5,6 +5,7 @@ VALUE PositiveInfinity;
 VALUE NegativeInfinity;
 
 static VALUE duckdb_s_library_version(VALUE self);
+static VALUE duckdb_s_vector_size(VALUE self);
 
 /*
  * call-seq:
@@ -18,6 +19,19 @@ static VALUE duckdb_s_library_version(VALUE self) {
     return rb_str_new2(duckdb_library_version());
 }
 
+/*
+ * call-seq:
+ *   DuckDB.vector_size -> Integer
+ *
+ * Returns the vector size of DuckDB. The vector size is the number of rows
+ * that are processed in a single vectorized operation.
+ *
+ *   DuckDB.vector_size # => 2048
+ */
+static VALUE duckdb_s_vector_size(VALUE self) {
+    return ULONG2NUM(duckdb_vector_size());
+}
+
 void
 Init_duckdb_native(void) {
     mDuckDB = rb_define_module("DuckDB");
@@ -25,6 +39,7 @@ Init_duckdb_native(void) {
     NegativeInfinity = rb_str_new_literal("-infinity");
 
     rb_define_singleton_method(mDuckDB, "library_version", duckdb_s_library_version, 0);
+    rb_define_singleton_method(mDuckDB, "vector_size", duckdb_s_vector_size, 0);
 
     rbduckdb_init_duckdb_error();
     rbduckdb_init_duckdb_database();

data/lib/duckdb/converter.rb

@@ -12,10 +12,14 @@ module DuckDB
     HALF_HUGEINT = 1 << HALF_HUGEINT_BIT
     FLIP_HUGEINT = 1 << 63
     EPOCH = Time.local(1970, 1, 1)
-    EPOCH_UTC = Time.new(1970, 1, 1, 0, 0, 0, 0)
+    EPOCH_UTC = Time.utc(1970, 1, 1)
 
     module_function
 
+    def default_timezone_utc?
+      defined?(DuckDB.default_timezone) && DuckDB.default_timezone == :utc
+    end
+
     def _to_infinity(value)
       if value.positive?
         DuckDB::Infinity::POSITIVE
@@ -30,11 +34,16 @@ module DuckDB
 
     # rubocop:disable Metrics/ParameterLists
     def _to_time(year, month, day, hour, minute, second, microsecond)
-      Time.local(year, month, day, hour, minute, second, microsecond)
+      Time.public_send(
+        default_timezone_utc? ? :utc : :local,
+        year, month, day, hour, minute, second, microsecond
+      )
     end
     # rubocop:enable Metrics/ParameterLists
 
     def _to_time_from_duckdb_time(hour, minute, second, microsecond)
+      return Time.utc(1970, 1, 1, hour, minute, second, microsecond) if default_timezone_utc?
+
       Time.parse(
         format(
           '%<hour>02d:%<minute>02d:%<second>02d.%<microsecond>06d',
@@ -47,17 +56,23 @@ module DuckDB
     end
 
     def _to_time_from_duckdb_timestamp_s(time)
-      EPOCH + time
+      if default_timezone_utc?
+        EPOCH_UTC + time
+      else
+        EPOCH + time
+      end
     end
 
     def _to_time_from_duckdb_timestamp_ms(time)
-      tm = EPOCH + (time / 1000)
-      Time.local(tm.year, tm.month, tm.day, tm.hour, tm.min, tm.sec, time % 1000 * 1000)
+      _to_time_from_duckdb_timestamp_s(time / 1000).then do |tm|
+        _to_time(tm.year, tm.month, tm.day, tm.hour, tm.min, tm.sec, time % 1000 * 1000)
+      end
     end
 
     def _to_time_from_duckdb_timestamp_ns(time)
-      tm = EPOCH + (time / 1_000_000_000)
-      Time.local(tm.year, tm.month, tm.day, tm.hour, tm.min, tm.sec, time % 1_000_000_000 / 1000)
+      _to_time_from_duckdb_timestamp_s(time / 1_000_000_000).then do |tm|
+        _to_time(tm.year, tm.month, tm.day, tm.hour, tm.min, tm.sec, time % 1_000_000_000 / 1000)
+      end
     end
 
     def _to_time_from_duckdb_time_tz(hour, min, sec, micro, timezone)

data/lib/duckdb/logical_type.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module DuckDB
-  class LogicalType
+  class LogicalType # rubocop:disable Metrics/ClassLength
     alias :alias get_alias
     alias :alias= set_alias
 
@@ -58,9 +58,11 @@ module DuckDB
       def resolve(symbol)
         return symbol if symbol.is_a?(DuckDB::LogicalType)
 
+        raise_resolve_error(symbol) unless symbol.respond_to?(:upcase)
+
         DuckDB::LogicalType.const_get(symbol.upcase)
       rescue NameError
-        raise ArgumentError, "Unknown logical type symbol: #{symbol}"
+        raise_resolve_error(symbol)
       end
 
       # Creates an array logical type with the given child type and size.
@@ -93,6 +95,12 @@ module DuckDB
       def create_list(type)
         _create_list_type(LogicalType.resolve(type))
       end
+
+      private
+
+      def raise_resolve_error(symbol)
+        raise DuckDB::Error, "Unknown logical type: `#{symbol.inspect}`"
+      end
     end
 
     # returns logical type's type symbol
@@ -239,5 +247,15 @@ module DuckDB
         yield dictionary_value_at(i)
       end
     end
+
+    # :nodoc:
+    def inspect
+      "<#{self.class}::#{type.upcase}>"
+    end
+
+    # :nodoc:
+    def to_s
+      inspect
+    end
   end
 end

data/lib/duckdb/scalar_function.rb

@@ -8,9 +8,9 @@ module DuckDB
     # Create and configure a scalar function in one call
     #
     # @param name [String, Symbol] the function name
-    # @param return_type [DuckDB::LogicalType] the return type
-    # @param parameter_type [DuckDB::LogicalType, nil] single parameter type (use this OR parameter_types)
-    # @param parameter_types [Array<DuckDB::LogicalType>, nil] multiple parameter types
+    # @param return_type [DuckDB::LogicalType|:logical_type_symbol] the return type
+    # @param parameter_type [DuckDB::LogicalType|:logical_type_symbol, nil] single parameter type
+    # @param parameter_types [Array<DuckDB::LogicalType|:logical_type_symbol>, nil] multiple parameter types
     # @yield [*args] the function implementation
     # @return [DuckDB::ScalarFunction] configured scalar function ready to register
     # @raise [ArgumentError] if block is not provided or both parameter_type and parameter_types are specified
@@ -18,21 +18,21 @@ module DuckDB
     # @example Single parameter function
     #   sf = DuckDB::ScalarFunction.create(
     #     name: :triple,
-    #     return_type: DuckDB::LogicalType::INTEGER,
-    #     parameter_type: DuckDB::LogicalType::INTEGER
+    #     return_type: :integer,
+    #     parameter_type: :integer
     #   ) { |v| v * 3 }
     #
     # @example Multiple parameters
     #   sf = DuckDB::ScalarFunction.create(
     #     name: :add,
-    #     return_type: DuckDB::LogicalType::INTEGER,
-    #     parameter_types: [DuckDB::LogicalType::INTEGER, DuckDB::LogicalType::INTEGER]
+    #     return_type: :integer,
+    #     parameter_types: [:integer, :integer]
     #   ) { |a, b| a + b }
     #
     # @example No parameters (constant function)
     #   sf = DuckDB::ScalarFunction.create(
     #     name: :random_num,
-    #     return_type: DuckDB::LogicalType::INTEGER
+    #     return_type: :integer
     #   ) { rand(100) }
     def self.create(name:, return_type:, parameter_type: nil, parameter_types: nil, &) # rubocop:disable Metrics/MethodLength
       raise ArgumentError, 'Block required' unless block_given?
@@ -80,17 +80,11 @@ module DuckDB
     # Currently supports BIGINT, BLOB, BOOLEAN, DATE, DOUBLE, FLOAT, INTEGER, SMALLINT, TIME, TIMESTAMP, TINYINT,
     # UBIGINT, UINTEGER, USMALLINT, UTINYINT, and VARCHAR types.
     #
-    # @param logical_type [DuckDB::LogicalType] the parameter type
+    # @param logical_type [DuckDB::LogicalType | :logical_type_symbol] the parameter type
     # @return [DuckDB::ScalarFunction] self
     # @raise [DuckDB::Error] if the type is not supported
     def add_parameter(logical_type)
-      raise DuckDB::Error, 'logical_type must be a DuckDB::LogicalType' unless logical_type.is_a?(DuckDB::LogicalType)
-
-      unless SUPPORTED_TYPES.include?(logical_type.type)
-        type_list = SUPPORTED_TYPES.map(&:upcase).join(', ')
-        raise DuckDB::Error,
-              "Only #{type_list} parameter types are currently supported"
-      end
+      logical_type = check_supported_type!(logical_type)
 
       _add_parameter(logical_type)
     end
@@ -99,19 +93,25 @@ module DuckDB
     # Currently supports BIGINT, BLOB, BOOLEAN, DATE, DOUBLE, FLOAT, INTEGER, SMALLINT, TIME, TIMESTAMP, TINYINT,
     # UBIGINT, UINTEGER, USMALLINT, UTINYINT, and VARCHAR types.
     #
-    # @param logical_type [DuckDB::LogicalType] the return type
+    # @param logical_type [DuckDB::LogicalType | :logical_type_symbol] the return type
     # @return [DuckDB::ScalarFunction] self
     # @raise [DuckDB::Error] if the type is not supported
     def return_type=(logical_type)
-      raise DuckDB::Error, 'logical_type must be a DuckDB::LogicalType' unless logical_type.is_a?(DuckDB::LogicalType)
+      logical_type = check_supported_type!(logical_type)
+
+      _set_return_type(logical_type)
+    end
+
+    private
+
+    def check_supported_type!(type)
+      logical_type = DuckDB::LogicalType.resolve(type)
 
       unless SUPPORTED_TYPES.include?(logical_type.type)
-        type_list = SUPPORTED_TYPES.map(&:upcase).join(', ')
-        raise DuckDB::Error,
-              "Only #{type_list} return types are currently supported"
+        raise DuckDB::Error, "Type `#{type}` is not supported. Only #{SUPPORTED_TYPES.inspect} are available."
       end
 
-      _set_return_type(logical_type)
+      logical_type
     end
   end
 end

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.0.1'
+  VERSION = '1.5.0.2'
 end

data/lib/duckdb.rb

@@ -27,4 +27,28 @@ require 'duckdb/casting'
 
 # DuckDB provides Ruby interface of DuckDB.
 module DuckDB
+  class << self
+    # Controls how DuckDB converts timestamp and time values without explicit
+    # time zone information.
+    #
+    # - `:utc`   - interpret values as UTC
+    # - `:local` - (default) interpret values as local time, preserving existing behavior
+    #
+    # Example:
+    #   DuckDB.default_timezone = :utc
+    #
+    # This setting only affects conversion of values without time zone. Values
+    # with explicit time zone are always interpreted according to their offset.
+    attr_reader :default_timezone
+
+    def default_timezone=(value)
+      raise ArgumentError, 'DuckDB.default_timezone must be either :utc or :local.' unless %i[local utc].include?(value)
+
+      @default_timezone = value
+    end
+  end
+
+  # Default to local time to preserve existing behavior unless explicitly
+  # configured otherwise.
+  self.default_timezone = :local
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: duckdb
 version: !ruby/object:Gem::Version
-  version: 1.5.0.1
+  version: 1.5.0.2
 platform: ruby
 authors:
 - Masaki Suketa