Contents

module Crusher
  
  class HttpLoadGenerator < LoadGenerator
    
    
    def initialize(crush_session, options)
      super(crush_session, options)
      @cookies = ::WebAgent::CookieManager.new 
    end
    
    def request(method, path, content = {}, &block)

      uri = URI.parse(path)

      http_args = content.keys.map do |key| 
        content[key] ? "#{URI.escape(key.to_s)}=#{URI.escape(content[key].to_s)}" : nil
      end
      
      qs, content = if method == :get
        [((uri.query || '').split('&') + http_args.compact).join("&"), nil]
      else
        [uri.query, http_args.join("&")] 
      end

      options = {
        :verb => method,
        :host => uri.host,
        :port => uri.port,
        :request => uri.path,
        :query_string => qs,
        :content => content,
        :contenttype => 'application/x-www-form-urlencoded',
        :cookie => @cookies.find(uri),
        :ssl => (uri.scheme == "https" ? true : false)
      }
      
      start_time = Time.now
      EM::Protocols::HttpClient.request(options).callback do |response|
        duration = ((Time.now - start_time) * 1000).to_i
        
        response[:headers].each do |header|
          matcher = /Set-Cookie:\s*(.*)/i.match(header)
          @cookies.parse(matcher[1], uri) if matcher
        end
        
        block.call(response, duration) if block
      end

    end
    
    
  end
  
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
crusher-0.0.7	lib/crusher/http_load_generator.rb
crusher-0.0.6	lib/crusher/http_load_generator.rb
crusher-0.0.5	lib/crusher/http_load_generator.rb
crusher-0.0.4	lib/crusher/http_load_generator.rb
crusher-0.0.3	lib/crusher/http_load_generator.rb
crusher-0.0.2	lib/crusher/http_load_generator.rb