Contents

module Quebert
  class Worker
    include Logging

    attr_accessor :exception_handler, :backend
    
    def initialize
      yield self if block_given?
    end
    
    # Start the worker backend and intercept exceptions if a handler is provided
    def start
      Signal.trap('TERM') { safe_stop }
      Signal.trap('INT') { safe_stop }

      logger.info "Worker started with #{backend.class.name} backend\n"
      while @controller = backend.reserve do      
        begin
          @controller.perform
        rescue Exception => e
          exception_handler ? exception_handler.call(e) : raise(e)
        end        
        @controller = nil

        stop if @terminate_sent
      end
    end
    
    def safe_stop
      if @terminate_sent
        logger.info "Ok! I get the point. Shutting down immediately."
        stop
      else
        logger.info "Finishing current job then shutting down."
        @terminate_sent = true
        stop unless @controller
      end
    end

    def stop
      logger.info "Worker stopping\n"
      exit 0
    end
    
  protected
    def backend
      @backend ||= Quebert.config.backend
    end
    
    def exception_handler
      @exception_handler ||= Quebert.config.worker.exception_handler
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
quebert-1.0.9	lib/quebert/worker.rb
quebert-1.0.8	lib/quebert/worker.rb
quebert-1.0.7	lib/quebert/worker.rb
quebert-1.0.6	lib/quebert/worker.rb
quebert-1.0.5	lib/quebert/worker.rb
quebert-1.0.4	lib/quebert/worker.rb
quebert-1.0.2	lib/quebert/worker.rb
quebert-1.0.1	lib/quebert/worker.rb