activerecord-turntable 4.0.0 → 4.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA256:
-  metadata.gz: 5bd7f48754157e32eaaf254a518645b9d1cde4e31840dba63b927886bedb74ee
-  data.tar.gz: 529f08173f847218298a34ed9832163e1d129b003e8ba0ef201ca63d8333b029
+SHA1:
+  metadata.gz: 31aa02feae321af087a80d9a1f9177c0ff7c139a
+  data.tar.gz: dbb3b7404067e3a2b756b96eaa5f76a51e27450d
 SHA512:
-  metadata.gz: 515cc5103600046d62ea3b91a5695787a2f34d423da084e15aa3dbef8accdba238abd4bb4fb4c4e2659cdbe9ac965f020afb2d6f75a84c4e097cff28e2f1d6cb
-  data.tar.gz: '0578e2aff069a944a6b512b04ae19d47fe2591813f9666f7324cb7d592a4febaa61b457dd71b75dfe8b592f1666bdf57880a3f552da0702c1362b64126db401c'
+  metadata.gz: 3bb4919921852779beb4b1d31f67f524e5985921c5d00d613799417a769c80eeaf3eec5cd51deaa5ea061cdc0869b8295b785806a5febb2ca7979924482ac79b
+  data.tar.gz: e5f313cc055e68af9520db17c29edcb717b8459b3ccb65819fe9d17e08010937a938277fc2baafa469b4fb8555a7f8f88a1344cf89064eff3a6d259e2cd6fe12

data/.travis.yml

@@ -5,10 +5,10 @@ cache:
   bundler: true
 
 rvm:
-  - 2.2.9
-  - 2.3.6
-  - 2.4.3
-  - 2.5.0
+  - 2.2.10
+  - 2.3.7
+  - 2.4.4
+  - 2.5.1
   - ruby-head
 
 gemfile:
@@ -18,12 +18,15 @@ gemfile:
   - gemfiles/rails5_0_3.gemfile
   - gemfiles/rails5_0_4.gemfile
   - gemfiles/rails5_0_5.gemfile
+  - gemfiles/rails5_0_6.gemfile
+  - gemfiles/rails5_0_7.gemfile
   - gemfiles/rails5_1_0.gemfile
   - gemfiles/rails5_1_1.gemfile
   - gemfiles/rails5_1_2.gemfile
   - gemfiles/rails5_1_3.gemfile
   - gemfiles/rails5_1_4.gemfile
   - gemfiles/rails5_1_5.gemfile
+  - gemfiles/rails5_1_6.gemfile
   - gemfiles/rails_edge.gemfile
 
 env:
@@ -37,6 +40,7 @@ services:
 before_install:
   - docker run -d --name turntable-katsubushi -p 11212:11212 katsubushi/katsubushi:latest -worker-id 1
   - gem update --system
+  - gem install bundler --no-doc
 
 before_script:
   - bundle exec rake $SETUP_TASK
@@ -48,14 +52,19 @@ matrix:
   exclude:
     - rvm: ruby-head
     - gemfile: gemfiles/rails_edge.gemfile
-    - rvm: 2.3.6
-    - rvm: 2.4.3
+    - rvm: 2.2.10
+    - rvm: 2.3.7
+    - rvm: 2.4.4
   include:
-    - rvm: 2.3.6
-      gemfile: gemfiles/rails5_1_5.gemfile
-    - rvm: 2.4.3
-      gemfile: gemfiles/rails5_1_5.gemfile
-    - rvm: 2.5.0
+    - rvm: 2.2.10
+      gemfile: gemfiles/rails5_0_7.gemfile
+    - rvm: 2.2.10
+      gemfile: gemfiles/rails5_1_6.gemfile
+    - rvm: 2.3.7
+      gemfile: gemfiles/rails5_1_6.gemfile
+    - rvm: 2.4.4
+      gemfile: gemfiles/rails5_1_6.gemfile
+    - rvm: 2.5.1
       gemfile: gemfiles/rails_edge.gemfile
     - rvm: ruby-head
       gemfile: gemfiles/rails_edge.gemfile

data/CHANGELOG.md

