blitz 0.1.15 → 0.1.16

data/blitz.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{blitz}
-  s.version = "0.1.15"
+  s.version = "0.1.16"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["pcapr"]
-  s.date = %q{2011-10-13}
+  s.date = %q{2011-10-25}
   s.default_executable = %q{blitz}
   s.description = %q{Make load and performance testing a fun sport}
   s.email = %q{support@blitz.io}
@@ -34,11 +34,14 @@ Gem::Specification.new do |s|
     "lib/blitz/command/couch.rb",
     "lib/blitz/command/curl.rb",
     "lib/blitz/command/help.rb",
+    "lib/blitz/command/traceroute.rb",
     "lib/blitz/command/version.rb",
     "lib/blitz/curl/error.rb",
     "lib/blitz/curl/rush.rb",
     "lib/blitz/curl/sprint.rb",
     "lib/blitz/helper.rb",
+    "lib/blitz/traceroute.rb",
+    "lib/blitz/traceroute/error.rb",
     "spec/command/curl_spec.rb",
     "test/helper.rb",
     "test/test_blitz.rb"

data/lib/blitz.rb

@@ -6,7 +6,7 @@ require 'pp'
 
 class Blitz # :nodoc:
     require 'blitz/helper'
-    Version = "0.1.15"
+    Version = "0.1.16"
         
     extend Blitz::Helper
     
@@ -32,4 +32,6 @@ require 'blitz/client'
 require 'blitz/curl/error'
 require 'blitz/curl/sprint'
 require 'blitz/curl/rush'
+require 'blitz/traceroute'
+require 'blitz/traceroute/error'
 require 'blitz/command'

data/lib/blitz/client.rb

@@ -18,6 +18,10 @@ class Client # :nodoc:
         JSON.parse blitz['/api/1/curl/execute'].post(data.to_json)
     end
     
+    def traceroute_execute data
+        JSON.parse blitz['/api/1/traceroute/execute'].post(data.to_json)
+    end
+    
     def login
         JSON.parse blitz['/login/api'].get
     end

data/lib/blitz/command.rb

@@ -29,6 +29,12 @@ class Command # :nodoc:
     include Test::Unit::Assertions
     include Helper
     include Term::ANSIColor
+    
+    def shift key, argv
+        val = argv.shift
+        assert_not_nil(val, "missing value for #{key}")
+        val
+    end
 end
 end # Blitz
 

data/lib/blitz/command/curl.rb

@@ -409,12 +409,6 @@ class Curl < Command # :nodoc:
 
         hash
     end
-
-    def shift key, argv
-        val = argv.shift
-        assert_not_nil(val, "missing value for #{key}")
-        val
-    end
 end # Curl
 end # Command
 end # Blitz

data/lib/blitz/command/help.rb

@@ -10,6 +10,8 @@ class Help < Command # :nodoc:
             { :cmd => 'couch:fuzz', :help => 'Auto generate blitz tests from CouchDB' },
             { :cmd => 'curl', :help => 'Run a sprint or a rush' },
             { :cmd => 'curl:help', :help => 'Show help on sprint and rushing' },
+            { :cmd => 'traceroute', :help => 'Run traceroute remotely' },
+            { :cmd => 'traceroute:help', :help => 'Show help on traceroute' },
             { :cmd => 'version', :help => 'Show the version of this Ruby gem' }
         ]
         

data/lib/blitz/command/traceroute.rb

@@ -0,0 +1,81 @@
+class Blitz
+class Command
+class Traceroute < Command # :nodoc:
+    def cmd_help argv
+        puts
+        msg "Usage: blitz traceroute [-r <region>] host"
+        puts
+    end
+
+    def cmd_default argv
+        args = parse_cli argv
+        
+        continue = true
+        last_index = nil
+        begin
+            [ 'INT', 'STOP', 'HUP' ].each do |s| 
+                trap(s) { continue = false }
+            end
+            
+            job = ::Blitz::Traceroute.queue args
+            msg "running from #{yellow(job.region)}..."
+            puts
+            job.result do |result|
+                print_result args, result, last_index
+                if not result.hops.empty?
+                    last_index = result.hops.size
+                end
+                sleep 2.0 if not continue
+                continue
+            end
+            puts
+            msg "[#{red('aborted')}]" if not continue
+        rescue ::Blitz::Traceroute::Error::Region => e
+            error "#{yellow(e.region)}: #{red(e.message)}"
+        rescue ::Blitz::Traceroute::Error => e
+            error red(e.message)
+        end
+    end
+    
+    def print_result args, result, last_index
+        if last_index and result.hops.size == last_index
+            return
+        end
+        
+        result.hops[(last_index || 0)..result.hops.size].each do |hop|
+            if hop =~ /!/
+                puts red(hop)
+            elsif hop =~ /\*/
+                puts yellow(hop)
+            else
+                puts hop
+            end
+        end
+    end
+    
+    private
+    def parse_cli argv
+        args = Hash.new
+        
+        while not argv.empty?
+            break if argv.first[0,1] != '-'
+            
+            k = argv.shift
+            if [ '-r', '--region' ].member? k
+                args['region'] = shift(k, argv)
+                next
+            end
+            
+            raise ArgumentError, "Unknown option #{k}"
+        end
+            
+        if argv.empty?
+            raise Test::Unit::AssertionFailedError, "missing host"
+        end
+        
+        args['host'] = argv.shift
+        return args
+    end
+end
+end # Command
+end # Blitz

