remote_run 0.1.3 → 0.1.4

data/Readme.md

@@ -12,6 +12,7 @@ machines until all tasks are complete.
 
         require 'rubygems'
         require 'remote_run'
+        require 'benchmark'
 
         hosts = ["broadway", "wall"]
         setup = "source ~/.profile; rvm use ree; bundle install;"
@@ -24,6 +25,15 @@ machines until all tasks are complete.
         runner = Runner.new do |config|
           config.hosts = hosts
           config.tasks = tasks
+          config.before_task = lambda do |host, task, remote_runner|
+            puts "#{host.hostname} is running '#{task.command}' "
+          end
+          config.around_task = lambda do |&block|
+            time = Benchmark.measure do
+              block.call
+            end
+            puts time
+          end
         end
 
         # kick off the run
@@ -43,6 +53,12 @@ machines until all tasks are complete.
         exclude - directories to exclude when rsyncing to remote host (default: [])
         login_as - the user used to log into ssh (default: current user)
 
+        Callbacks (optional):
+        before_run and after_run - the code to be executed before and after the run respectively, receives the remote_run instance as a paramater
+        before_task and after_task - the code to be executed before and after each task respectively, receives host, task and the remote_run instance as paramaters
+        around_run - the code to execute around the run, receives the block to call
+        around_task - the code to execute around each task, receives the block to call
+
 
 ## Accessible Attributes:
 
@@ -70,7 +86,7 @@ machines until all tasks are complete.
 
 Dependencies:
 ----------------------------------------------------------------------
-* HighLine 
+* HighLine
 
 
 License:

data/lib/remote_run/configuration.rb

@@ -1,6 +1,6 @@
 module RemoteRun
   class Configuration
-    attr_accessor :remote_path, :local_path, :login_as, :exclude, :temp_path, :quiet
+    attr_accessor :remote_path, :local_path, :login_as, :exclude, :temp_path, :quiet, :before_task, :after_task, :before_run, :after_run, :around_task, :around_run
     attr_reader :local_hostname, :identifier, :start_time
     attr_reader :host_manager, :task_manager
 
@@ -20,6 +20,12 @@ module RemoteRun
       @identifier = `echo $RANDOM`.strip
       @local_hostname = `hostname`.strip
 
+      @before_run = Proc.new{}
+      @after_run = Proc.new{}
+      @around_run = Proc.new {|&block| block.call }
+      @before_task = Proc.new{}
+      @after_task = Proc.new{}
+      @around_task = Proc.new {|&block| block.call }
       $runner = self
       yield self
     end
@@ -45,7 +51,9 @@ module RemoteRun
     end
 
     def run
-      Runner.new(self).run
+      @before_run.call(self)
+      @around_run.call { Runner.new(self).run }
+      @after_run.call(self)
     end
 
     private

data/lib/remote_run/runner.rb

@@ -12,6 +12,9 @@ module RemoteRun
       @task_manager = @configuration.task_manager
       @host_manager = @configuration.host_manager
       @starting_number_of_tasks = @task_manager.count
+      @before_task = @configuration.before_task
+      @after_task = @configuration.after_task
+      @around_task = @configuration.around_task
     end
 
     def run
@@ -84,15 +87,17 @@ module RemoteRun
 
     def start_forked_task(host, task)
       begin
+        @before_task.call(host, task, self)
         this_host = host.dup
         unless this_host.copy_codebase
           @task_manager.add(task)
           status = 0
         end
-        status = this_host.run(task.command)
+        @around_task.call { status = this_host.run(task.command) }
         host.unlock
       rescue Errno::EPIPE
       ensure
+        @after_task.call(host, task, self)
         Process.exit!(status)
       end
     end

data/lib/remote_run/version.rb

@@ -1,3 +1,3 @@
 module RemoteRun
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

data/lib/remote_run.rb

@@ -3,6 +3,5 @@ require File.join(File.dirname(__FILE__), "remote_run", "host")
 require File.join(File.dirname(__FILE__), "remote_run", "runner")
 require File.join(File.dirname(__FILE__), "remote_run", "configuration")
 require 'highline'
-
 module RemoteRun
 end

data/spec/remote_run/configuration_spec.rb

@@ -1,4 +1,24 @@
 require 'spec_helper'
+describe RemoteRun::Configuration do
+  let(:hosts){ ["foo", "bar"] }
+  let(:user){ "baz" }
+  let!(:pre_run) { double(call: true) }
+  let!(:post_run) { double(call: true) }
+  subject do
+    RemoteRun::Configuration.new do |config|
+      config.hosts = hosts
+      config.login_as = user
+      config.pre_run = pre_run
+      config.post_run = post_run
+    end
+  end
+
+  it "pre_run should have a callable block" do
+    pre_run.should_receive(:call).once
+    post_run.should_receive(:call).once
+    subject.run
+  end
+end
 
 describe RemoteRun::Configuration::HostManager do
   let(:host) { double(:host, is_up?: true, name: "foobar") }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: remote_run
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-10-31 00:00:00.000000000Z
+date: 2011-11-04 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: highline
-  requirement: &2169342220 !ruby/object:Gem::Requirement
+  requirement: &2165476480 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2169342220
+  version_requirements: *2165476480
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &2169341540 !ruby/object:Gem::Requirement
+  requirement: &2165475960 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,7 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *2169341540
+  version_requirements: *2165475960
 description: Can be used as a parallel unit test runner
 email:
 - casecommons-dev@googlegroups.com
@@ -71,18 +71,12 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: -1981659961289074609
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: -1981659961289074609
 requirements: []
 rubyforge_project: remote_run
 rubygems_version: 1.8.11