@@ -1,3 +1,15 @@
+## activerecord-turntable 4.1.0 ##
+
+### Major Changes
+
+* Support activerecord versions v5.0.6, v5.0.7 and v5.1.6
+* Remove unneeded monkey patches
+* Add `ActiveRecord::Base.turntable_pool_list` instead of `ActiveRecord::Base.turntable_connections`
+
+### BugFixes
+
+* Fix connection leaks on fork
+
 ## activerecord-turntable 4.0.0 ##
 
 ### Major Changes

data/Gemfile

@@ -2,7 +2,5 @@ source "https://rubygems.org"
 
 gemspec
 
-gem "activerecord", "~> 5.1.1"
-gem "activesupport", "~> 5.1.1"
-
-gem "mysql2", "~> 0.4.4"
+gem "activerecord", "~> 5.1.6"
+gem "activesupport", "~> 5.1.6"

data/README.md

@@ -27,7 +27,7 @@ MySQL only.
 Add to Gemfile:
 
 ```ruby
-gem 'activerecord-turntable', '~> 4.0.0'
+gem 'activerecord-turntable', '~> 4.1.0'
 ```
 
 Run a bundle install:

data/gemfiles/rails5_0_6.gemfile

@@ -0,0 +1,8 @@
+source "https://rubygems.org"
+
+gem "activerecord", "5.0.6"
+gem "activesupport", "5.0.6"
+
+gem "mysql2", "~> 0.4.4"
+
+gemspec :path => '../'

data/gemfiles/rails5_0_7.gemfile

@@ -0,0 +1,6 @@
+source "https://rubygems.org"
+
+gem "activerecord", "5.0.7"
+gem "activesupport", "5.0.7"
+
+gemspec :path => '../'

data/gemfiles/rails5_1_6.gemfile

@@ -0,0 +1,7 @@
+source "https://rubygems.org"
+
+gem "activerecord", "5.1.6"
+gem "activesupport", "5.1.6"
+gem "actionview", "5.1.6"
+
+gemspec :path => '../'

data/lib/active_record/turntable/active_record_ext.rb

@@ -27,7 +27,7 @@ module ActiveRecord::Turntable
       ActiveRecord::Persistence.include(Persistence)
       ActiveRecord::Locking::Optimistic.include(LockingOptimistic)
       ActiveRecord::Relation.include(CleverLoad)
-      ActiveRecord::Relation.prepend(Relation)
+      ActiveRecord::Relation.prepend(Relation) unless Util.ar_version_equals_or_later?("5.1.6")
       ActiveRecord::Migration.include(ActiveRecord::Turntable::Migration)
       ActiveRecord::ConnectionAdapters::ConnectionHandler.prepend(ConnectionHandlerExtension)
       ActiveRecord::Associations::Preloader::Association.prepend(AssociationPreloader)

data/lib/active_record/turntable/active_record_ext/fixtures.rb

@@ -133,10 +133,5 @@ module ActiveRecord
       instantiate_fixtures if use_instantiated_fixtures
     end
     # rubocop:enable Style/ClassVars, Style/RedundantException
-
-    def enlist_fixture_connections
-      ActiveRecord::Base.connection_handler.connection_pool_list.map(&:connection) +
-        ActiveRecord::Base.turntable_connections.values.map(&:connection)
-    end
   end
 end

data/lib/active_record/turntable/active_record_ext/locking_optimistic.rb

@@ -1,7 +1,47 @@
 module ActiveRecord::Turntable
   module ActiveRecordExt
     module LockingOptimistic
-      if Util.ar51_or_later?
+      if Util.ar_version_equals_or_later?("5.1.6")
+        ::ActiveRecord::Locking::Optimistic.class_eval <<-EOD
+          private
+          def _update_row(attribute_names, attempted_action = "update")
+            return super unless locking_enabled?
+
+            begin
+              locking_column = self.class.locking_column
+              previous_lock_value = read_attribute_before_type_cast(locking_column)
+              attribute_names << locking_column
+
+              self[locking_column] += 1
+
+              constraints = {
+                self.class.primary_key => id_in_database,
+                locking_column => previous_lock_value
+              }
+              if self.class.sharding_condition_needed?
+                constraints[self.class.turntable_shard_key] = self[self.class.turntable_shard_key]
+              end
+
+              affected_rows = self.class.unscoped._update_record(
+                arel_attributes_with_values(attribute_names),
+                constraints,
+              )
+
+              if affected_rows != 1
+                raise ActiveRecord::StaleObjectError.new(self, attempted_action)
+              end
+
+              affected_rows
+
+            # If something went wrong, revert the locking_column value.
+            rescue Exception
+              self[locking_column] = previous_lock_value.to_i
+              raise
+            end
+          end
+        EOD
+
+      elsif Util.ar51_or_later?
         ::ActiveRecord::Locking::Optimistic.class_eval <<-EOD
           private
           # @note Override to add sharding condition on optimistic locking

