switchman 3.3.5 → 3.3.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b02846fe7319e31d24a50618d5b2f7120109e09c4a4dc40125a25f7299207bf0
-  data.tar.gz: bd42f42242e7f0b8cc88ca76206625607ad00e33d58247ac6bd9d090f7c3722a
+  metadata.gz: 07ef0eeff511916fa7913f453f2b3cde7d5317dfee9e03b58cf3645b2830c845
+  data.tar.gz: 2c2c12374387e6856674f14cf965087a0d27c1ccdacc6470b69aec0eb77af22f
 SHA512:
-  metadata.gz: 4109620cb05413f9ebfc56290e74e73d94682439a62cd1813b93ffb2b28aff2ec84aafb5bb888a283e418c8fcbe743b384bc09d3cd18f2d5008920e2707d491f
-  data.tar.gz: c5ce42651aed4080935281b4a131d56517bf84dc78c2dc38f0ce3ec6db62387b0848120392eb2187e8f86a2b61557c0a1ad34e1482cbed2f9b3f4cfbf90bfd22
+  metadata.gz: d62c36fe017fbf801312eb13295e19fbb66c5e9b2aa23996400c3654acca130525580728602328377f4b601d5a9ba09847ef6082948f29af7d1217d7471ffa70
+  data.tar.gz: b6afb2bd1e0f69d93dd7b1e7a739cd7a1b6896eee43be063f8452d102de21991f8ab49da3fecc65859e24d66dd54e19f49c6352e22a50dcbd1add00518f154c0

data/lib/switchman/active_record/attribute_methods.rb

@@ -251,7 +251,7 @@ module Switchman
 
       def local_attribute(attr_name)
         if self.class.sharded_column?(attr_name)
-          ::Switchman::Shard.local_id_for(attribute(attr_name), shard).first
+          ::Switchman::Shard.local_id_for(attribute(attr_name)).first
         else
           attribute(attr_name)
         end

data/lib/switchman/active_record/migration.rb

@@ -44,6 +44,23 @@ module Switchman
     end
 
     module MigrationContext
+      def migrate(...)
+        connection = ::ActiveRecord::Base.connection
+        connection_pool = ::ActiveRecord::Base.connection_pool
+        previous_schema_cache = connection_pool.get_schema_cache(connection)
+        temporary_schema_cache = ::ActiveRecord::ConnectionAdapters::SchemaCache.new(connection)
+
+        reset_column_information
+        connection_pool.set_schema_cache(temporary_schema_cache)
+
+        begin
+          super(...)
+        ensure
+          connection_pool.set_schema_cache(previous_schema_cache)
+          reset_column_information
+        end
+      end
+
       def migrations
         return @migrations if instance_variable_defined?(:@migrations)
 
@@ -51,6 +68,12 @@ module Switchman
         key = Digest::MD5.hexdigest(migration_files.sort.join(','))
         @migrations = migrations_cache[key] ||= super
       end
+
+      private
+
+      def reset_column_information
+        ::ActiveRecord::Base.descendants.reject { |m| m <= UnshardedRecord }.each(&:reset_column_information)
+      end
     end
   end
 end

data/lib/switchman/database_server.rb

@@ -221,12 +221,10 @@ module Switchman
 
           unless schema == false
             shard.activate do
-              reset_column_information
-
               ::ActiveRecord::Base.connection.transaction(requires_new: true) do
                 ::ActiveRecord::Base.connection.migration_context.migrate
               end
-              reset_column_information
+
               ::ActiveRecord::Base.descendants.reject do |m|
                 m <= UnshardedRecord || !m.table_exists?
               end.each(&:define_attribute_methods)
@@ -240,7 +238,6 @@ module Switchman
       rescue
         shard&.destroy
         shard&.drop_database rescue nil unless schema_already_existed
-        reset_column_information unless schema == false rescue nil
         raise
       ensure
         self.class.creating_new_shard = false
@@ -272,11 +269,5 @@ module Switchman
       end
       @primary_shard
     end
-
-    private
-
-    def reset_column_information
-      ::ActiveRecord::Base.descendants.reject { |m| m <= UnshardedRecord }.each(&:reset_column_information)
-    end
   end
 end

data/lib/switchman/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Switchman
-  VERSION = '3.3.5'
+  VERSION = '3.3.7'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: switchman
 version: !ruby/object:Gem::Version
-  version: 3.3.5
+  version: 3.3.7
 platform: ruby
 authors:
 - Cody Cutrer
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-03-21 00:00:00.000000000 Z
+date: 2023-03-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord