Contents

require 'system_timer'
require 'qs'
require 'qs/runner'

module Qs

  class QsRunner < Runner

    attr_reader :timeout

    def initialize(handler_class, args = nil)
      super(handler_class, args)
      @timeout = handler_class.timeout || Qs.config.timeout
    end

    def run
      OptionalTimeout.new(self.timeout) do
        run_callbacks self.handler_class.before_callbacks
        self.handler.init
        self.handler.run
        run_callbacks self.handler_class.after_callbacks
      end
    rescue TimeoutError => exception
      error = TimeoutError.new "#{handler_class} timed out (#{timeout}s)"
      error.set_backtrace(exception.backtrace)
      raise error
    end

    private

    def run_callbacks(callbacks)
      callbacks.each{ |proc| self.handler.instance_eval(&proc) }
    end

    module OptionalTimeout
      def self.new(timeout, &block)
        if !timeout.nil?
          SystemTimer.timeout_after(timeout, TimeoutError, &block)
        else
          block.call
        end
      end
    end

  end

  TimeoutError = Class.new(RuntimeError)

end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
qs-0.5.0	lib/qs/qs_runner.rb
qs-0.4.0	lib/qs/qs_runner.rb
qs-0.3.0	lib/qs/qs_runner.rb
qs-0.2.0	lib/qs/qs_runner.rb
qs-0.1.0	lib/qs/qs_runner.rb