data/lib/active_record/turntable/active_record_ext/log_subscriber.rb

@@ -20,7 +20,7 @@ module ActiveRecord::Turntable
 
         unless (payload[:binds] || []).empty?
           if Util.ar_version_equals_or_later?("5.0.3")
-            casted_params = if Util.ar_version_equals_or_later?("5.1.5")
+            casted_params = if Util.ar_version_satisfy?(">= 5.1.5") || Util.ar_version_satisfy?([">= 5.0.7", "< 5.1"])
                               type_casted_binds(payload[:type_casted_binds])
                             else
                               type_casted_binds(payload[:binds], payload[:type_casted_binds])

data/lib/active_record/turntable/active_record_ext/persistence.rb

@@ -26,54 +26,56 @@ module ActiveRecord::Turntable
           self
         end
 
-        # @note Override to add sharding scope on `touch`
-        # rubocop:disable Style/UnlessElse
-        def touch(*names, time: nil)
-          unless persisted?
-            raise ActiveRecord::ActiveRecordError, <<-MSG.squish
-              cannot touch on a new or destroyed record object. Consider using
-              persisted?, new_record?, or destroyed? before touching
-            MSG
-          end
+        unless Util.ar_version_equals_or_later?("5.1.6")
+          # @note Override to add sharding scope on `touch`
+          # rubocop:disable Style/UnlessElse
+          def touch(*names, time: nil)
+            unless persisted?
+              raise ActiveRecord::ActiveRecordError, <<-MSG.squish
+                cannot touch on a new or destroyed record object. Consider using
+                persisted?, new_record?, or destroyed? before touching
+              MSG
+            end
 
-          time ||= current_time_from_proper_timezone
-          attributes = timestamp_attributes_for_update_in_model
-          attributes.concat(names)
+            time ||= current_time_from_proper_timezone
+            attributes = timestamp_attributes_for_update_in_model
+            attributes.concat(names)
 
-          unless attributes.empty?
-            changes = {}
+            unless attributes.empty?
+              changes = {}
 
-            attributes.each do |column|
-              column = column.to_s
-              changes[column] = write_attribute(column, time)
-            end
+              attributes.each do |column|
+                column = column.to_s
+                changes[column] = write_attribute(column, time)
+              end
 
-            clear_attribute_changes(changes.keys) unless Util.ar51_or_later?
-            primary_key = self.class.primary_key
-            scope = if turntable_enabled? && primary_key != self.class.turntable_shard_key.to_s
-                      self.class.unscoped.where(self.class.turntable_shard_key => _read_attribute(turntable_shard_key))
-                    else
-                      self.class.unscoped
-                    end
-            scope = scope.where(primary_key => _read_attribute(primary_key))
-
-            if locking_enabled?
-              locking_column = self.class.locking_column
-              scope = scope.where(locking_column => _read_attribute(locking_column))
-              changes[locking_column] = increment_lock
-            end
+              clear_attribute_changes(changes.keys) unless Util.ar51_or_later?
+              primary_key = self.class.primary_key
+              scope = if turntable_enabled? && primary_key != self.class.turntable_shard_key.to_s
+                        self.class.unscoped.where(self.class.turntable_shard_key => _read_attribute(turntable_shard_key))
+                      else
+                        self.class.unscoped
+                      end
+              scope = scope.where(primary_key => _read_attribute(primary_key))
 
-            clear_attribute_changes(changes.keys) if Util.ar51_or_later?
-            result = scope.update_all(changes) == 1
+              if locking_enabled?
+                locking_column = self.class.locking_column
+                scope = scope.where(locking_column => _read_attribute(locking_column))
+                changes[locking_column] = increment_lock
+              end
 