data/lib/blitz/traceroute.rb

@@ -0,0 +1,79 @@
+class Blitz
+class Traceroute
+    class Result
+        # The region from which the traceroute was executed
+        attr_reader :region
+        
+        # The timeline of the rush containing various statistics.
+        attr_reader :hops
+        
+        def initialize json # :nodoc:
+            result = json['result']
+            @region = result['region']
+            @hops = result['hops']
+        end        
+    end
+    
+    def self.execute args
+        self.queue(args).result
+    end
+    
+    def self.queue args # :nodoc:
+        res = Command::API.client.traceroute_execute args
+        raise Error.new(res) if res['error']
+        return self.new res
+    end
+    
+    attr_reader :job_id # :nodoc:
+    attr_reader :region # :nodoc:
+    
+    def initialize json # :nodoc:
+        @job_id = json['job_id']
+        @region = json['region']
+    end
+    
+    def result &block # :nodoc:
+        last = nil
+        while true
+            sleep 2.0
+
+            job = Command::API.client.job_status job_id
+            if job['error']
+                raise Error
+            end
+            
+            result = job['result']
+            next if job['status'] == 'queued'
+            next if job['status'] == 'running' and not result
+
+            raise Error if not result
+
+            error = result['error']
+            if error
+                if error == 'dns'
+                    raise Error::DNS.new(result)
+                elsif error == 'parse'
+                    raise Error::Parse.new(result)
+                else
+                    raise Error
+                end
+            end
+            
+            last = Result.new(job)
+            continue = yield last rescue false
+            if continue == false
+                abort!
+                break
+            end
+            
+            break if job['status'] == 'completed'
+        end
+        
+        return last
+    end
+    
+    def abort! # :nodoc:
+        Command::API.client.abort_job job_id rescue nil
+    end    
+end
+end # blitz

data/lib/blitz/traceroute/error.rb

@@ -0,0 +1,28 @@
+class Blitz
+class Traceroute
+class Error < StandardError # :nodoc:
+    def initialize json={}
+        super json['reason'] || "Hmmm, something went wrong. Try again in a little bit?"
+    end
+    
+    # The base class for all exceptions thrown by the distributed engines
+    class Region < Error
+        # The region from which the test was run
+        attr_reader :region
+        
+        def initialize json # :nodoc:
+            @region = json['region']
+            super
+        end
+    end
+    
+    # This exception is raised when the DNS resolution fails
+    class DNS < Region
+    end
+    
+    # This exception is raised when the arguments to traceroute are invalid
+    class Parse < Region
+    end    
+end
+end # Traceroute
+end # Blitz

metadata

@@ -2,7 +2,7 @@
 name: blitz
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.15
+  version: 0.1.16
 platform: ruby
 authors: 
 - pcapr
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-10-13 00:00:00 -07:00
+date: 2011-10-25 00:00:00 -07:00
 default_executable: blitz
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -148,11 +148,14 @@ files:
 - lib/blitz/command/couch.rb
 - lib/blitz/command/curl.rb
 - lib/blitz/command/help.rb
+- lib/blitz/command/traceroute.rb
 - lib/blitz/command/version.rb
 - lib/blitz/curl/error.rb
 - lib/blitz/curl/rush.rb
 - lib/blitz/curl/sprint.rb
 - lib/blitz/helper.rb
+- lib/blitz/traceroute.rb
+- lib/blitz/traceroute/error.rb
 - spec/command/curl_spec.rb
 - test/helper.rb
 - test/test_blitz.rb
@@ -170,7 +173,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 510668115
+      hash: -370311181
       segments: 
       - 0
       version: "0"