Sha256: 80229f7827f9a42687889acc71f8fcccfc0ad7044a75a834fb94a59fb6f7b8af
Contents?: true
Size: 1.13 KB
Versions: 12
Compression:
Stored size: 1.13 KB
Contents
# frozen_string_literal: true
require 'elastic_apm/sql_summarizer'
module ElasticAPM
# @api private
module Spies
# @api private
class SequelSpy
TYPE = 'db.sequel.sql'.freeze
def self.summarizer
@summarizer ||= SqlSummarizer.new
end
# rubocop:disable Metrics/MethodLength
def install
require 'sequel/database/logging'
::Sequel::Database.class_eval do
alias log_connection_yield_without_apm log_connection_yield
def log_connection_yield(sql, *args, &block)
unless ElasticAPM.current_transaction
return log_connection_yield_without_apm(sql, *args, &block)
end
summarizer = ElasticAPM::Spies::SequelSpy.summarizer
name = summarizer.summarize sql
context = Span::Context.new(
statement: sql,
type: 'sql',
user: opts[:user]
)
ElasticAPM.span(name, TYPE, context: context, &block)
end
end
end
# rubocop:enable Metrics/MethodLength
end
register 'Sequel', 'sequel', SequelSpy.new
end
end
Version data entries
12 entries across 12 versions & 1 rubygems