Contents

# frozen_string_literal: true

module SafeRequestTimeout
  # Rack middleware that adds a timeout block to all requests.
  class RackMiddleware
    # @param app [Object] The Rack application to wrap.
    # @param timeout [Integer, Proc, nil] The timeout in seconds.
    def initialize(app, timeout = nil)
      @app = app
      @timeout = timeout
      @timeout_block = true if timeout.is_a?(Proc) && timeout.arity == 1
    end

    def call(env)
      SafeRequestTimeout.timeout(@timeout_block ? @timeout.call(env) : @timeout) do
        @app.call(env)
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
safe_request_timeout-1.0.2	lib/safe_request_timeout/rack_middleware.rb
safe_request_timeout-1.0.1	lib/safe_request_timeout/rack_middleware.rb
safe_request_timeout-1.0.0	lib/safe_request_timeout/rack_middleware.rb