red-arrow-flight 16.1.0 → 18.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9afcc53db63194b4d083cfb495de23387ca1147706196d4bed8b1bad5fa58e74
-  data.tar.gz: ec0dc4f922cef09878c71f546771fba26d9e74fdb87ae7a80d590004879b9a49
+  metadata.gz: 658c5a7dad91b476c435276383713b91730dd3192e033532eaea8e88b4554349
+  data.tar.gz: ff25e6c1927a7d56b0c547933da0e9201998e56ac7df04b292385311acddebed
 SHA512:
-  metadata.gz: 158620d95f5ebb76fab3604de3eabcada6b025e6ae9fa2e13aac4fd53dfd570065a32b1bd2714aa7059a15c8f8d81189eb7e7a7ce2e9b38cd755842cb0c0d4e4
-  data.tar.gz: dba5936e3d99b8a148ceea71442c363ae38c841403366f6ed48feb5237fd06a51e3176deb14507c655943576c81c652fad7c9081799db3d22a1cb7edb51e266d
+  metadata.gz: 87247bf8ddcb3ca84f38b0e269a5d04989683439b1638a33c8b9e77d89547d22c621bd716037edeaa4ca4afc07fe8b1c2f22ee25d177a0108e8503dcf7f8048e
+  data.tar.gz: e911e98bf8b68eaa3d6ef809a1b77ebb20f9658805ccdab4446d27d2acca03bb2bb31301fc2d158ac7e1b09128ebabffe55062abb906b0a5f6a06121540e1397

data/lib/arrow-flight/client.rb

@@ -47,5 +47,49 @@ module ArrowFlight
       end
       options
     end
+
+    alias_method :do_put_raw, :do_put
+    # Upload data to a Flight described by the given descriptor. The
+    # caller must call `#close` on the returned stream once they are
+    # done writing. Note that it's automatically done when you use
+    # block.
+    #
+    # The reader and writer are linked; closing the writer will also
+    # close the reader. Use GArrowFlight::StreamWriter#done_writing to
+    # only close the write side of the channel.
+    #
+    # @param descriptor [GArrowFlight::Descriptor] Descriptor to be uploaded.
+    # @param schema [GArrow::Schema] Schema of uploaded data.
+    # @param options [ArrowFlight::CallOptions, Hash, nil] (nil)
+    #   The options to be used.
+    #
+    # @yieldparam writer [GArrowFlight::StreamWriter] The writer to upload
+    #   data to the given descriptor.
+    #
+    #   This is closed automatically after the given block is finished.
+    #
+    # @yieldparam reader [GArrowFlight::MetadataReader] The reader to read
+    #   metadata from the server.
+    #
+    # @return [Array<GArrowFlight::MetadataReader, GArrowFlight::StreamWriter>, Object]
+    #   The reader and the writer if block isn't given.
+    #
+    #   The return value from block if block is given.
+    #
+    # @since 18.0.0
+    def do_put(descriptor, schema, options=nil)
+      result = do_put_raw(descriptor, schema, options)
+      reader = result.reader
+      writer = result.writer
+      if block_given?
+        begin
+          yield(reader, writer)
+        ensure
+          writer.close unless writer.closed?
+        end
+      else
+        return reader, writer
+      end
+    end
   end
 end

data/lib/arrow-flight/version.rb

@@ -16,7 +16,7 @@
 # under the License.
 
 module ArrowFlight
-  VERSION = "16.1.0"
+  VERSION = "18.0.0"
 
   module Version
     numbers, TAG = VERSION.split("-")

data/test/helper/server.rb

@@ -21,6 +21,8 @@ module Helper
   class Server < ArrowFlight::Server
     type_register
 
+    attr_reader :uploaded_table
+
     private
     def virtual_do_list_flights(context, criteria)
       generator = InfoGenerator.new
@@ -35,5 +37,14 @@ module Helper
       table = generator.page_view_table
       ArrowFlight::RecordBatchStream.new(table)
     end
+
+    def virtual_do_do_put(context, reader, writer)
+      @uploaded_table = reader.read_all
+      writer.write(Arrow::Buffer.new("done"))
+      if @uploaded_table.n_rows.zero?
+        raise Arrow::Error::Invalid.new("empty table")
+      end
+      true
+    end
   end
 end

data/test/test-client.rb

@@ -43,4 +43,35 @@ class TestClient < Test::Unit::TestCase
     assert_equal(generator.page_view_table,
                  reader.read_all)
   end
+
+  def test_do_put_with_block
+    client = ArrowFlight::Client.new(@location)
+    generator = Helper::InfoGenerator.new
+    descriptor = generator.page_view_descriptor
+    table = generator.page_view_table
+    client.do_put(descriptor, table.schema) do |reader, writer|
+      writer.write_table(table)
+      writer.done_writing
+      metadata = reader.read
+      assert_equal(["done", table],
+                   [metadata.data.to_s, @server.uploaded_table])
+    end
+  end
+
+  def test_do_put_without_block
+    client = ArrowFlight::Client.new(@location)
+    generator = Helper::InfoGenerator.new
+    descriptor = generator.page_view_descriptor
+    table = generator.page_view_table
+    reader, writer = client.do_put(descriptor, table.schema)
+    begin
+      writer.write_table(table)
+      writer.done_writing
+      metadata = reader.read
+      assert_equal(["done", table],
+                   [metadata.data.to_s, @server.uploaded_table])
+    ensure
+      writer.close
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: red-arrow-flight
 version: !ruby/object:Gem::Version
-  version: 16.1.0
+  version: 18.0.0
 platform: ruby
 authors:
 - Apache Arrow Developers
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-05-27 00:00:00.000000000 Z
+date: 2024-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: red-arrow
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 16.1.0
+        version: 18.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 16.1.0
+        version: 18.0.0
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -106,7 +106,7 @@ homepage: https://arrow.apache.org/
 licenses:
 - Apache-2.0
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -121,8 +121,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.5
-signing_key: 
+rubygems_version: 3.4.20
+signing_key:
 specification_version: 4
 summary: Red Arrow Flight is the Ruby bindings of Apache Arrow Flight
 test_files: