40 lines
1.5 KiB
Ruby
40 lines
1.5 KiB
Ruby
module Elasticsearch
|
|
module Rails
|
|
module Lograge
|
|
|
|
# Rails initializer class to require Elasticsearch::Rails::Instrumentation files,
|
|
# set up Elasticsearch::Model and add Lograge configuration to display Elasticsearch-related duration
|
|
#
|
|
# Require the component in your `application.rb` file and enable Lograge:
|
|
#
|
|
# require 'elasticsearch/rails/lograge'
|
|
#
|
|
# You should see the full duration of the request to Elasticsearch as part of each log event:
|
|
#
|
|
# method=GET path=/search ... status=200 duration=380.89 view=99.64 db=0.00 es=279.37
|
|
#
|
|
# @see https://github.com/roidrage/lograge
|
|
#
|
|
class Railtie < ::Rails::Railtie
|
|
initializer "elasticsearch.lograge" do |app|
|
|
require 'elasticsearch/rails/instrumentation/publishers'
|
|
require 'elasticsearch/rails/instrumentation/log_subscriber'
|
|
require 'elasticsearch/rails/instrumentation/controller_runtime'
|
|
|
|
Elasticsearch::Model::Searching::SearchRequest.class_eval do
|
|
include Elasticsearch::Rails::Instrumentation::Publishers::SearchRequest
|
|
end if defined?(Elasticsearch::Model::Searching::SearchRequest)
|
|
|
|
ActiveSupport.on_load(:action_controller) do
|
|
include Elasticsearch::Rails::Instrumentation::ControllerRuntime
|
|
end
|
|
|
|
config.lograge.custom_options = lambda do |event|
|
|
{ es: event.payload[:elasticsearch_runtime].to_f.round(2) }
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|