Sha256: 050df97c6f1484aa0c371d6915eee4f730f382324e9e1a46e9e16a5e6a3764a4
Contents?: true
Size: 1.21 KB
Versions: 24
Compression:
Stored size: 1.21 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'
def self.summarizer
@summarizer ||= SqlSummarizer.new
end
def self.build_context(sql, opts)
Span::Context.new(
db: { statement: sql, type: 'sql', user: opts[:user] }
)
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 = ElasticAPM::Spies::SequelSpy.build_context(sql, opts)
ElasticAPM.with_span(name, TYPE, context: context, &block)
end
end
end
# rubocop:enable Metrics/MethodLength
end
register 'Sequel', 'sequel', SequelSpy.new
end
end
Version data entries
24 entries across 24 versions & 1 rubygems