switchman 1.9.6 → 1.9.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b8f3e448e1f627fcd836971b4e1e866e43082987
-  data.tar.gz: 204b5b664a1349b4e74652e18fbc66281d240f6f
+  metadata.gz: 5cb122072743392c94ee7f184136a0a54710618d
+  data.tar.gz: edd7dac621e1abb34fde5f727d55943789d79076
 SHA512:
-  metadata.gz: 2ed76354d01310128a8a0003611810091bfa79b6148f2e7d58e3adef4d810fef6bbe4dc2f819cf2cfe61ac1ea91d86195e30a7da2490b8293afe406356eeaab5
-  data.tar.gz: 0f78ee14df104f97acdf4dd01dea9d823f40c6b606d916d893c0273bfec6fd1b76d97014267163f1ec748f33e7e1f1fe522b1148f45d4a4e5df5cda88bbbdddd
+  metadata.gz: a64792365604ca68c6a7aa162ab073ec8820e3aa9944bb101e681fb5bd9b25caa29a4b207b0df283274e57b853315db2979981cf5f9f08ec43d0c1f1ffc86cd6
+  data.tar.gz: b4e6287a9bc2ea138aa4de7705c87a602c7dc8646d69791e149dba5b54855235b2ddda355544e1b26fdc76f96be37cf1b2fbe9a88b7b8f0025600c6594f25089

data/lib/switchman/active_record/migration.rb

@@ -0,0 +1,13 @@
+module Switchman
+  module ActiveRecord
+    module Migration
+      def connection
+        conn = super
+        if conn.shard != ::ActiveRecord::Base.connection_pool.current_pool.shard
+          ::ActiveRecord::Base.connection_pool.current_pool.switch_database(conn)
+        end
+        conn
+      end
+    end
+  end
+end

data/lib/switchman/engine.rb

@@ -72,6 +72,7 @@ module Switchman
         require "switchman/active_record/connection_pool"
         require "switchman/active_record/finder_methods"
         require "switchman/active_record/log_subscriber"
+        require "switchman/active_record/migration"
         require "switchman/active_record/model_schema"
         require "switchman/active_record/persistence"
         require "switchman/active_record/predicate_builder"
@@ -127,6 +128,8 @@ module Switchman
         ::ActiveRecord::ConnectionAdapters::QueryCache.send(:remove_method, :select_all) if ::Rails.version < '5.0.1'
 
         ::ActiveRecord::LogSubscriber.prepend(ActiveRecord::LogSubscriber)
+        ::ActiveRecord::Migration.prepend(ActiveRecord::Migration)
+
         ::ActiveRecord::Reflection::AbstractReflection.include(ActiveRecord::Reflection::AbstractReflection)
         ::ActiveRecord::Reflection::AssociationReflection.prepend(ActiveRecord::Reflection::AssociationScopeCache)
         ::ActiveRecord::Reflection::ThroughReflection.prepend(ActiveRecord::Reflection::AssociationScopeCache)

data/lib/switchman/version.rb

@@ -1,3 +1,3 @@
 module Switchman
-  VERSION = "1.9.6"
+  VERSION = "1.9.7"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: switchman
 version: !ruby/object:Gem::Version
-  version: 1.9.6
+  version: 1.9.7
 platform: ruby
 authors:
 - Cody Cutrer
@@ -189,6 +189,7 @@ files:
 - lib/switchman/active_record/connection_pool.rb
 - lib/switchman/active_record/finder_methods.rb
 - lib/switchman/active_record/log_subscriber.rb
+- lib/switchman/active_record/migration.rb
 - lib/switchman/active_record/model_schema.rb
 - lib/switchman/active_record/persistence.rb
 - lib/switchman/active_record/postgresql_adapter.rb
@@ -242,7 +243,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.10
+rubygems_version: 2.6.11
 signing_key: 
 specification_version: 4
 summary: Rails 4 sharding magic