-            if !result && locking_enabled?
-              raise ActiveRecord::StaleObjectError.new(self, "touch")
-            end
+              clear_attribute_changes(changes.keys) if Util.ar51_or_later?
+              result = scope.update_all(changes) == 1
 
-            @_trigger_update_callback = result
-            result
-          else
-            true
+              if !result && locking_enabled?
+                raise ActiveRecord::StaleObjectError.new(self, "touch")
+              end
+
+              @_trigger_update_callback = result
+              result
+            else
+              true
+            end
           end
         end
         # rubocop:enable Style/UnlessElse
@@ -115,25 +117,39 @@ module ActiveRecord::Turntable
             relation
           end
 
-          # @note Override to add sharding scope on updating
-          def _update_record(attribute_names = self.attribute_names)
-            klass = self.class
-            attributes_values = arel_attributes_with_values_for_update(attribute_names)
-            if attributes_values.empty?
-              rows_affected = 0
-              @_trigger_update_callback = true
-            else
-              scope = if klass.turntable_enabled? && (klass.primary_key != klass.turntable_shard_key.to_s)
-                        klass.unscoped.where(klass.turntable_shard_key => self.send(turntable_shard_key))
-                      end
-              previous_id = Util.ar51_or_later? ? id_in_database : id_was
-              rows_affected = klass.unscoped._update_record attributes_values, id, previous_id, scope
-              @_trigger_update_callback = rows_affected > 0
+          if Util.ar_version_equals_or_later?("5.1.6")
+            def _update_row(attribute_names, attempted_action = "update")
+              constraints = { self.class.primary_key => id_in_database }
+              if self.class.sharding_condition_needed?
+                constraints[self.class.turntable_shard_key] = self[self.class.turntable_shard_key]
+              end
+
+              self.class.unscoped._update_record(
+                arel_attributes_with_values(attribute_names),
+                constraints,
+              )
+            end
+          else
+            # @note Override to add sharding scope on updating
+            def _update_record(attribute_names = self.attribute_names)
+              klass = self.class
+              attributes_values = arel_attributes_with_values_for_update(attribute_names)
+              if attributes_values.empty?
+                rows_affected = 0
+                @_trigger_update_callback = true
+              else
+                scope = if klass.turntable_enabled? && (klass.primary_key != klass.turntable_shard_key.to_s)
+                          klass.unscoped.where(klass.turntable_shard_key => self.send(turntable_shard_key))
+                        end
+                previous_id = Util.ar51_or_later? ? id_in_database : id_was
+                rows_affected = klass.unscoped._update_record attributes_values, id, previous_id, scope
+                @_trigger_update_callback = rows_affected > 0
+              end
+
+              yield(self) if block_given?
+
+              rows_affected
             end
-
-            yield(self) if block_given?
-
-            rows_affected
           end
       end
     end

data/lib/active_record/turntable/active_record_ext/query_cache.rb

@@ -14,7 +14,7 @@ module ActiveRecord::Turntable
           def run
             result = super
 
-            pools = ActiveRecord::Base.turntable_connections.values
+            pools = ActiveRecord::Base.turntable_pool_list
             pools.each do |pool|
               pool.enable_query_cache!
             end
@@ -36,7 +36,7 @@ module ActiveRecord::Turntable
           def run
             result = super
 
-            pools = ActiveRecord::Base.turntable_connections.values
+            pools = ActiveRecord::Base.turntable_pool_list
             pools.each do |pool|
               pool.enable_query_cache!
             end
@@ -58,7 +58,7 @@ module ActiveRecord::Turntable
           def run
             result = super
 
-            pools = ActiveRecord::Base.turntable_connections.values
+            pools = ActiveRecord::Base.turntable_pool_list
             pools.each do |k|
               k.connection.enable_query_cache!
             end
@@ -70,7 +70,7 @@ module ActiveRecord::Turntable
             enabled, _connection_id = state
             super
 
-            klasses = ActiveRecord::Base.turntable_connections.values
+            klasses = ActiveRecord::Base.turntable_pool_list
             klasses.each do |k|
               k.connection.clear_query_cache
               k.connection.disable_query_cache! unless enabled

