lib/kumogata/post_processing.rb in kumogata-0.3.12 vs lib/kumogata/post_processing.rb in kumogata-0.3.13

@@ -130,11 +130,16 @@
     args = [host, user]
     args << ssh['options'] if ssh['options']
 
     command = evaluate_command_template(command, outputs)
 
+    connect_tries = (ssh['connect_tries'] || 36).to_i
+    retry_interval = (ssh['retry_interval'] || 5).to_i
+
     begin
-      Net::SSH.start(*args) {|ssh| ssh_exec!(ssh, command) }
+      retryable(:tries => connect_tries, :on => Net::SSH::Disconnect, :sleep => retry_interval) do
+        Net::SSH.start(*args) {|ssh| ssh_exec!(ssh, command) }
+      end
     rescue Net::SSH::HostKeyMismatch => e
       e.remember_host!
       retry
     end
   end