Contents

module Crusher
  
  class LoadGenerator
  
    def initialize(crush_session, options)
      @crush_session = crush_session
      @options = options
      
      Dir.mkdir @options[:stats_log_dir] if @options[:stats_log_dir] && !File.exist?(@options[:stats_log_dir])

    end
    
    def act!; end
  
    def prepare; end
        
    def log(message)
      @crush_session.log(message)
    end
    
    def log_stats(event_type, data)
      keys, values = data.map{|k,v| [k.to_s,v.to_s] }.transpose
      stats_log_file(event_type) do |file|
        file.syswrite "#{Time.now.strftime('%Y-%m-%d %H:%M:%S')},#{values.join(',')}\n"
        file.flush unless file.fsync
      end
    end
    
    def stats_log_file(event_type)
      event_type = event_type.to_sym
      File.open(File.join(@options[:stats_log_dir], "#{event_type}.csv"), 'a') do |file|
        yield file
      end
    end  
  end
  
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
crusher-0.0.8.4	lib/crusher/load_generator.rb
crusher-0.0.8.3	lib/crusher/load_generator.rb
crusher-0.0.8.2	lib/crusher/load_generator.rb
crusher-0.0.8.1	lib/crusher/load_generator.rb
crusher-0.0.8	lib/crusher/load_generator.rb
crusher-0.0.7	lib/crusher/load_generator.rb