lib/capistrano/tasks/migrations.rake in capistrano-rails-1.5.0 vs lib/capistrano/tasks/migrations.rake in capistrano-rails-1.6.0

@@ -20,11 +20,11 @@
   desc 'Runs rake db:migrate'
   task migrating: [:set_rails_env] do
     on fetch(:migration_servers) do
       within release_path do
         with rails_env: fetch(:rails_env) do
-          execute :rake, 'db:migrate'
+          execute :rake, fetch(:migration_command)
         end
       end
     end
   end
 
@@ -34,7 +34,8 @@
 namespace :load do
   task :defaults do
     set :conditionally_migrate, fetch(:conditionally_migrate, false)
     set :migration_role, fetch(:migration_role, :db)
     set :migration_servers, -> { primary(fetch(:migration_role)) }
+    set :migration_command, -> { fetch(:migration_command, 'db:migrate') }
   end
 end