module Toiler

# File lib/toiler.rb, line 31
def active_worker_class_registry
  active_queues = options[:active_queues]
  if active_queues
    active_queues.each_with_object({}) do |q, registry|
      worker = @worker_class_registry[q]
      if worker.nil?
        logger.warn "No worker assigned to queue: #{q}"
      else
        registry[q] = worker
      end
    end
  else
    @worker_class_registry
  end
end

# File lib/toiler.rb, line 63
def default_options
  {
    auto_visibility_timeout: false,
    concurrency: 1,
    auto_delete: false,
    shutdown_timeout: 5,
    batch: false
  }
end

# File lib/toiler.rb, line 47
def fetcher(queue)
  fetchers["fetcher_#{queue}".to_sym]
end

# File lib/toiler.rb, line 23
def logger
  Toiler::Utils::Logging.logger
end

# File lib/toiler.rb, line 55
def processor_pool(queue)
  processor_pools["processor_pool_#{queue}".to_sym]
end

# File lib/toiler.rb, line 27
def queues
  worker_class_registry.keys
end

# File lib/toiler.rb, line 73
def register_worker(queue, worker)
  @worker_class_registry[queue] = worker
end

# File lib/toiler.rb, line 51
def set_fetcher(queue, val)
  fetchers["fetcher_#{queue}".to_sym] = val
end

# File lib/toiler.rb, line 59
def set_processor_pool(queue, val)
  processor_pools["processor_pool_#{queue}".to_sym] = val
end

# File lib/toiler.rb, line 77
def worker_class_registry=(val)
  @worker_class_registry = val
end