123456789_123456789_123456789_123456789_123456789_

Class: Bundler::Worker

Relationships & Source Files
Namespace Children
Exceptions:
Inherits: Object
Defined in: lib/bundler/worker.rb

Constant Summary

Class Method Summary

Instance Attribute Summary

Instance Method Summary

Constructor Details

.new(size, name, func) ⇒ Worker

Creates a worker pool of specified size

Parameters:

  • size (Integer)

    Size of pool

  • name (String)

    name the name of the worker

  • func (Proc)

    job to run in inside the worker pool

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 24

def initialize(size, name, func)
  @name = name
  @request_queue = Queue.new
  @response_queue = Queue.new
  @func = func
  @size = size
  @threads = nil
  SharedHelpers.trap("INT") { abort_threads }
end

Instance Attribute Details

#nameString (readonly)

Returns:

  • (String)

    the name of the worker

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 17

attr_reader :name

Instance Method Details

#abort_threads (private)

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 78

def abort_threads
  return unless @threads
  Bundler.ui.debug("\n#{caller.join("\n")}")
  @threads.each(&:exit)
  exit 1
end

#apply_func(obj, i) (private)

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 63

def apply_func(obj, i)
  @func.call(obj, i)
rescue Exception => e
  WrappedException.new(e)
end

#create_threads (private)

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 85

def create_threads
  creation_errors = []

  @threads = Array.new(@size) do |i|
    begin
      Thread.start { process_queue(i) }.tap do |thread|
        thread.name = "#{name} Worker ##{i}" if thread.respond_to?(:name=)
      end
    rescue ThreadError => e
      creation_errors << e
      nil
    end
  end.compact

  return if creation_errors.empty?

  message = "Failed to create threads for the #{name} worker: #{creation_errors.map(&:to_s).uniq.join(", ")}"
  raise ThreadCreationError, message if @threads.empty?
  Bundler.ui.info message
end

#deq

Retrieves results of job function being executed in worker pool

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 43

def deq
  result = @response_queue.deq
  raise result.exception if result.is_a?(WrappedException)
  result
end

#enq(obj)

Enqueue a request to be executed in the worker pool

Parameters:

  • obj (String)

    mostly it is name of spec that should be downloaded

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 37

def enq(obj)
  create_threads unless @threads
  @request_queue.enq obj
end

#process_queue(i) (private)

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 55

def process_queue(i)
  loop do
    obj = @request_queue.deq
    break if obj.equal? POISON
    @response_queue.enq apply_func(obj, i)
  end
end

#stop

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 49

def stop
  stop_threads
end

#stop_threads (private)

Stop the worker threads by sending a poison object down the request queue so as worker threads after retrieving it, shut themselves down

[ GitHub ]

  
# File 'lib/bundler/worker.rb', line 71

def stop_threads
  return unless @threads
  @threads.each { @request_queue.enq POISON }
  @threads.each(&:join)
  @threads = nil
end