better_batch-active_record 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: fc3cedf70ce459807c93814252ed8c4db04ccba0288e829b40f92d49df1efe21
+  data.tar.gz: 8062fff1d56372dd65eb20a4bd016022c8c81fc50363648a286a0485c0ad67bd
+SHA512:
+  metadata.gz: 793b02a682e20a0ac01cde09d59a91d642b24bf0453eef62d51924111335d0e7f307526274cf434b3b8467934106a91b85357b9c23794a9378e7009a0eb6dfc5
+  data.tar.gz: 542b42bc0e4b7891936d97f0513f5b10ab0ea889d8e5d68c5454a52c140cf89b33e10a8c405f45bb5f1edc5ad6ac419c832cfafff806c4e8f330ff493b63ae38

data/lib/better_batch/active_record/model.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require 'better_batch/active_record/query'
+
+module BetterBatch
+  module ActiveRecord
+    module Model
+      def better_batch
+        Query.new(self)
+      end
+    end
+  end
+end

data/lib/better_batch/active_record/query.rb

@@ -0,0 +1,131 @@
+# frozen_string_literal: true
+
+require 'better_batch/query'
+
+module BetterBatch
+  module ActiveRecord
+    class Query
+      def initialize(model)
+        @model = model
+      end
+
+      def upsert(data, unique_by:, returning:)
+        query = build_query(data, unique_by:, returning:)
+        result = exec_query(:upsert, query, data)
+        case returning
+        when Symbol
+          result.rows.map(&:first)
+        when nil
+          nil
+        else
+          hash_rows(query.returning, result.rows)
+        end
+      end
+
+      def with_upserted_pk(data, unique_by:)
+        query = build_query(data, unique_by:, returning: primary_key)
+        data.zip(exec_query(:upsert, query, data).rows.map(&:first))
+      end
+
+      def set_upserted_pk(data, unique_by:)
+        with_upserted_pk(data, unique_by:).each do |row, pk|
+          row[primary_key] = pk
+        end
+        nil
+      end
+
+      def select(data, unique_by:, returning:)
+        query = build_query(data, unique_by:, returning:)
+        result = exec_query(:select, query, data)
+        case returning
+        when Symbol
+          result.rows.map(&:first)
+        when nil
+          nil
+        else
+          hash_rows(query.returning, result.rows)
+        end
+      end
+
+      def with_selected_pk(data, unique_by:)
+        query = build_query(data, unique_by:, returning: primary_key)
+        data.zip(exec_query(:select, query, data).rows.map(&:first))
+      end
+
+      def set_selected_pk(data, unique_by:)
+        with_upserted_pk(data, unique_by:).each do |row, pk|
+          row[primary_key] = pk
+        end
+        nil
+      end
+
+      private
+
+      attr_reader :model
+
+      def build_query(data, unique_by:, returning:)
+        array_data = data.to_a
+        unique_columns = Array(unique_by)
+        input_columns = array_data.first.keys
+        returning = Array(returning)
+        BetterBatch::Query.new(table_name:, primary_key:, input_columns:, column_types:, unique_columns:,
+                               now_on_insert:, now_on_update:, returning:)
+      end
+
+      def exec_query(type, query, data)
+        sql = build_sql(type, query)
+        json_data = JSON.generate(data)
+        db_exec(sql, query, json_data)
+      end
+
+      def build_sql(type, query)
+        query.public_send(type)
+      rescue StandardError
+        raise query.inspect
+      end
+
+      def db_exec(sql, query, json_data)
+        model.connection.exec_query(sql, nil, [json_data])
+      rescue StandardError
+        raise [query.inspect, query.upsert_formatted].join("\n")
+      end
+
+      def table_name
+        @table_name ||= model.table_name
+      end
+
+      def primary_key
+        model.primary_key.to_sym
+      end
+
+      def column_types
+        @column_types ||= model.columns.to_h { |c| [c.name.to_sym, c.sql_type] }
+      end
+
+      def now_on_insert
+        [created_at_if_present, updated_at_if_present].compact
+      end
+
+      def now_on_update
+        updated_at_if_present
+      end
+
+      def created_at_if_present
+        :created_at if column_types.key?(:created_at)
+      end
+
+      def updated_at_if_present
+        :updated_at if column_types.key?(:updated_at)
+      end
+
+      def hash_rows(returning, rows)
+        # avoid building an entire new hash (which would rehash keys) for each row
+        # we only need to sub in the values
+        indexes = returning.each_with_index.to_h.freeze
+        rows.map do |row|
+          indexes.transform_values { |index| row[index] }
+        end
+      end
+    end
+  end
+end

data/lib/better_batch/active_record/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module BetterBatch
+  module ActiveRecord
+    VERSION = '0.1.0'
+  end
+end

data/lib/better_batch/active_record.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module BetterBatch
+  module ActiveRecord
+  end
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification
+name: better_batch-active_record
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Tyler Hartland
+bindir: exe
+cert_chain: []
+date: 2025-04-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '9'
+- !ruby/object:Gem::Dependency
+  name: better_batch
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+email:
+- tylerhartland7@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/.DS_Store
+- lib/better_batch/.DS_Store
+- lib/better_batch/active_record.rb
+- lib/better_batch/active_record/model.rb
+- lib/better_batch/active_record/query.rb
+- lib/better_batch/active_record/version.rb
+homepage: https://github.com/th7/better_batch-active_record
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  source_code_uri: https://github.com/th7/better_batch-active_record
+  rubygems_mfa_required: 'true'
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.2
+specification_version: 4
+summary: Better batch operations.
+test_files: []