mirror of
https://github.com/github/rails.git
synced 2026-04-26 03:00:59 -04:00
27 lines
701 B
Ruby
27 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
|