active_record-pool 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1b121f2e0e34ac06dbb412139c8f27ed201bae07
+  data.tar.gz: 999918bde34952536d00121b133c77bfad2a705d
+SHA512:
+  metadata.gz: 984ccc454a7187c02852bc8d2aec247dc62ca52334b6fe9f22b7c95457268da834a3c23dc76c8465d898952781f454758c38edc48cfa5660ec4503faa9201809
+  data.tar.gz: c17a8f1f8dda3a4ff0deede9382add61171a68e1428085168fd4742d3bd3b09409e9fb82ea9154f1a828bc824d0de33195ff5b3e82e0f8353914690e196314de

data/lib/active_record-pool.rb

@@ -0,0 +1,5 @@
+require 'active_record'
+
+module ActiveRecord
+  require_relative 'active_record/pool'
+end

data/lib/active_record/base_extension.rb

@@ -0,0 +1,7 @@
+module ActiveRecord
+  class Base
+    def self.pool(columns:, query: self, table: self.table_name, size: ActiveRecord::Pool::DEFAULT_SIZE, serializer: ActiveRecord::Pool::DEFAULT_SERIALIZER, &iteration)
+      ActiveRecord::Pool.new(columns: columns, query: query, table: table, size: size, serializer: serializer, model: self, &iteration)
+    end
+  end
+end

data/lib/active_record/pool.rb

@@ -0,0 +1,104 @@
+module ActiveRecord
+  require_relative "base_extension"
+
+  class Pool
+    require_relative "pool/version"
+
+    DEFAULT_SIZE = 24
+    DEFAULT_SERIALIZER = ::JSON
+    EMPTY_HASH = {}
+
+    # `query` is either an ActiveRecord query object or arel
+    # `columns` is a list of columns you want to have during the transaction
+    # `table` is the table you want to talk to
+    # `size` is the maximum number of running iterations in the pool, default: 24
+    # `serializer` is the #dump duck for Array & Hash values, default: JSON
+    # `model` is an ActiveRecord model
+    # `transaction` is the process you want to run against your database
+    def initialize(query:, columns:, table:, size:, serializer:, model:, &transaction)
+      @query = query
+      @serializer = serializer
+      @table = Arel::Table.new(table)
+      qutex = Mutex.new
+
+      queue = case
+      when activerecord?
+        @query.pluck(*@columns)
+      when arel?
+        ActiveRecord::Base.connection.execute(@query.to_sql).map(&:values)
+      when tuple?
+        @query.map { |result| result.slice(*columns).values }
+      when twodimensional?
+        @query
+      else
+        raise ArgumentError, 'query wasn\'t recognizable, please use some that looks like a: ActiveRecord::Base, Arel::SelectManager, Array[Hash], Array[Array]'
+      end
+
+      puts "Migrating #{queue.count} #{table} records"
+
+      # Spin up a number of threads based on the `maximum` given
+      1.upto(size).map do
+        Thread.new do
+          loop do
+            # Try to get a new queue item
+            item = qutex.synchronize { queue.shift }
+
+            if item.nil?
+              # There is no more work
+              break
+            else
+              # Wait for a free connection
+              model.connection_pool.with_connection do
+                model.transaction do
+                  # Execute each statement coming back
+                  Array[instance_exec(*item, &transaction)].each do |instruction|
+                    next if instruction.nil?
+                    model.connection.execute(instruction.to_sql)
+                  end
+                end
+              end
+            end
+          end
+        end
+      end.map(&:join)
+    end
+
+    private def activerecord?
+      @query.kind_of?(ActiveRecord::Base) || @query.kind_of?(ActiveRecord::Relation) || @query < ActiveRecord::Base
+    end
+
+    private def arel?
+      @query.kind_of?(Arel::SelectManager)
+    end
+
+    private def tuple?
+      @query.kind_of?(Array) && @query.first.kind_of?(Hash)
+    end
+
+    private def twodimensional?
+      @query.kind_of?(Array) && @query.first.kind_of?(Array)
+    end
+
+    private def update(id, data)
+      Arel::UpdateManager.new.table(@table).where(@table[:id].eq(id)).set(serialize(data))
+    end
+
+    private def insert(data)
+      Arel::InsertManager.new.tap { |m| m.insert(serialize(data)) }
+    end
+
+    private def delete(id)
+      Arel::DeleteManager.new.from(@table).where(@table[:id].eq(id))
+    end
+
+    private def serialize(data)
+      data.inject(EMPTY_HASH) do |state, (key, value)|
+        if value.is_a?(Array) || value.is_a?(Hash)
+          state.merge(@table[key] => @serializer.dump(value))
+        else
+          state.merge(@table[key] => value)
+        end
+      end
+    end
+  end
+end

