switchman 1.5.21 → 1.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a6c8d63c04c23a59c2cda990cbed599098887d5e
-  data.tar.gz: 421b3c6dad49777eb5b032d82a579bc0a2802d89
+  metadata.gz: 5216239c0a0c3b84a975c7f5a7c63250a36c3960
+  data.tar.gz: 9eff38ae0cdfe742c47d25667dfa9bd5f807ad44
 SHA512:
-  metadata.gz: d428c6d5e5af925e71f8517c99ec3346a769360082114b8e3590c3459e9e104b7db6431ec41f5703d87fb9dc8b43ce64b51ac59629c0365a3afe6701a26b5aff
-  data.tar.gz: 24c0021a45980b354394c1c3f3138bf54bb7a236c18f903bb83143cb34c3b16527c55823e4fd24ca608c588d88562256d3840167ad82636b686b60b8744d0f3b
+  metadata.gz: 1a8afa7d5bbedf8bf89d797cab7eab0285a36d01d7f1efada6fb6e7aabb459ee44abfb3f82bc9e043399cf943a95bb1bee6b2f932ea92e65b97d77d7c2510801
+  data.tar.gz: fa08be1de119479e4e9a0ad3ac6f5c6a6a18f8c3400d2d784bf30be78aaff3665875eafb2a621bd2e3a4f6181affe95a49a74b6d73e55244c0810c7409b318da

data/lib/switchman/version.rb

@@ -1,3 +1,3 @@
 module Switchman
-  VERSION = "1.5.21"
+  VERSION = "1.6.0"
 end

data/lib/tasks/switchman.rake

@@ -44,7 +44,11 @@ module Switchman
       { parallel: ENV['PARALLEL'].to_i, max_procs: ENV['MAX_PARALLEL_PROCS'] }
     end
 
-    def self.shardify_task(task_name)
+    # categories - an array or proc, to activate as the current shard during the
+    # task. tasks which modify the schema may want to pass all categories in
+    # so that schema updates for non-default tables happen against all shards.
+    # this is handled automatically for the default migration tasks, below.
+    def self.shardify_task(task_name, categories: [:default])
       old_task = ::Rake::Task[task_name]
       old_actions = old_task.actions.dup
       old_task.actions.clear
@@ -57,7 +61,8 @@ module Switchman
 
         ::Shackles.activate(:deploy) do
           begin
-            Shard.with_each_shard(scope, Shard.categories, options) do
+            categories = categories.call if categories.respond_to?(:call)
+            Shard.with_each_shard(scope, categories, options) do
               shard = Shard.current
               puts "#{shard.id}: #{shard.description}"
               ::ActiveRecord::Base.connection_pool.spec.config[:shard_name] = Shard.current.name
@@ -75,7 +80,9 @@ module Switchman
       end
     end
 
-    %w{db:migrate db:migrate:up db:migrate:down db:rollback}.each { |task_name| shardify_task(task_name) }
+    %w{db:migrate db:migrate:up db:migrate:down db:rollback}.each do |task_name|
+      shardify_task(task_name, categories: ->{ Shard.categories })
+    end
 
     private
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: switchman
 version: !ruby/object:Gem::Version
-  version: 1.5.21
+  version: 1.6.0
 platform: ruby
 authors:
 - Cody Cutrer
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-09-09 00:00:00.000000000 Z
+date: 2016-09-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: railties
@@ -226,7 +226,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.4
+rubygems_version: 2.4.5.1
 signing_key: 
 specification_version: 4
 summary: Rails 4 sharding magic