data/lib/active_record/turntable/base.rb

@@ -5,10 +5,9 @@ module ActiveRecord::Turntable
     extend ActiveSupport::Concern
 
     included do
-      class_attribute :turntable_connections, :turntable_clusters, :turntable_sequencers,
+      class_attribute :turntable_clusters, :turntable_sequencers,
                       :turntable_enabled, :turntable_sequencer_enabled, :turntable_configuration
 
-      self.turntable_connections = {}
       self.turntable_clusters = {}.with_indifferent_access
       self.turntable_sequencers = {}.with_indifferent_access
       self.turntable_enabled = false
@@ -52,6 +51,10 @@ module ActiveRecord::Turntable
         turntable_clusters[turntable_cluster_name]
       end
 
+      def turntable_pool_list
+        turntable_clusters.values.map { |cluster| cluster.shards.map(&:connection_pool) }.flatten.uniq
+      end
+
       def turntable_replace_connection_pool
         ch = connection_handler
         cp = ConnectionProxy.new(self, turntable_cluster)
@@ -61,10 +64,6 @@ module ActiveRecord::Turntable
         ch.send(:owner_to_pool)[connection_specification_name] = pp
       end
 
-      def clear_all_connections!
-        turntable_connections.values.each(&:disconnect!)
-      end
-
       def sequencer(sequencer_name, *args)
         class_attribute :turntable_sequencer_name
         class << self
@@ -83,6 +82,10 @@ module ActiveRecord::Turntable
         turntable_enabled
       end
 
+      def sharding_condition_needed?
+        turntable_enabled? && primary_key != turntable_shard_key.to_s
+      end
+
       def sequencer_enabled?
         turntable_sequencer_enabled
       end

data/lib/active_record/turntable/cluster_helper_methods.rb

@@ -13,9 +13,9 @@ module ActiveRecord::Turntable
     module ClassMethods
       def force_transaction_all_shards!(options = {}, &block)
         force_connect_all_shards!
-        shards = turntable_connections.values
-        shards += [ActiveRecord::Base.connection_pool]
-        recursive_transaction(shards, options, &block)
+        pools = turntable_pool_list
+        pools += [ActiveRecord::Base.connection_pool]
+        recursive_transaction(pools, options, &block)
       end
 
       def recursive_transaction(pools, options, &block)
@@ -30,16 +30,7 @@ module ActiveRecord::Turntable
       end
 
       def force_connect_all_shards!
-        conf = configurations[::ActiveRecord::ConnectionHandling::DEFAULT_ENV.call.to_sym]
-        return unless conf
-
-        shards = HashWithIndifferentAccess.new
-        shards = shards.merge(conf[:shards]) if conf[:shards]
-        shards = shards.merge(conf[:seq]) if conf[:seq]
-        shards.each do |name, config|
-          turntable_connections[name] ||=
-            ActiveRecord::ConnectionAdapters::ConnectionPool.new(spec_for(config))
-        end
+        turntable_pool_list.each(&:connection)
       end
 
       def spec_for(config)

data/lib/active_record/turntable/connection_proxy.rb

@@ -59,16 +59,16 @@ module ActiveRecord::Turntable
     def enable_query_cache!
       default_connection.enable_query_cache!
 
-      klass.turntable_connections.each do |_k, v|
-        v.connection.enable_query_cache!
+      cluster.shards.each do |shard|
+        shard.connection.enable_query_cache!
       end
     end
 
     def disable_query_cache!
       default_connection.disable_query_cache!
 
-      klass.turntable_connections.each do |_k, v|
-        v.connection.disable_query_cache!
+      cluster.shards.each do |shard|
+        shard.connection.disable_query_cache!
       end
     end
 
@@ -79,12 +79,11 @@ module ActiveRecord::Turntable
     def clear_query_cache
       default_connection.clear_query_cache
 
-      klass.turntable_connections.each do |_k, v|
-        v.connection.clear_query_cache
+      cluster.shards.each do |shard|
+        shard.connection.clear_query_cache
       end
     end
 
-
     # rubocop:disable Style/MethodMissing
     def method_missing(method, *args, &block)
       clear_query_cache_if_needed(method)

