Module: HireFire::Macro::Sidekiq

Extended by:
Sidekiq
Included in:
Sidekiq
Defined in:
lib/hirefire/macro/sidekiq.rb

Instance Method Summary collapse

Instance Method Details

#queue(*queues) ⇒ Integer

Counts the amount of jobs in the (provided) Sidekiq queue(s).

Examples:

Sidekiq Macro Usage

HireFire::Macro::Sidekiq.queue # all queues
HireFire::Macro::Sidekiq.queue("email") # only email queue
HireFire::Macro::Sidekiq.queue("audio", "video") # audio and video queues
HireFire::Macro::Sidekiq.queue("email", skip_scheduled: true) # only email, will not count scheduled queue
HireFire::Macro::Sidekiq.queue("audio", skip_retries: true) # only audio, will not count the retries queue

Parameters:

  • queues (Array) —

    provide one or more queue names, or none for “all”.

Returns:

  • (Integer) —

    the number of jobs in the queue(s).



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/hirefire/macro/sidekiq.rb', line 20

def queue(*queues)
  require "sidekiq/api"

  queues.flatten!

  if queues.last.is_a?(Hash)
    options = queues.pop
  else
    options = {}
  end

  queues = queues.map(&:to_s)
  queues = ::Sidekiq::Stats.new.queues.map { |name, _| name } if queues.empty?

  in_queues = queues.inject(0) do |memo, name|
    memo += ::Sidekiq::Queue.new(name).size
    memo
  end

  if !options[:skip_scheduled]
    max = options[:max_scheduled]
    in_schedule = ::Sidekiq::ScheduledSet.new.inject(0) do |memo, job|
      memo += 1 if queues.include?(job["queue"]) && job.at <= Time.now
      break memo if max && memo >= max
      memo
    end
  end

  if !options[:skip_retries]
    in_retry = ::Sidekiq::RetrySet.new.inject(0) do |memo, job|
      memo += 1 if queues.include?(job["queue"]) && job.at <= Time.now
      memo
    end
  end

  i = ::Sidekiq::VERSION >= "3.0.0" ? 2 : 1
  in_progress = ::Sidekiq::Workers.new.inject(0) do |memo, job|
    memo += 1 if queues.include?(job[i]["queue"]) && job[i]["run_at"] <= Time.now.to_i
    memo
  end

  [in_queues, in_schedule, in_retry, in_progress].compact.inject(&:+)
end