Sha256: deab596750294160fc299120cb7789fd44cc3f35369b9517edbfdbd1937d682c
Contents?: true
Size: 1.93 KB
Versions: 13
Compression:
Stored size: 1.93 KB
Contents
# frozen_string_literal: true
require_relative '../../metadata/ext'
require_relative '../analytics'
require_relative 'ext'
require_relative 'utils'
module Datadog
module Tracing
module Contrib
module Sequel
# Adds instrumentation to Sequel::Database
module Database
def self.included(base)
base.prepend(InstanceMethods)
end
# Instance methods for instrumenting Sequel::Database
module InstanceMethods
def run(sql, options = ::Sequel::OPTS)
opts = parse_opts(sql, options)
response = nil
Tracing.trace(Ext::SPAN_QUERY) do |span|
span.service = Datadog.configuration_for(self, :service_name) \
|| Datadog.configuration.tracing[:sequel][:service_name] \
|| Contrib::SpanAttributeSchema.fetch_service_name(
'',
adapter_name
)
span.resource = opts[:query]
span.type = Tracing::Metadata::Ext::SQL::TYPE
Utils.set_common_tags(span, self)
span.set_tag(Ext::TAG_DB_VENDOR, adapter_name)
response = super(sql, options)
end
response
end
private
def adapter_name
Utils.adapter_name(self)
end
def parse_opts(sql, opts)
db_opts = if ::Sequel::VERSION < '3.41.0' && self.class.to_s !~ /Dataset$/
@opts
elsif instance_variable_defined?(:@pool) && @pool
@pool.db.opts
end
sql = sql.is_a?(::Sequel::SQL::Expression) ? literal(sql) : sql.to_s
Utils.parse_opts(sql, opts, db_opts)
end
end
end
end
end
end
end
Version data entries
13 entries across 13 versions & 1 rubygems