mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
26 lines
701 B
Ruby
26 lines
701 B
Ruby
module ActionDispatch
|
|
class Rescue
|
|
def initialize(app, rescuers = {}, &block)
|
|
@app, @rescuers = app, {}
|
|
rescuers.each { |exception, rescuer| rescue_from(exception, rescuer) }
|
|
instance_eval(&block) if block_given?
|
|
end
|
|
|
|
def call(env)
|
|
@app.call(env)
|
|
rescue Exception => exception
|
|
if rescuer = @rescuers[exception.class.name]
|
|
env['action_dispatch.rescue.exception'] = exception
|
|
rescuer.call(env)
|
|
else
|
|
raise exception
|
|
end
|
|
end
|
|
|
|
protected
|
|
def rescue_from(exception, rescuer)
|
|
exception = exception.class.name if exception.is_a?(Exception)
|
|
@rescuers[exception.to_s] = rescuer
|
|
end
|
|
end
|
|
end
|