github-migration
/
hydra
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added ability to run arbitrary commands on remote machines (useful for updating bundles!)

This commit is contained in:
Sean Kirby 2011-04-26 12:18:21 -04:00
parent b89a60d897
commit beae81621d
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lib/hydra/tasks.rb
View File

@ -243,8 +243,9 @@ module Hydra #:nodoc:
include Open3 include Open3
# Create a new hydra remote task with the given name. # Create a new hydra remote task with the given name.
# The task will be named hydra:remote:<name> # The task will be named hydra:remote:<name>
def initialize(name) def initialize(name, command=nil)
@name = name @name = name
@command = command
yield self if block_given? yield self if block_given?
@config = find_config_file @config = find_config_file
if @config if @config
@ -262,6 +263,7 @@ module Hydra #:nodoc:
environment = config.fetch('environment') { 'test' } environment = config.fetch('environment') { 'test' }
workers = config.fetch('workers') { [] } workers = config.fetch('workers') { [] }
workers = workers.select{|w| w['type'] == 'ssh'} workers = workers.select{|w| w['type'] == 'ssh'}
@command = "RAILS_ENV=#{environment} rake #{@name}" unless @command
$stdout.write "==== Hydra Running #{@name} ====\n" $stdout.write "==== Hydra Running #{@name} ====\n"
Thread.abort_on_exception = true Thread.abort_on_exception = true
@ -270,7 +272,7 @@ module Hydra #:nodoc:
workers.each do |worker| workers.each do |worker|
@listeners << Thread.new do @listeners << Thread.new do
begin begin
@results[worker] = if run_task(worker, environment) @results[worker] = if run_command(worker, @command)
"==== #{@name} passed on #{worker['connect']} ====\n" "==== #{@name} passed on #{worker['connect']} ====\n"
else else
"==== #{@name} failed on #{worker['connect']} ====\nPlease see above for more details.\n" "==== #{@name} failed on #{worker['connect']} ====\nPlease see above for more details.\n"
@ -286,13 +288,13 @@ module Hydra #:nodoc:
end end
end end
def run_task worker, environment def run_command worker, command
$stdout.write "==== Hydra Running #{@name} on #{worker['connect']} ====\n" $stdout.write "==== Hydra Running #{@name} on #{worker['connect']} ====\n"
ssh_opts = worker.fetch('ssh_opts') { '' } ssh_opts = worker.fetch('ssh_opts') { '' }
writer, reader, error = popen3("ssh -tt #{ssh_opts} #{worker['connect']} ") writer, reader, error = popen3("ssh -tt #{ssh_opts} #{worker['connect']} ")
writer.write("cd #{worker['directory']}\n") writer.write("cd #{worker['directory']}\n")
writer.write "echo BEGIN HYDRA\n" writer.write "echo BEGIN HYDRA\n"
writer.write("RAILS_ENV=#{environment} rake #{@name}\n") writer.write(command + "\r")
writer.write "echo END HYDRA\n" writer.write "echo END HYDRA\n"
writer.write("exit\n") writer.write("exit\n")
writer.close writer.close