mirror of
https://github.com/samsonjs/grape_logging.git
synced 2026-04-27 14:57:40 +00:00
pass logger as option to middleware so it's easily usable when grape is mounted inside rails
This commit is contained in:
parent
b908496d42
commit
4b662e5c23
1 changed files with 15 additions and 5 deletions
|
|
@ -4,24 +4,25 @@ module GrapeLogging
|
||||||
module Middleware
|
module Middleware
|
||||||
class RequestLogger < Grape::Middleware::Base
|
class RequestLogger < Grape::Middleware::Base
|
||||||
def before
|
def before
|
||||||
@env[:db_duration] = 0
|
env[:db_duration] = 0
|
||||||
|
|
||||||
ActiveSupport::Notifications.subscribe('sql.active_record') do |*args|
|
ActiveSupport::Notifications.subscribe('sql.active_record') do |*args|
|
||||||
event = ActiveSupport::Notifications::Event.new(*args)
|
event = ActiveSupport::Notifications::Event.new(*args)
|
||||||
@env[:db_duration] += event.duration
|
env[:db_duration] += event.duration
|
||||||
end if defined?(ActiveRecord)
|
end if defined?(ActiveRecord)
|
||||||
end
|
end
|
||||||
|
|
||||||
def call!(env)
|
def call!(env)
|
||||||
original_response = nil
|
original_response = nil
|
||||||
duration = Benchmark.realtime { original_response = super(env) }
|
|
||||||
|
|
||||||
@app.logger.info parameters(::Rack::Request.new(env), original_response, duration)
|
duration = Benchmark.realtime { original_response = super(env) }
|
||||||
|
logger.info parameters(original_response, duration)
|
||||||
|
|
||||||
original_response
|
original_response
|
||||||
end
|
end
|
||||||
|
|
||||||
protected
|
protected
|
||||||
def parameters(request, response, duration)
|
def parameters(response, duration)
|
||||||
{
|
{
|
||||||
path: request.path,
|
path: request.path,
|
||||||
params: request.params,
|
params: request.params,
|
||||||
|
|
@ -31,6 +32,15 @@ module GrapeLogging
|
||||||
status: response.first
|
status: response.first
|
||||||
}
|
}
|
||||||
end
|
end
|
||||||
|
|
||||||
|
private
|
||||||
|
def logger
|
||||||
|
@logger ||= @options[:logger] || Logger.new(STDOUT)
|
||||||
|
end
|
||||||
|
|
||||||
|
def request
|
||||||
|
@request ||= ::Rack::Request.new(env)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
Loading…
Reference in a new issue