active_record_shards 4.0.0.beta9 → 5.0.0

data/lib/active_record_shards/default_slave_patches.rb

@@ -1,136 +0,0 @@
-# frozen_string_literal: true
-module ActiveRecordShards
-  module DefaultSlavePatches
-    def self.wrap_method_in_on_slave(class_method, base, method)
-      base_methods =
-        if class_method
-          base.methods + base.private_methods
-        else
-          base.instance_methods + base.private_instance_methods
-        end
-
-      return unless base_methods.include?(method)
-      _, method, punctuation = method.to_s.match(/^(.*?)([\?\!]?)$/).to_a
-      base.class_eval <<-RUBY, __FILE__, __LINE__ + 1
-        #{class_method ? 'class << self' : ''}
-          def #{method}_with_default_slave#{punctuation}(*args, &block)
-            on_slave_unless_tx do
-              #{method}_without_default_slave#{punctuation}(*args, &block)
-            end
-          end
-
-          alias_method :#{method}_without_default_slave#{punctuation}, :#{method}#{punctuation}
-          alias_method :#{method}#{punctuation}, :#{method}_with_default_slave#{punctuation}
-        #{class_method ? 'end' : ''}
-      RUBY
-    end
-
-    def columns_with_force_slave(*args, &block)
-      force_cx_switch_slave_block do
-        columns_without_force_slave(*args, &block)
-      end
-    end
-
-    def table_exists_with_force_slave?(*args, &block)
-      force_cx_switch_slave_block do
-        table_exists_without_force_slave?(*args, &block)
-      end
-    end
-
-    def transaction_with_slave_off(*args, &block)
-      if on_slave_by_default?
-        begin
-          old_val = Thread.current[:_active_record_shards_slave_off]
-          Thread.current[:_active_record_shards_slave_off] = true
-          transaction_without_slave_off(*args, &block)
-        ensure
-          Thread.current[:_active_record_shards_slave_off] = old_val
-        end
-      else
-        transaction_without_slave_off(*args, &block)
-      end
-    end
-
-    module InstanceMethods
-      # fix ActiveRecord to do the right thing, and use our aliased quote_value
-      def quote_value(*args, &block)
-        self.class.quote_value(*args, &block)
-      end
-
-      def reload_with_slave_off(*args, &block)
-        self.class.on_master { reload_without_slave_off(*args, &block) }
-      end
-    end
-
-    CLASS_SLAVE_METHODS = [:find_by_sql, :count_by_sql, :calculate, :find_one, :find_some, :find_every, :exists?].freeze
-
-    def self.extended(base)
-      CLASS_SLAVE_METHODS.each { |m| ActiveRecordShards::DefaultSlavePatches.wrap_method_in_on_slave(true, base, m) }
-
-      base.class_eval do
-        include InstanceMethods
-
-        alias_method :reload_without_slave_off, :reload
-        alias_method :reload, :reload_with_slave_off
-
-        class << self
-          alias_method :columns_without_force_slave, :columns
-          alias_method :columns, :columns_with_force_slave
-
-          alias_method :table_exists_without_force_slave?, :table_exists?
-          alias_method :table_exists?, :table_exists_with_force_slave?
-
-          alias_method :transaction_without_slave_off, :transaction
-          alias_method :transaction, :transaction_with_slave_off
-        end
-      end
-      if ActiveRecord::Associations.const_defined?(:HasAndBelongsToManyAssociation)
-        ActiveRecordShards::DefaultSlavePatches.wrap_method_in_on_slave(false, ActiveRecord::Associations::HasAndBelongsToManyAssociation, :construct_sql)
-        ActiveRecordShards::DefaultSlavePatches.wrap_method_in_on_slave(false, ActiveRecord::Associations::HasAndBelongsToManyAssociation, :construct_find_options!)
-      end
-    end
-
-    def on_slave_unless_tx
-      if on_slave_by_default? && !Thread.current[:_active_record_shards_slave_off]
-        on_slave { yield }
-      else
-        yield
-      end
-    end
-
-    module ActiveRelationPatches
-      def self.included(base)
-        [:calculate, :exists?, :pluck, :find_with_associations].each do |m|
-          ActiveRecordShards::DefaultSlavePatches.wrap_method_in_on_slave(false, base, m)
-        end
-      end
-
-      def on_slave_unless_tx
-        @klass.on_slave_unless_tx { yield }
-      end
-    end
-
-    # in rails 4.1+, they create a join class that's used to pull in records for HABTM.
-    # this simplifies the hell out of our existence, because all we have to do is inerit on-slave-by-default
-    # down from the parent now.
-    module HasAndBelongsToManyBuilderExtension
-      def self.included(base)
-        base.class_eval do
-          alias_method :through_model_without_inherit_default_slave_from_lhs, :through_model
-          alias_method :through_model, :through_model_with_inherit_default_slave_from_lhs
-        end
-      end
-
-      def through_model_with_inherit_default_slave_from_lhs
-        model = through_model_without_inherit_default_slave_from_lhs
-        def model.on_slave_by_default?
-          left_reflection.klass.on_slave_by_default?
-        end
-
-        model.extend(ActiveRecordShards::Ext::ShardedModel) if model.left_reflection.klass.is_sharded?
-
-        model
-      end
-    end
-  end
-end

