actionpack/lib/action_dispatch/middleware/callbacks.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

module ActionDispatch
  # Provides callbacks to be executed before and after dispatching the request.
  class Callbacks
    include ActiveSupport::Callbacks

    define_callbacks :call

    class << self
      def to_prepare(*args, &block)
        ActiveSupport::Reloader.to_prepare(*args, &block)
      end

      def to_cleanup(*args, &block)
        ActiveSupport::Reloader.to_complete(*args, &block)
      end

      deprecate to_prepare: 'use ActiveSupport::Reloader.to_prepare instead',
        to_cleanup: 'use ActiveSupport::Reloader.to_complete instead'

      def before(*args, &block)
        set_callback(:call, :before, *args, &block)
      end

      def after(*args, &block)
        set_callback(:call, :after, *args, &block)
      end
    end

    def initialize(app)
      @app = app
    end

    def call(env)
      error = nil
      result = run_callbacks :call do
        begin
          @app.call(env)
        rescue => error
        end
      end
      raise error if error
      result
    end
  end
end