Contents

require 'serverengine'

module DummyLogGenerator
  module Worker
    def initialize
      reload
    end

    def reload
      @rate = config[:rate]
      @formatter = config[:formatter]
      @generator = config[:generator]
    end

    # thanks! ref. https://github.com/tagomoris/fluent-plugin-dummydata-producer/blob/a550fd4424f71cd9227e138c3c89f600ba40a0d5/lib/fluent/plugin/in_dummydata_producer.rb#L63
    def run
      batch_num = (@rate / 9).to_i + 1
      prev_data  = {}
      while !@stop
        current_time = Time.now.to_i
        rate_count = 0

        while !@stop && rate_count < @rate && Time.now.to_i == current_time
          batch_num.times do
            # ToDo: what if generation is slower than I/O?
            # We may should generate data and output in parallel
            prev_data = @generator.generate(prev_data)
            STDOUT.puts @formatter.format(prev_data)
          end
          rate_count += batch_num
          sleep 0.1
        end
        # wait for next second
        while !@stop && Time.now.to_i == current_time
          sleep 0.04
        end
      end
    end

    def stop
      @stop = true
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
dummy_log_generator-0.0.4	lib/dummy_log_generator/worker.rb
dummy_log_generator-0.0.3	lib/dummy_log_generator/worker.rb
dummy_log_generator-0.0.2	lib/dummy_log_generator/worker.rb