hgmnz/keikokuc

module HandlesTimeout
  def self.included(base)
    base.extend ClassMethods
  end

  RequestTimeout = Class.new

  module ClassMethods
    def handle_timeout(method_name)
      alias_method "#{method_name}_without_timeout", method_name
      define_method method_name do |*args|
        begin
          if args.empty?
            Timeout::timeout(5) { send("#{method_name}_without_timeout") }
          else
            Timeout::timeout(5) { send("#{method_name}_without_timeout", *args) }
          end
        rescue Timeout::Error
          [nil, RequestTimeout]
        end
      end
    end
  end
end