GH-35589: [Ruby] Add support or JRuby

c_glib/arrow-glib/array-builder.cpp

@@ -6320,7 +6320,7 @@ garrow_union_array_builder_class_init(GArrowUnionArrayBuilderClass *klass)
  * garrow_union_array_builder_append_child:
  * @builder: A #GArrowUnionArrayBuilder.
  * @child: A #GArrowArrayBuilder for new child.
- * @filed_name: (nullable): A field name for new child.
+ * @field_name: (nullable): A field name for new child.
  *
  * Returns: The type ID for the appended child.
  *

c_glib/arrow-glib/array-builder.h

@@ -1820,7 +1820,7 @@ GARROW_AVAILABLE_IN_12_0
 gint8
 garrow_union_array_builder_append_child(GArrowUnionArrayBuilder *builder,
                                         GArrowArrayBuilder *child,
-                                        const gchar *filed_name);
+                                        const gchar *field_name);
 
 GARROW_AVAILABLE_IN_12_0
 gboolean

ruby/red-arrow/lib/arrow.rb

@@ -15,16 +15,11 @@
 # specific language governing permissions and limitations
 # under the License.
 
-require "extpp/setup"
-require "gio2"
-
 require "arrow/version"
 
-require "arrow/loader"
-
 module Arrow
   class Error < StandardError
   end
-
-  Loader.load
 end
+
+require_relative "arrow/#{RUBY_ENGINE}"

ruby/red-arrow/lib/arrow/array.rb

@@ -33,9 +33,10 @@ def new(*args)
       end
 
       def builder_class
-        builder_class_name = "#{name}Builder"
-        return nil unless const_defined?(builder_class_name)
-        const_get(builder_class_name)
+        local_name = name.split("::").last
+        builder_class_name = "#{local_name}Builder"
+        return nil unless Arrow.const_defined?(builder_class_name)
+        Arrow.const_get(builder_class_name)
       end
 
       # @api private
@@ -92,6 +93,8 @@ def equal_array?(other, options=nil)
       equal_options(other, options)
     end
 
+    alias_method :size, :length
+
     def each
       return to_enum(__method__) unless block_given?
 
@@ -250,7 +253,7 @@ def resolve(other_array)
             "[array][resolve] need to implement " +
             "a feature that building #{value_data_type} array " +
             "from raw Ruby Array"
-          raise NotImplemented, message
+          raise NotImplementedError, message
         end
         other_array
       elsif other_array.respond_to?(:value_data_type)

ruby/red-arrow/lib/arrow/jruby.rb

@@ -0,0 +1,52 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+if File.exist?("../red-arrow_jars")
+  # installed gems
+  require_relative "../red-arrow_jars"
+else
+  # local development
+  require "red-arrow_jars"
+end
+
+module Arrow
+  class << self
+    def allocator
+      @allocator ||= org.apache.arrow.memory.RootAllocator.new
+    end
+  end
+end
+
+require_relative "jruby/array"
+require_relative "jruby/array-builder"
+require_relative "jruby/chunked-array"
+require_relative "jruby/compression-type"
+require_relative "jruby/csv-read-options"
+require_relative "jruby/decimal128"
+require_relative "jruby/decimal256"
+require_relative "jruby/error"
+require_relative "jruby/file-system"
+require_relative "jruby/function"
+require_relative "jruby/record-batch"
+require_relative "jruby/record-batch-iterator"
+require_relative "jruby/sort-key"
+require_relative "jruby/sort-options"
+require_relative "jruby/stream-listener-raw"
+require_relative "jruby/table"
+require_relative "jruby/writable"
+
+require_relative "libraries"

ruby/red-arrow/lib/arrow/jruby/array-builder.rb

