/
logger.rb
80 lines (70 loc) · 2.24 KB
/
logger.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
# frozen_string_literal: true
require "active_support/core_ext/time/conversions"
require "active_support/log_subscriber"
require "rack/body_proxy"
module Rails
module Rack
# Sets log tags, logs the request, calls the app, and flushes the logs.
#
# Log tags (+taggers+) can be an Array containing: methods that the +request+
# object responds to, objects that respond to +to_s+ or Proc objects that accept
# an instance of the +request+ object.
class Logger < ActiveSupport::LogSubscriber
def initialize(app, taggers = nil)
@app = app
@taggers = taggers || []
end
def call(env)
request = ActionDispatch::Request.new(env)
if logger.respond_to?(:tagged)
logger.tagged(*compute_tags(request)) { call_app(request, env) }
else
call_app(request, env)
end
end
private
def call_app(request, env) # :doc:
instrumenter = ActiveSupport::Notifications.instrumenter
handle = instrumenter.build_handle("request.action_dispatch", { request: request })
handle.start
logger.info { started_request_message(request) }
status, headers, body = response = @app.call(env)
body = ::Rack::BodyProxy.new(body, &handle.method(:finish))
if response.frozen?
[status, headers, body]
else
response[2] = body
response
end
rescue Exception
handle.finish
raise
ensure
ActiveSupport::LogSubscriber.flush_all!
end
# Started GET "/session/new" for 127.0.0.1 at 2012-09-26 14:51:42 -0700
def started_request_message(request) # :doc:
sprintf('Started %s "%s" for %s at %s',
request.raw_request_method,
request.filtered_path,
request.remote_ip,
Time.now)
end
def compute_tags(request) # :doc:
@taggers.collect do |tag|
case tag
when Proc
tag.call(request)
when Symbol
request.send(tag)
else
tag
end
end
end
def logger
Rails.logger
end
end
end
end