Class: Evilution::Parallel::WorkQueue

Inherits:
Object
  • Object
show all
Defined in:
lib/evilution/parallel/work_queue.rb

Defined Under Namespace

Classes: WorkerStat

Constant Summary collapse

SHUTDOWN = 
:__shutdown__
STATS = 
:__stats__
TIMING_GRACE_PERIOD = 
5

Instance Method Summary collapse

Constructor Details

#initialize(size:, hooks: nil, prefetch: 1, item_timeout: nil) ⇒ WorkQueue

Returns a new instance of WorkQueue.

Raises:

  • (ArgumentError) 


24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/evilution/parallel/work_queue.rb', line 24

def initialize(size:, hooks: nil, prefetch: 1, item_timeout: nil)
  raise ArgumentError, "pool size must be a positive integer, got #{size.inspect}" unless size.is_a?(Integer) && size >= 1
  raise ArgumentError, "prefetch must be a positive integer, got #{prefetch.inspect}" unless prefetch.is_a?(Integer) && prefetch >= 1
  unless item_timeout.nil? || (item_timeout.is_a?(Numeric) && item_timeout.positive?)
    raise ArgumentError, "item_timeout must be nil or a positive number, got #{item_timeout.inspect}"
  end

  @size = size
  @hooks = hooks
  @prefetch = prefetch
  @item_timeout = item_timeout
  @worker_stats = []
end

Instance Method Details

#map(items) ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/evilution/parallel/work_queue.rb', line 38

def map(items, &)
  return [] if items.empty?

  worker_count = [@size, items.length].min
  workers = spawn_workers(worker_count, &)

  begin
    distribute_and_collect(items, workers)
  ensure
    shutdown_workers(workers)
    @worker_stats = build_worker_stats(workers)
  end
end

#worker_stats ⇒ Object 



52
53
54
 
# File 'lib/evilution/parallel/work_queue.rb', line 52

def worker_stats
  @worker_stats.map { |stat| stat.dup.freeze }
end