data/lib/active_record/pool/version.rb

@@ -0,0 +1,5 @@
+module ActiveRecord
+  class Pool
+    VERSION = "1.0.0"
+  end
+end

data/spec/lib/active_record/pool/version_spec.rb

@@ -0,0 +1,7 @@
+require "spec_helper"
+
+RSpec.describe ActiveRecord::Pool::VERSION do
+  it "is a string" do
+    expect(ActiveRecord::Pool::VERSION).to be_kind_of(String)
+  end
+end

data/spec/lib/active_record/pool_spec.rb

@@ -0,0 +1,59 @@
+require "spec_helper"
+
+RSpec.describe ActiveRecord::Pool do
+  with_model :Activity do
+    table do |schema|
+      schema.string :kind
+      schema.string :category
+    end
+  end
+  let(:model) { Activity }
+
+  before do
+    allow(Thread).to receive(:new).and_yield.and_return(instance_double("Thread", join: true))
+    75.times do
+      Activity.create(kind: "like")
+    end
+    25.times do
+      Activity.create(kind: "dislike")
+    end
+  end
+
+  describe ".pool" do
+    context 'with an delete' do
+      let(:pool) do
+        model.pool(columns: [:id, :kind]) do |id, kind|
+          if kind == "like" then delete(id) end
+        end
+      end
+
+      it "deletes 75 likes" do
+        expect { pool }.to change(Activity.where(kind: "like"), :count).from(75).to(0)
+      end
+    end
+
+    context 'with an insert' do
+      let(:pool) do
+        model.pool(columns: [:id, :kind]) do |id, kind|
+          if kind == "like" then insert(kind: "response", category: "like") end
+        end
+      end
+
+      it "creates 75 responses likes" do
+        expect { pool }.to change(Activity.where(kind: "response", category: "like"), :count).from(0).to(75)
+      end
+    end
+
+    context 'with an update' do
+      let(:pool) do
+        model.pool(columns: [:id, :kind]) do |id, kind|
+          if kind == "like" then update(id, kind: "dislike") end
+        end
+      end
+
+      it "updates 75 likes to dislikes" do
+        expect { pool }.to change(Activity.where(kind: "dislike"), :count).from(25).to(100)
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,27 @@
+require "pry"
+require "rspec"
+require "with_model"
+require "active_record-pool"
+
+ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => ":memory:")
+
+RSpec.configure do |let|
+
+  # Exit the spec after the first failure
+  let.fail_fast = true
+
+  # Only run a specific file, using the ENV variable
+  # Example: FILE=spec/write/version_spec.rb bundle exec rake spec
+  let.pattern = ENV["FILE"]
+
+  # Show the slowest examples in the suite
+  let.profile_examples = true
+
+  # Colorize the output
+  let.color = true
+
+  # Output as a document string
+  let.default_formatter = "doc"
+
+  let.extend WithModel
+end

metadata

@@ -0,0 +1,168 @@
+--- !ruby/object:Gem::Specification
+name: active_record-pool
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Kurtis Rainbolt-Greene
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-06-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: with_model
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.1'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: pry-doc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+description: active_record-pool is an extension to active_record that gives you an
+  interface to write high-speed inserts, updates, & delete in a manageable way.
+email:
+- kurtis@laurelandwolf.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/active_record-pool.rb
+- lib/active_record/base_extension.rb
+- lib/active_record/pool.rb
+- lib/active_record/pool/version.rb
+- spec/lib/active_record/pool/version_spec.rb
+- spec/lib/active_record/pool_spec.rb
+- spec/spec_helper.rb
+homepage: http://laurelandwolf.github.io/active_record-pool
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.12
+signing_key: 
+specification_version: 4
+summary: active_record-pool is an extension to active_record that gives you an interface
+  to write high-speed inserts, updates, & delete in a manageable way.
+test_files:
+- spec/lib/active_record/pool/version_spec.rb
+- spec/lib/active_record/pool_spec.rb
+- spec/spec_helper.rb