benchy 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in benchy.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/benchy.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "benchy/version"
+
+Gem::Specification.new do |s|
+  s.name        = "benchy"
+  s.version     = Benchy::VERSION
+  s.authors     = ["Brad Gessler"]
+  s.email       = ["brad@bradgessler.com"]
+  s.homepage    = ""
+  s.summary     = %q{Benchmark HTTP applications}
+  s.description = %q{A dirty-simple HTTP benchmarking application}
+
+  s.rubyforge_project = "benchy"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  s.add_runtime_dependency "em-http-request"
+  s.add_runtime_dependency "thor"
+end

data/bin/benchy

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+$:.unshift File.expand_path('../../lib', __FILE__)
+
+require 'rubygems'
+require 'benchy'
+
+Benchy::CLI.start

data/lib/benchy.rb

@@ -0,0 +1,125 @@
+require "benchy/version"
+require 'logger'
+require "thor"
+
+module Benchy
+  def self.logger
+    @logger ||= Logger.new($stdout)
+  end
+
+  class Dispatcher
+    attr_accessor :concurrency, :request
+
+    def initialize(request, concurrency=1)
+      @request, @concurrency = request, concurrency
+    end
+
+    def run
+      workers.each(&:run)
+    end
+
+    def halt
+      workers.each(&:halt)
+    end
+
+    def workers
+      @workers ||= (0...concurrency).map{|n| Worker.new(request, "worker.#{n}") }
+    end
+  end
+
+  class Worker
+    attr_accessor :request, :name
+
+    def initialize(request, name)
+      @request, @name = request, name
+    end
+
+    # Run, and keep running!
+    def run
+      return if halted?
+
+      http = request.em
+      http.callback {
+        Benchy.logger.info "#{name}\t| #{request.method.upcase} #{request.url} - HTTP #{http.response_header.status}"
+        run
+      }
+      http.errback {
+        Benchy.logger.debug "Connection error!"
+        halt # TODO - Make this fail the ping and try again, not halt
+      }
+    end
+
+    def halt
+      @halted = false
+    end
+
+    def halted?
+      !!@halted
+    end
+  end
+
+  # Represents an HTTP Request, but can't actually be executed
+  class Request
+    attr_accessor :url, :method, :headers, :body
+
+    def initialize(url, method, headers, body=nil)
+      @url, @method, @headers, @body = url, method, (headers || {}), body
+    end
+
+    # Grab an instance of an Em::Http request so we can run it somewhere.
+    def em
+      EventMachine::HttpRequest.new(url).send(method.downcase,
+        :head => default_headers.merge(headers),
+        :body => body,
+        :connect_timeout => 9000,    # Disable
+        :inactivity_timeout => 9000  # Disable
+      )
+    end
+
+    # Setup smart default headers to minimize the chances that a request gets rejected.
+    def default_headers
+      default_headers = {}
+      default_headers['Content-Type'] = 'application/binary-octet' if body
+      default_headers['Accepts']      = '*/*'
+      default_headers
+    end
+  end
+
+  # Parse out some command line goodness
+  class CLI < Thor
+    desc "benchmark URL", "Run benchmarks against server"
+    method_option :body,
+      :type => :string,
+      :aliases => '-b',
+      :desc => "Request body"
+    method_option :file,
+      :type => :string,
+      :aliases => '-f',
+      :desc => "File for request body"
+    method_option :headers,
+      :type => :hash,
+      :aliases => '-h',
+      :desc => 'HTTP headers'
+    method_option :method,
+      :type => :string, 
+      :desc => "Request method",
+      :aliases => '-m',
+      :default => 'GET'
+    method_option :concurrency,
+      :type => :numeric,
+      :desc => "Concurrent requests",
+      :aliases => '-c',
+      :default => 1
+
+    def benchmark(url)
+      req = Request.new(url, options[:method], options[:headers], self.class.body(options))
+      EM.run { Dispatcher.new(req, options[:concurrency]).run }
+    end
+
+  private
+    # Normalize the request body if its a file or a text string.
+    def self.body(options)
+      options[:file] ? File.read(options[:file]) : options[:body]
+    end
+  end
+end

data/lib/benchy/version.rb

@@ -0,0 +1,5 @@
+require 'em-http-request'
+
+module Benchy
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,76 @@
+--- !ruby/object:Gem::Specification
+name: benchy
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Brad Gessler
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: em-http-request
+  requirement: &70165834461960 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70165834461960
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: &70165834461540 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70165834461540
+description: A dirty-simple HTTP benchmarking application
+email:
+- brad@bradgessler.com
+executables:
+- benchy
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- benchy.gemspec
+- bin/benchy
+- lib/benchy.rb
+- lib/benchy/version.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: benchy
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Benchmark HTTP applications
+test_files: []
+has_rdoc: