Merge pull request #508 from kevinjcoleman/feature/updating-octopus-v…

…ersions only migrate specified shards.
thiagopradi · Oct 21, 2018 · b6cedf8 · b6cedf8
2 parents 28aac38 + 906a4f1
commit b6cedf8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/octopus/migration.rb b/lib/octopus/migration.rb
@@ -109,8 +109,8 @@ def run_with_octopus(&block)
 
       def migrate_with_octopus(&block)
         return migrate_without_octopus(&block) unless connection.is_a?(Octopus::Proxy)
-
-        connection.send_queries_to_multiple_shards(connection.shard_names) do
+        shards = migrations.map(&:shards).flatten.map(&:to_s)
+        connection.send_queries_to_multiple_shards(shards) do
           migrate_without_octopus(&block)
         end
         rescue ActiveRecord::UnknownMigrationVersionError => e