Contents

module Chillout
  class Worker
    attr_reader :dispatcher, :queue, :logger

    def initialize(dispatcher, queue, logger, container_class=CreationsContainer)
      @dispatcher = dispatcher
      @queue = queue
      @logger = logger
      @container_class = container_class
    end

    def get_all_containers_to_process
      logger.debug "Waiting for at least one container."
      all_containers = [queue.pop]
      logger.debug "Received at least one container."
      loop do
        begin
          all_containers << queue.pop(true)
        rescue ThreadError
          break
        end
      end
      logger.debug "Received containers to process: #{all_containers.count}"
      all_containers
    end

    def merge_containers(containers_to_merge)
      creations_container = @container_class.new
      for container in containers_to_merge
        creations_container.merge(container)
      end
      creations_container
    end

    def send_creations(creations_container)
      logger.debug "Trying to send creations"
      dispatcher.send_creations(creations_container)
      logger.info "Metrics sent"
    rescue Dispatcher::SendCreationsFailed
      queue << creations_container
      logger.error "Sending metrics failed"
    end

    def send_startup_message
      dispatcher.send_startup_message
      logger.debug "Sending startup message"
    end

    def run
      logger.info "Worker started"
      send_startup_message
      loop do
        containers_to_merge = get_all_containers_to_process
        creations_container = merge_containers(containers_to_merge)
        send_creations(creations_container)
        logger.flush
        sleep 5
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
chillout-0.8.1	lib/chillout/worker.rb
chillout-0.8.0	lib/chillout/worker.rb