data/lib/active_record_shards/no_shard_selection.rb

@@ -1,18 +0,0 @@
-# frozen_string_literal: true
-module ActiveRecordShards
-  class NoShardSelection
-    class NoShardSelected < RuntimeError; end
-
-    def shard
-      raise NoShardSelected, "Missing shard information on connection"
-    end
-
-    def connection_config
-      raise NoShardSelected, "No shard selected, can't connect"
-    end
-
-    def on_shard?
-      false
-    end
-  end
-end

data/lib/active_record_shards/sharded_model.rb

@@ -1,31 +0,0 @@
-module ActiveRecordShards
-  module Ext
-    module ShardedModel
-      def self.extended(base)
-        base.extend(ActiveRecordShards::ConnectionSwitcher)
-        base.include(InstanceMethods)
-        base.after_initialize :initialize_shard
-      end
-
-      def is_sharded? # rubocop:disable Naming/PredicateName
-        true
-      end
-
-      module InstanceMethods
-        def initialize_shard
-          @from_shard = self.class.current_shard_selection.shard
-        end
-
-        def from_shard
-          @from_shard
-        end
-      end
-    end
-  end
-
-  class ShardedModel < ActiveRecord::Base
-    self.abstract_class = true
-
-    extend ActiveRecordShards::Ext::ShardedModel
-  end
-end

data/lib/active_record_shards/slave_db.rb

@@ -1,105 +0,0 @@
-module ActiveRecordShards
-  module SlaveDb
-    def on_slave_if(condition, &block)
-      condition ? on_slave(&block) : yield
-    end
-
-    def on_slave_unless(condition, &block)
-      on_slave_if(!condition, &block)
-    end
-
-    def on_master_if(condition, &block)
-      condition ? on_master(&block) : yield
-    end
-
-    def on_master_unless(condition, &block)
-      on_master_if(!condition, &block)
-    end
-
-    def on_master_or_slave(which, &block)
-      if block_given?
-        on_cx_switch_block(which, &block)
-      else
-        MasterSlaveProxy.new(self, which)
-      end
-    end
-
-    def on_slave?
-      current_slave_selection
-    end
-
-    # Executes queries using the slave database. Fails over to master if no slave is found.
-    # if you want to execute a block of code on the slave you can go:
-    #   Account.on_slave do
-    #     Account.first
-    #   end
-    # the first account will be found on the slave DB
-    #
-    # For one-liners you can simply do
-    #   Account.on_slave.first
-    def on_slave(&block)
-      on_master_or_slave(:slave, &block)
-    end
-
-    def on_master(&block)
-      on_master_or_slave(:master, &block)
-    end
-
-    def force_cx_switch_slave_block
-      old_options = current_slave_selection
-      switch_slave_connection(slave: true)
-      yield
-    ensure
-      switch_slave_connection(slave: old_options)
-    end
-
-    def on_cx_switch_block(which, &block)
-      @disallow_slave ||= 0
-      @disallow_slave += 1 if which == :master
-
-      switch_to_slave = @disallow_slave.zero?
-      old_options = current_slave_selection
-
-      switch_slave_connection(slave: switch_to_slave)
-
-      # we avoid_readonly_scope to prevent some stack overflow problems, like when
-      # .columns calls .with_scope which calls .columns and onward, endlessly.
-      if self == ActiveRecord::Base || !switch_to_slave
-        yield
-      else
-        readonly.scoping(&block)
-      end
-    ensure
-      @disallow_slave -= 1 if which == :master
-      switch_slave_connection(slave: old_options)
-    end
-
-    def current_slave_selection=(on_slave)
-      Thread.current[:slave_selection] = on_slave
-    end
-
-    def current_slave_selection
-      !!Thread.current[:slave_selection]
-    end
-
-    def connection_config
-      super.merge(slave: current_slave_selection)
-    end
-
-    def switch_slave_connection(options)
-      self.current_slave_selection = options[:slave]
-      ensure_shard_connection
-    end
-
-    class MasterSlaveProxy
-      def initialize(target, which)
-        @target = target
-        @which = which
-      end
-
-      def method_missing(method, *args, &block) # rubocop:disable Style/MethodMissing
-        @target.on_master_or_slave(@which) { @target.send(method, *args, &block) }
-      end
-    end
-  end
-end