examples: a scratchy launcher helper

examples/itchy.rb

@@ -6,14 +6,6 @@
 
 id = 'itchy'
 
-OptionParser.new do |opts|
-  opts.banner = "Usage: itchy.rb [options]"
-
-  opts.on("--id ID", "Assign node ID") do |v|
-    id = v
-  end
-end.parse!
-
 class Itchy
   include Celluloid
 
@@ -47,6 +39,14 @@ def work(value)
 end
 
 if __FILE__ == $0
+  OptionParser.new do |opts|
+    opts.banner = "Usage: itchy.rb [options]"
+
+    opts.on("--id ID", "Assign node ID") do |v|
+      id = v
+    end
+  end.parse!
+
   Itchy.supervise_as :itchy
   puts "Starting itchy with ID '#{id}'"
   DCell.start :id =>id, :registry => registry

examples/massive-scratchy.rb

@@ -2,6 +2,17 @@
 require 'dcell'
 require_relative 'registry'
 require_relative 'itchy'
+require 'optparse'
+
+local = true
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: massive-scratchy.rb [options]"
+
+  opts.on("--no-local", "Do not run local celluloid actor tests") do
+    local = false
+  end
+end.parse!
 
 DCell.start :registry => registry
 puts "Making itchy work hard everywhere!"
@@ -56,6 +67,8 @@ def run_test(itchies, method, info, args=[])
 run_test itchies, :async, "remote async", repeat
 itchies.each {|itchy| itchy.terminate}
 
+exit unless local
+
 itchies = max.times.map {Itchy.new}
 run_test itchies, :future, "local futures", repeat
 run_test itchies, :async, "local async", repeat

examples/scratchy-launcher.rb

@@ -0,0 +1,43 @@
+#!/usr/bin/env ruby
+
+require 'optparse'
+
+pids = Array.new
+count = 1
+
+args = []
+scratchy = 'scratchy.rb'
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: scratchy-launcher.rb [options]"
+
+  opts.on("--count COUNT", OptionParser::DecimalInteger, "Number of massive-scratchy instances") do |v|
+    count = v
+  end
+  opts.on("--massive", "execute massive scratchy instead of generic") do
+    args = ['--no-local']
+    scratchy = 'massive-scratchy.rb'
+  end
+end.parse!
+
+Signal.trap(:INT) do
+  pids.each do |pid|
+    begin
+      Process.kill :TERM, pid if pid
+    rescue
+    end
+  end
+end
+
+count.times do |id|
+  pid = Process.spawn Gem.ruby, File.expand_path("./examples/#{scratchy}"), *args
+  unless pid
+    STDERR.print "ERROR: Couldn't start test node."
+  end
+  pids << pid
+end
+
+start = Time.now
+Process.waitall
+stop = Time.now
+puts "All scratchies completed within #{stop-start} seconds"