Module: Sidekiq::LimitFetch::Queues

Extended by:
Queues
Included in:
Queues
Defined in:
lib/sidekiq/limit_fetch/queues.rb

Constant Summary collapse

THREAD_KEY = 
:acquired_queues

Instance Method Summary collapse

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

#namespace ⇒ Object 



48
49
50
51
52
53
54
55
56
 
# File 'lib/sidekiq/limit_fetch/queues.rb', line 48

def namespace
  @namespace ||= Sidekiq.redis do |it|
    if it.respond_to?(:namespace) and it.namespace
      it.namespace + ':'
    else
      ''
    end
  end
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

#start(options) ⇒ Object 



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

def start(options)
  @queues    = options[:queues]
  @dynamic   = options[:dynamic]

  options[:strict] ? strict_order! : weighted_order!

  set :process_limit, options[:process_limits]
  set :limit, options[:limits]
  set_blocks options[:blocking]
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