Import

Author: kou

.github/dependabot.yml

@@ -0,0 +1,6 @@
+version: 2
+updates:
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: "weekly"

.github/workflows/test.yaml

@@ -0,0 +1,39 @@
+name: Test
+
+on:
+  push:
+  pull_request:
+
+jobs:
+  test:
+    name: "Ruby ${{ matrix.ruby-version }}: ${{ matrix.runs-on }}"
+    strategy:
+      fail-fast: false
+      matrix:
+        ruby-version:
+          - "3.0"
+          - "3.1"
+          - "3.2"
+        runs-on:
+          - macos-latest
+          - ubuntu-latest
+          # - windows-latest
+    runs-on: ${{ matrix.runs-on }}
+    steps:
+      - uses: actions/checkout@v3
+      - name: Prepare the Apache Arrow APT repository
+        if: |
+          runner.os == 'Linux'
+        run: |
+          sudo apt update
+          sudo apt install -y -V ca-certificates lsb-release wget
+          wget https://apache.jfrog.io/artifactory/arrow/$(lsb_release --id --short | tr 'A-Z' 'a-z')/apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb
+          sudo apt install -y -V ./apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb
+          sudo apt update
+      - uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: ${{ matrix.ruby-version }}
+          bundler-cache: true
+      - name: Test
+        run: |
+          bundle exec rake

.gitignore

@@ -0,0 +1 @@
+/Gemfile.lock

Gemfile

@@ -0,0 +1,31 @@
+# -*- ruby -*-
+
+source "https://rubygems.org/"
+
+gemspec
+
+gem "bundler"
+gem "rake"
+gem "test-unit"
+
+local_rails = File.expand_path(File.join(__dir__, "..", "rails"))
+if File.exist?(local_rails)
+  path local_rails do
+    gem "activerecord"
+  end
+else
+  git "https://github.com/rails/rails.git" do
+    gem "activerecord"
+  end
+end
+
+local_adbc = File.expand_path(File.join(__dir__, "..", "arrow-adbc"))
+if File.exist?(local_adbc)
+  path "#{local_adbc}/ruby" do
+    gem "red-adbc"
+  end
+else
+  git "https://github.com/apache/arrow-adbc.git" do
+    gem "red-adbc"
+  end
+end

LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright 2023 Sutou Kouhei <[email protected]>
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

README.md

@@ -0,0 +1,20 @@
+# Active Record ADBC adapter
+
+## Description
+
+Active Record ADBC adapter provides an ADBC adapter for Active Record.
+
+This adapter is optimized for extracting and loading large data
+from/to DBs. The optimization is powered by Apache Arrow.
+
+## Install
+
+TODO
+
+## Usage
+
+TODO
+
+## License
+
+The MIT license. See `LICENSE.txt` for details.

Rakefile

@@ -0,0 +1,21 @@
+# -*- ruby -*-
+
+require "rubygems"
+require "bundler/gem_helper"
+
+base_dir = File.join(__dir__)
+
+helper = Bundler::GemHelper.new(base_dir)
+helper.install
+
+release_task = Rake::Task["release"]
+release_task.prerequisites.replace(["build", "release:rubygem_push"])
+
+desc "Run tests"
+task :test do
+  cd(base_dir) do
+    ruby("test/run.rb")
+  end
+end
+
+task default: :test

activerecord-adbc-adapter.gemspec

@@ -0,0 +1,28 @@
+# -*- ruby -*-
+
+require "fileutils"
+
+require_relative "lib/activerecord-adbc-adapter/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "activerecord-adbc-adapter"
+  spec.version = ActiveRecordADBCAdapter::VERSION
+  spec.homepage = "https://github.com/red-data-tools/activerecord-adbc-adapter"
+  spec.authors = ["Sutou Kouhei"]
+  spec.email = ["[email protected]"]
+
+  spec.summary = "Active Record's ADBC adapter"
+  spec.description =
+    "This gem provides an ADBC adapter for Active Record. " +
+    "This adapter is optimized for extracting and loading large data " +
+    "from/to DBs. The optimization is powered by Apache Arrow."
+  spec.license = "MIT"
+  spec.files = [
+    "LICENSE.txt",
+    "README.md",
+  ]
+  spec.files += Dir.glob("lib/**/*.rb")
+
+  spec.add_runtime_dependency("activerecord")
+  spec.add_runtime_dependency("red-arrow")
+end

lib/active_record/connection_adapters/adbc/column.rb

@@ -0,0 +1,8 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module ADBC
+      class Column < ConnectionAdapters::Column # :nodoc:
+      end
+    end
+  end
+end

lib/active_record/connection_adapters/adbc/database_statements.rb

@@ -0,0 +1,22 @@
+module ActiveRecord
+  module ConnectionAdapters
+    module ADBC
+      module DatabaseStatements
+        def internal_exec_query(sql,
+                                name = "SQL",
+                                binds = [],
+                                prepare: false,
+                                async: false,
+                                allow_retry: false,
+                                materialize_transactions: true) # :nodoc:
+          casted_binds = type_casted_binds(binds)
+          log(sql, name, binds, casted_binds, async: async) do
+            with_raw_connection do |conn|
+              build_arrow_result(conn.query(sql, casted_binds))
+            end
+          end
+        end
+      end
+    end
+  end
+end