Class: Rollbar::Middleware::Rails::RollbarMiddleware

Inherits:
Object
  • Object
show all
Includes:
ExceptionReporter, RequestDataExtractor
Defined in:
lib/rollbar/middleware/rails/rollbar.rb

Constant Summary

Constants included from RequestDataExtractor

RequestDataExtractor::ALLOWED_BODY_PARSEABLE_METHODS, RequestDataExtractor::ALLOWED_HEADERS_REGEX

Instance Method Summary collapse

Methods included from ExceptionReporter

#report_exception_to_rollbar

Methods included from RequestDataExtractor

#extract_person_data_from_controller, #extract_request_data_from_rack, #scrub_params, #scrub_url

Constructor Details

#initialize(app) ⇒ RollbarMiddleware

Returns a new instance of RollbarMiddleware.



11
12
13
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 11

def initialize(app)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 15

def call(env)
  self.request_data = nil

  Rollbar.reset_notifier!

  env['rollbar.scope'] = scope = fetch_scope(env)

  Rollbar.scoped(scope) do
    begin
      response = @app.call(env)

      if (framework_exception = env['action_dispatch.exception'])
        report_exception_to_rollbar(env, framework_exception)
      end

      response
    rescue Exception => exception
      report_exception_to_rollbar(env, exception)
      raise
    end
  end
end

#context(request_data) ⇒ Object 



68
69
70
71
72
73
74
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 68

def context(request_data)
  return unless request_data[:params]

  route_params = request_data[:params]
  # make sure route is a hash built by RequestDataExtractor
  return route_params[:controller].to_s + '#' + route_params[:action].to_s if route_params.is_a?(Hash) && !route_params.empty?
end

#extract_request_data(env) ⇒ Object 



57
58
59
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 57

def extract_request_data(env)
  extract_request_data_from_rack(env)
end

#fetch_scope(env) ⇒ Object 



38
39
40
41
42
43
44
45
46
47
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 38

def fetch_scope(env)
  # Scope a new notifier with request data and a Proc for person data
  # for any reports that happen while a controller is handling a request

  {
    :request => proc { request_data(env) },
    :person => person_data_proc(env),
    :context => proc { context(request_data(env)) }
  }
end

#person_data_proc(env) ⇒ Object 



61
62
63
64
65
66
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 61

def person_data_proc(env)
  block = proc { extract_person_data_from_controller(env) }
  return block unless defined?(ActiveRecord::Base) && ActiveRecord::Base.connected?

  proc { ActiveRecord::Base.connection_pool.with_connection(&block) }
end

#request_data(env) ⇒ Object 



49
50
51
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 49

def request_data(env)
  Thread.current[:'_rollbar.rails.request_data'] ||= extract_request_data(env)
end

#request_data=(value) ⇒ Object 



53
54
55
 
# File 'lib/rollbar/middleware/rails/rollbar.rb', line 53

def request_data=(value)
  Thread.current[:'_rollbar.rails.request_data'] = value
end