red-adbc 0.11.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 97bc5368186dbbb7e3f23357bba96d58ab2401e3c37a6cf2bcd039b9880875a7
-  data.tar.gz: b3e8197a27f928f8e29cd617928a2eccc2ac08717244f8395b055440d730a7ac
+  metadata.gz: b3e38661bb51d363808e530520d34e2db0c9ee10760e3131a3fc29ce8d24603c
+  data.tar.gz: bfa56ab21c18d1d415c80b1f702f74a8b65b249e414ae81fe175ac32fa1668d4
 SHA512:
-  metadata.gz: 03f0422fea96fed1cb60caa6b5757b8744aab4c2c19e6a75b1c76fbdd6890baabb9df438cd1f1903472001029755e25ea90cd06195fd043e0d5127ccb189d9a8
-  data.tar.gz: d4bd60c141d0cb11cd4214f4189528214c2c4780fcb92a1c98f64d6bc87c178446f637dd4318e1567f761b49938751a86a79d7186056ab1b2fd0559c4315f177
+  metadata.gz: 227b2de351ca2a2470a87899255eb0d42f563549c556de83e79c5198f1591de5bb620c6f359c91061471bed5dcdb70a8fe2db3c5175eaadd490a5cda50098e9f
+  data.tar.gz: 640804850814b5bab21f6c3cc5f50b443104c79b15bbb440d271d78870bdec43f55f5739dc73aebb9cbf08577c512aa4466dfc7efab18edf08a9af34d69220a1

data/lib/adbc/arrow-connection.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 ADBCArrow
+  class Connection
+    include ADBC::ConnectionOperations
+
+    def open_statement(&block)
+      Statement.open(self, &block)
+    end
+
+    alias_method :get_info_raw, :get_info
+    def get_info(codes)
+      reader = get_info_raw(codes)
+      begin
+        yield(reader.read_all)
+      ensure
+        reader.unref
+      end
+    end
+  end
+end

data/lib/adbc/connection-operations.rb

@@ -0,0 +1,56 @@
+# 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 ADBC
+  module ConnectionOperations
+    def query(sql, &block)
+      open_statement do |statement|
+        statement.query(sql, &block)
+      end
+    end
+
+    def ingest(table_name, values, mode: :create)
+      open_statment do |statement|
+        statement.ingest(table_name, values, mode: mode)
+      end
+    end
+
+    def info(codes=nil)
+      unless codes.nil?
+        codes = codes.collect do |code|
+          ADBC::Info.try_convert(code)
+        end
+      end
+      get_info(codes) do |table|
+        values = {}
+        table.raw_records.each do |code, value|
+          value = value.values[0] if value.is_a?(Hash)
+          code = ADBC::Info.try_convert(code)
+          values[code.nick.gsub("-", "_").to_sym] = value
+        end
+        values
+      end
+    end
+
+    ADBC::Info.values.each do |value|
+      name = value.nick.gsub("-", "_").to_sym
+      define_method(name) do
+        info([name])[name]
+      end
+    end
+  end
+end

data/lib/adbc/connection.rb

@@ -15,42 +15,23 @@
 # specific language governing permissions and limitations
 # under the License.
 
+require_relative "connection-operations"
+
 module ADBC
   class Connection
+    include ConnectionOperations
+
     def open_statement(&block)
       Statement.open(self, &block)
     end
 
-    def query(sql, &block)
-      open_statement do |statement|
-        statement.query(sql, &block)
-      end
-    end
-
-    def ingest(table_name, values, mode: :create)
-      open_statment do |statement|
-        statement.ingest(table_name, values, mode: mode)
-      end
-    end
-
-    def info(codes=nil)
-      unless codes.nil?
-        codes = codes.collect do |code|
-          ADBC::Info.try_convert(code)
-        end
-      end
-      c_abi_array_stream = get_info(codes)
+    alias_method :get_info_raw, :get_info
+    def get_info(codes)
+      c_abi_array_stream = get_info_raw(codes)
       begin
         reader = Arrow::RecordBatchReader.import(c_abi_array_stream)
         begin
-          table = reader.read_all
-          values = {}
-          table.raw_records.each do |code, value|
-            value = value.values[0] if value.is_a?(Hash)
-            code = ADBC::Info.try_convert(code)
-            values[code.nick.gsub("-", "_").to_sym] = value
-          end
-          values
+          yield(reader.read_all)
         ensure
           reader.unref
         end
@@ -58,12 +39,5 @@ module ADBC
         GLib.free(c_abi_array_stream)
       end
     end
-
-    ADBC::Info.values.each do |value|
-      name = value.nick.gsub("-", "_").to_sym
-      define_method(name) do
-        info([name])[name]
-      end
-    end
   end
 end

data/lib/adbc/loader.rb

@@ -58,6 +58,7 @@ module ADBCArrow
     end
 
     def require_libraries
+      require_relative "arrow-connection"
       require_relative "arrow-statement"
     end
   end

data/lib/adbc/version.rb

@@ -16,7 +16,7 @@
 # under the License.
 
 module ADBC
-  VERSION = "0.11.0"
+  VERSION = "1.0.0"
 
   module Version
     MAJOR, MINOR, MICRO, TAG = VERSION.split(".").collect(&:to_i)

data/lib/adbc.rb

@@ -32,6 +32,10 @@ begin
 rescue GObjectIntrospection::RepositoryError
 else
   module ADBC
+    RawConnection = Connection
+    remove_const(:Connection)
+    Connection = ADBCArrow::Connection
+
     RawStatement = Statement
     remove_const(:Statement)
     Statement = ADBCArrow::Statement

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: red-adbc
 version: !ruby/object:Gem::Version
-  version: 0.11.0
+  version: 1.0.0
 platform: ruby
 authors:
 - Apache Arrow Developers
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-31 00:00:00.000000000 Z
+date: 2024-05-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: red-arrow
@@ -38,7 +38,9 @@ files:
 - README.md
 - dependency-check/Rakefile
 - lib/adbc.rb
+- lib/adbc/arrow-connection.rb
 - lib/adbc/arrow-statement.rb
+- lib/adbc/connection-operations.rb
 - lib/adbc/connection.rb
 - lib/adbc/database.rb
 - lib/adbc/loader.rb