Sha256: 939baf2dacb752446811f945d57e5e6c2f94b24dcf1317eb51c814cfc73d6a50
Contents?: true
Size: 999 Bytes
Versions: 24
Compression:
Stored size: 999 Bytes
Contents
# frozen_string_literal: true
module ElasticAPM
# @api private
module Spies
# @api private
class ElasticsearchSpy
NAME_FORMAT = '%s %s'
TYPE = 'db.elasticsearch'
# rubocop:disable Metrics/MethodLength
def install
::Elasticsearch::Transport::Client.class_eval do
alias perform_request_without_apm perform_request
def perform_request(method, path, *args, &block)
name = format(NAME_FORMAT, method, path)
statement = args[0].is_a?(String) ? args[0] : args[0].to_json
context = Span::Context.new(db: { statement: statement })
ElasticAPM.with_span name, TYPE, context: context do
perform_request_without_apm(method, path, *args, &block)
end
end
end
end
# rubocop:enable Metrics/MethodLength
end
register(
'Elasticsearch::Transport::Client',
'elasticsearch-transport',
ElasticsearchSpy.new
)
end
end
Version data entries
24 entries across 24 versions & 1 rubygems