data/lib/active_record/turntable/seq_shard.rb

@@ -10,10 +10,15 @@ module ActiveRecord::Turntable
 
     private
 
-      def create_connection_class
-        klass = connection_class_instance
-        klass.remove_connection
-        klass.establish_connection ActiveRecord::Base.connection_pool.spec.config[:seq][name].with_indifferent_access
+      def connection_class_instance
+        if Connections.const_defined?(name.classify)
+          klass = Connections.const_get(name.classify)
+        else
+          klass = Class.new(ActiveRecord::Base)
+          Connections.const_set(name.classify, klass)
+          klass.abstract_class = true
+          klass.establish_connection ActiveRecord::Base.connection_pool.spec.config[:seq][name].with_indifferent_access
+        end
         klass
       end
   end

data/lib/active_record/turntable/shard.rb

@@ -11,7 +11,6 @@ module ActiveRecord::Turntable
       @cluster = cluster
       @name = name
       @slaves = slaves.map { |s| SlaveShard.new(cluster, s) }
-      ActiveRecord::Base.turntable_connections[name] = connection_pool
     end
 
     def connection_pool
@@ -43,14 +42,7 @@ module ActiveRecord::Turntable
     private
 
       def connection_klass
-        @connection_klass ||= create_connection_class
-      end
-
-      def create_connection_class
-        klass = connection_class_instance
-        klass.remove_connection
-        klass.establish_connection ActiveRecord::Base.connection_pool.spec.config[:shards][name].with_indifferent_access
-        klass
+        @connection_klass ||= connection_class_instance
       end
 
       def connection_class_instance
@@ -60,6 +52,7 @@ module ActiveRecord::Turntable
           klass = Class.new(ActiveRecord::Base)
           Connections.const_set(name.classify, klass)
           klass.abstract_class = true
+          klass.establish_connection ActiveRecord::Base.connection_pool.spec.config[:shards][name].with_indifferent_access
         end
         klass
       end

data/lib/active_record/turntable/util.rb

@@ -12,6 +12,13 @@ module ActiveRecord::Turntable
       ActiveRecord.gem_version.release
     end
 
+    def ar_version_satisfy?(requirement)
+      unless requirement.is_a?(Gem::Requirement)
+        requirement = Gem::Requirement.new(requirement)
+      end
+      requirement.satisfied_by?(ar_version)
+    end
+
     def ar51_or_later?
       ar_version_equals_or_later?("5.1")
     end
@@ -19,6 +26,7 @@ module ActiveRecord::Turntable
     module_function :ar_version_equals_or_later?,
                     :ar_version_earlier_than?,
                     :ar_version,
+                    :ar_version_satisfy?,
                     :ar51_or_later?
   end
 end

data/lib/active_record/turntable/version.rb

@@ -1,5 +1,5 @@
 module ActiveRecord
   module Turntable
-    VERSION = "4.0.0".freeze
+    VERSION = "4.1.0".freeze
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-turntable
 version: !ruby/object:Gem::Version
-  version: 4.0.0
+  version: 4.1.0
 platform: ruby
 authors:
 - gussan
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-04-03 00:00:00.000000000 Z
+date: 2018-04-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -529,12 +529,15 @@ files:
 - gemfiles/rails5_0_3.gemfile
 - gemfiles/rails5_0_4.gemfile
 - gemfiles/rails5_0_5.gemfile
+- gemfiles/rails5_0_6.gemfile
+- gemfiles/rails5_0_7.gemfile
 - gemfiles/rails5_1_0.gemfile
 - gemfiles/rails5_1_1.gemfile
 - gemfiles/rails5_1_2.gemfile
 - gemfiles/rails5_1_3.gemfile
 - gemfiles/rails5_1_4.gemfile
 - gemfiles/rails5_1_5.gemfile
+- gemfiles/rails5_1_6.gemfile
 - gemfiles/rails_edge.gemfile
 - lib/active_record/turntable.rb
 - lib/active_record/turntable/active_record_ext.rb
@@ -632,7 +635,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: activerecord-turntable
-rubygems_version: 2.7.6
+rubygems_version: 2.6.14
 signing_key: 
 specification_version: 4
 summary: ActiveRecord sharding extension