red-arrow-flight-sql 13.0.0 → 14.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e525a6ce012f2ed5e39f3753c92d4310d3e338cf8ad14cb2521513bcb9031e45
-  data.tar.gz: bbd0a34d90d72f8618a174c1d44fd6c49b76793b7164750eaf89e6d4079f8dc0
+  metadata.gz: 3edaa6d2d45fb6c78a565cb41d0786df654932e35e91615a95248076d77631d3
+  data.tar.gz: 980519b323808a7134d70aa1ea285901bc5b588001eda6cf6772b4ffefe487b4
 SHA512:
-  metadata.gz: b628f5f9be098d7f198fc6968eba772b1f08034391e1cccc2ffbab2c78c5a5906eb0c30e55b0f205533572c15aa0b4c5eb90b35f466b02a603df0da79998c0a4
-  data.tar.gz: 941b59ad57580e1940cfbf85bbeac8773d7ce5df8d2f0d5cc11a1a9c3c8aa7c51e54a08ed203af166a03f6b2bf676fb241ddf9e707e27c64a2c8ffb58382868f
+  metadata.gz: ee44d8f7f9ea9ddc5bf5eaf413af8be396f2983d844f3f7341e986035450d0ad431df370838e0265532ee97f446ec7722b9989a3d58b435765e3547b96e05ea4
+  data.tar.gz: ae7c613ce9bfcb085cbba33bd554666be820c00287f67a69e26bbb57ceff0e9b6a406114769f28bbf283b617b4843986b5cd1446ec9f7813f2bff0c9afe10398

data/lib/arrow-flight-sql/client.rb

@@ -0,0 +1,34 @@
+# 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 ArrowFlightSQL
+  class Client
+    alias_method :prepare_raw, :prepare
+    def prepare(query, options=nil)
+      statement = prepare_raw(query, options)
+      if block_given?
+        begin
+          yield(statement)
+        ensure
+          statement.close(options) unless statement.closed?
+        end
+      else
+        statement
+      end
+    end
+  end
+end

data/lib/arrow-flight-sql/loader.rb

@@ -29,6 +29,7 @@ module ArrowFlightSQL
     end
 
     def require_libraries
+      require_relative "client"
       require_relative "server"
     end
 

data/lib/arrow-flight-sql/version.rb

@@ -16,7 +16,7 @@
 # under the License.
 
 module ArrowFlightSQL
-  VERSION = "13.0.0"
+  VERSION = "14.0.0"
 
   module Version
     numbers, TAG = VERSION.split("-")

data/test/helper/server.rb

@@ -37,5 +37,36 @@ module Helper
       table = generator.page_view_table
       ArrowFlight::RecordBatchStream.new(table)
     end
+
+    def virtual_do_create_prepared_statement(context, request)
+      unless request.query == "INSERT INTO page_view_table VALUES ($1, true)"
+        raise Arrow::Error::Invalid.new("invalid SQL")
+      end
+      result = ArrowFlightSQL::CreatePreparedStatementResult.new
+      generator = InfoGenerator.new
+      table = generator.page_view_table
+      result.dataset_schema = table.schema
+      result.parameter_schema = table.schema.remove_field(1)
+      result.handle = "valid-handle"
+      result
+    end
+
+    def virtual_do_do_put_prepared_statement_update(context, command, reader)
+      unless command.handle.to_s == "valid-handle"
+        raise Arrow::Error::Invalid.new("invalid handle")
+      end
+      reader.read_all.n_rows
+    end
+
+    def virtual_do_close_prepared_statement(context, request)
+      unless request.handle.to_s == "valid-handle"
+        raise Arrow::Error::Invalid.new("invalid handle")
+      end
+      access_key = context.incoming_headers.assoc("x-access-key")
+      unless access_key == ["x-access-key", "secret"]
+        message = "invalid access key: #{access_key.inspect}"
+        raise Arrow::Error::Invalid.new(message)
+      end
+    end
   end
 end

data/test/test-client.rb

@@ -39,4 +39,27 @@ class TestClient < Test::Unit::TestCase
     assert_equal(generator.page_view_table,
                  reader.read_all)
   end
+
+  def test_prepare
+    insert_sql = "INSERT INTO page_view_table VALUES ($1, true)"
+    block_called = false
+    options = ArrowFlight::CallOptions.new
+    options.add_header("x-access-key", "secret")
+    @sql_client.prepare(insert_sql, options) do |statement|
+      block_called = true
+      assert_equal([
+                     Arrow::Schema.new(count: :uint64, private: :boolean),
+                     Arrow::Schema.new(count: :uint64),
+                   ],
+                   [
+                     statement.dataset_schema,
+                     statement.parameter_schema,
+                   ])
+      counts = Arrow::UInt64Array.new([1, 2, 3])
+      parameters = Arrow::RecordBatch.new(count: counts)
+      statement.set_record_batch(parameters)
+      assert_equal(3, statement.execute_update)
+    end
+    assert_true(block_called)
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: red-arrow-flight-sql
 version: !ruby/object:Gem::Version
-  version: 13.0.0
+  version: 14.0.0
 platform: ruby
 authors:
 - Apache Arrow Developers
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-08-28 00:00:00.000000000 Z
+date: 2023-11-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: red-arrow-flight
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 13.0.0
+        version: 14.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 13.0.0
+        version: 14.0.0
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -82,6 +82,7 @@ files:
 - Rakefile
 - dependency-check/Rakefile
 - lib/arrow-flight-sql.rb
+- lib/arrow-flight-sql/client.rb
 - lib/arrow-flight-sql/loader.rb
 - lib/arrow-flight-sql/server.rb
 - lib/arrow-flight-sql/version.rb
@@ -95,7 +96,7 @@ homepage: https://arrow.apache.org/
 licenses:
 - Apache-2.0
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -110,8 +111,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.5.0.dev
+signing_key:
 specification_version: 4
 summary: Red Arrow Flight SQL is the Ruby bindings of Apache Arrow Flight SQL
 test_files: