red-arrow-flight-sql 13.0.0 → 14.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e525a6ce012f2ed5e39f3753c92d4310d3e338cf8ad14cb2521513bcb9031e45
-  data.tar.gz: bbd0a34d90d72f8618a174c1d44fd6c49b76793b7164750eaf89e6d4079f8dc0
+  metadata.gz: b3f60f22e7fa72b79a9ce10057081c7159cbae5d5f4201998fb67d527e2c6714
+  data.tar.gz: 3c9dfbcc05fc74b3fefd7ca6431f83d12fea9abd3422f2ae06b7c23d59c17b43
 SHA512:
-  metadata.gz: b628f5f9be098d7f198fc6968eba772b1f08034391e1cccc2ffbab2c78c5a5906eb0c30e55b0f205533572c15aa0b4c5eb90b35f466b02a603df0da79998c0a4
-  data.tar.gz: 941b59ad57580e1940cfbf85bbeac8773d7ce5df8d2f0d5cc11a1a9c3c8aa7c51e54a08ed203af166a03f6b2bf676fb241ddf9e707e27c64a2c8ffb58382868f
+  metadata.gz: 50038a8e3fbe80cd494e04288a84ddeacc8ec09aa00947ed13e653f9617281bf082c879d5a1e7dd837b562fae867542ed1b91529edf301213ea6791bf1c064b8
+  data.tar.gz: 68fdd9b8a7032e5194b479e6bd66d274cc13922569d2e462ed78cff742240dd6d6fa9e7a08b037d5cfd4666586854d78d371d3c1e552e5a19343b8c99fc61b80

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.1"
 
   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.1
 platform: ruby
 authors:
 - Apache Arrow Developers
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-08-28 00:00:00.000000000 Z
+date: 2023-11-13 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.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 13.0.0
+        version: 14.0.1
 - !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