lib/capistrano/asg/rolling/plugin.rb in capistrano-asg-rolling-0.3.0 vs lib/capistrano/asg/rolling/plugin.rb in capistrano-asg-rolling-0.4.0

@@ -26,10 +26,11 @@
           after 'deploy', 'rolling:update'
           after 'deploy:failed', 'rolling:cleanup'
 
           after 'rolling:update', 'rolling:cleanup'
           after 'rolling:create_ami', 'rolling:cleanup'
+          after 'rolling:update', 'rolling:instance_refresh_status'
 
           # Register an exit hook to do some cleanup when Capistrano
           # terminates without calling our after cleanup hook.
           at_exit { cleanup }
         end
@@ -50,9 +51,11 @@
           instances = config.instances.auto_terminate
           return if instances.empty?
 
           logger.info 'Terminating instance(s)...'
           instances.terminate
+        rescue Capistrano::ASG::Rolling::InstanceTerminateFailed => e
+          logger.warning "Failed to terminate Instance **#{e.instance.id}**: #{e.message}"
         end
       end
     end
   end
 end