Class: Sidekiq::LimitFetch::Queues

Inherits:
Object
  • Object
show all
Defined in:
lib/sidekiq/limit_fetch/queues.rb

Constant Summary collapse

THREAD_KEY = 
:acquired_queues

Instance Method Summary collapse

Constructor Details

#initialize(options) ⇒ Queues

Returns a new instance of Queues.



5
6
7
8
9
10
11
12
13
14
15
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 5

def initialize(options)
  @queues    = options[:queues]
  @namespace = options[:namespace]
  @dynamic   = options[:dynamic]

  options[:strict] ? strict_order! : weighted_order!

  set :process_limit, options[:process_limits]
  set :limit, options[:limits]
  set_blocks options[:blocking]
end

Instance Method Details

#acquire ⇒ Object 



17
18
19
20
21
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 17

def acquire
  selector.acquire(ordered_queues, @namespace)
    .tap {|it| save it }
    .map {|it| "queue:#{it}" }
end

#add(queues) ⇒ Object 



33
34
35
36
37
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 33

def add(queues)
  queues.each do |queue|
    @queues.push queue unless @queues.include? queue
  end
end

#dynamic? ⇒ Boolean

Returns:

  • (Boolean) 


29
30
31
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 29

def dynamic?
  @dynamic
end

#ordered_queues ⇒ Object 



41
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 41

def ordered_queues; @queues end

#release_except(full_name) ⇒ Object 



23
24
25
26
27
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 23

def release_except(full_name)
  queues = restore
  queues.delete full_name[/queue:(.*)/, 1] if full_name
  selector.release queues, @namespace
end

#strict_order! ⇒ Object 



39
40
41
42
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 39

def strict_order!
  @queues.uniq!
  def ordered_queues; @queues end
end

#weighted_order! ⇒ Object 



44
45
46
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 44

def weighted_order!
  def ordered_queues; @queues.shuffle.uniq end
end