@@ -0,0 +1,114 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+require_relative "array"
+
+module Arrow
+  module ArrayBuildable
+    ValueVector = org.apache.arrow.vector.ValueVector
+    def buildable?(args)
+      return false if args.size == 1 and args.first.is_a?(ValueVector)
+      super
+    end
+  end
+
+  class ArrayBuilder
+    class << self
+      prepend ArrayBuildable
+    end
+
+    def initialize
+      @vector = self.class::Array::Vector.new("", Arrow.allocator)
+      @vector.allocate_new
+      @index = 0
+    end
+
+    def append_value(value)
+      @vector.set(@index, value)
+      @index += 1
+    end
+
+    def append_values(values, is_valids=nil)
+      if is_valids
+        values.zip(is_valids) do |value, is_valid|
+          if is_valid
+            @vector.set(@index, value)
+          else
+            @vector.set_null(@index)
+          end
+          @index += 1
+        end
+      else
+        values.each do |value|
+          @vector.set(@index, value)
+          @index += 1
+        end
+      end
+    end
+
+    def append_nulls(n)
+      n.times do
+        @vector.set_null(@index)
+        @index += 1
+      end
+    end
+
+    def finish
+      @vector.set_value_count(@index)
+      vector, @vector = @vector, nil
+      self.class::Array.new(vector)
+    end
+  end
+
+  class Int8ArrayBuilder < ArrayBuilder
+    Array = Int8Array
+  end
+
+  class Int32ArrayBuilder < ArrayBuilder
+    Array = Int32Array
+  end
+
+  class FixedSizeBinaryArrayBuilder < ArrayBuilder
+  end
+
+  class Decimal128ArrayBuilder < FixedSizeBinaryArrayBuilder
+  end
+
+  class Decimal256ArrayBuilder < FixedSizeBinaryArrayBuilder
+  end
+
+  class ListArrayBuilder < ArrayBuilder
+  end
+
+  class MapArrayBuilder < ArrayBuilder
+  end
+
+  class StructArrayBuilder < ArrayBuilder
+  end
+
+  class UnionArrayBuilder < ArrayBuilder
+    def append_child(child, filed_name)
+      raise NotImplementedError
+    end
+  end
+
+  class DenseUnionArrayBuilder < UnionArrayBuilder
+  end
+
+  class SparseUnionArrayBuilder < UnionArrayBuilder
+  end
+end

ruby/red-arrow/lib/arrow/jruby/array.rb

@@ -0,0 +1,109 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+require_relative "data-type"
+
+module Arrow
+  class Array
+    VectorAppender = org.apache.arrow.vector.util.VectorAppender
+    VectorEqualsVisitor = org.apache.arrow.vector.compare.VectorEqualsVisitor
+
+    attr_reader :vector
+
+    def initialize(vector)
+      @vector = vector
+    end
+
+    def ==(other_array)
+      return false unless other_array.is_a?(self.class)
+      VectorEqualsVisitor.vector_equals(@vector, other_array.vector)
+    end
+
+    def null?(i)
+      @vector.null?(i)
+    end
+
+    def get_value(i)
+      @vector.get_object(i)
+    end
+
+    def to_s
+      @vector.to_s
+    end
+
+    def inspect
+      super.sub(/>\z/) do
+        " #{to_s}>"
+      end
+    end
+
+    def close
+      @vector.close
+    end
+
+    def length
+      @vector.value_count
+    end
+
+    def value_data_type
+      self.class::ValueDataType.new
+    end
+
+    def values
+      each.to_a
+    end
+
+    def cast(other_value_data_type)
+      other_value_data_type.build_array(to_a)
+    end
+
+    def is_in(values)
+      raise NotImplementedError
+    end
+
+    def concatenate(other_arrays)
+      total_size = length + other_arrays.sum(&:length)
+      vector = self.class::Vector.new("", Arrow.allocator)
+      vector.allocate_new(total_size)
+      appender = VectorAppender.new(vector)
+      @vector.accept(appender, nil)
+      other_arrays.each do |other_array|
+        other_array.vector.accept(appender, nil)
+      end
+      self.class.new(vector)
+    end
+  end
+
+  class Int8Array < Array
+    Vector = org.apache.arrow.vector.SmallIntVector
+    ValueDataType = Int8DataType
+  end
+
+  class Int32Array < Array
+    Vector = org.apache.arrow.vector.IntVector
+    ValueDataType = Int32DataType
+  end
+
+  class FixedSizeBinaryArray < Array
+  end
+
+  class StructArray < Array
+    def fields
+      raise NotImplementedError
+    end
+  end
+end

ruby/red-arrow/lib/arrow/jruby/chunked-array.rb

@@ -0,0 +1,36 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+module Arrow
+  class ChunkedArray
+    def initialize(arrays)
+      @arrays = arrays
+    end
+
+    def n_rows
+      @arrays.sum(&:size)
+    end
+
+    def chunks
+      @arrays
+    end
+
+    def get_chunk(i)
+      @arrays[i]
+    end
+  end
+end