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_request(start_time, uri, block, options)

    end
    
    private
    
    def em_request(start_time, uri, block, options, times_to_retry=1)
      begin
        http = EM::Protocols::HttpClient.request(options)
      rescue EventMachine::ConnectionError => error
        unless times_to_retry > 3
          @crush_session.logger.warn "retrying for #{times_to_retry} time with error \"#{error}\" failed request #{options.inspect}"
          em_request(start_time, uri, block, options, (times_to_retry + 1))
        end
        @crush_session.logger.error "Failed after 3 tries with error \"#{error}\": #{options.inspect}"
        raise error
      end
      http.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

4 entries across 4 versions & 1 rubygems

Version	Path
crusher-0.0.8.3	lib/crusher/http_load_generator.rb
crusher-0.0.8.2	lib/crusher/http_load_generator.rb
crusher-0.0.8.1	lib/crusher/http_load_generator.rb
crusher-0.0.8	lib/crusher/http_load_generator.rb