Sha256: 2fbc9d79b4fe2b421167afe8f081b623bcbf7fefc1d74295fe99cac15eeebc67

Contents?: true

Size: 1.49 KB

Versions: 74

Compression:

Stored size: 1.49 KB

Contents

# frozen_string_literal: true

module Appsignal
  class Hooks
    # @api private
    module SequelLogExtension
      # Add query instrumentation
      def log_yield(sql, args = nil)
        Appsignal.instrument(
          "sql.sequel",
          nil,
          sql,
          Appsignal::EventFormatter::SQL_BODY_FORMAT
        ) do
          super
        end
      end
    end

    module SequelLogConnectionExtension
      # Add query instrumentation
      def log_connection_yield(sql, conn, args = nil)
        Appsignal.instrument(
          "sql.sequel",
          nil,
          sql,
          Appsignal::EventFormatter::SQL_BODY_FORMAT
        ) do
          super
        end
      end
    end

    class SequelHook < Appsignal::Hooks::Hook
      register :sequel

      def dependencies_present?
        defined?(::Sequel::Database) &&
          Appsignal.config &&
          Appsignal.config[:instrument_sequel]
      end

      def install
        # Register the extension...
        if ::Sequel::MAJOR >= 4 && ::Sequel::MINOR >= 35
          ::Sequel::Database.register_extension(
            :appsignal_integration,
            Appsignal::Hooks::SequelLogConnectionExtension
          )
        else
          ::Sequel::Database.register_extension(
            :appsignal_integration,
            Appsignal::Hooks::SequelLogExtension
          )
        end

        # ... and automatically add it to future instances.
        ::Sequel::Database.extension(:appsignal_integration)
      end
    end
  end
end

Version data entries

74 entries across 74 versions & 1 rubygems

Version Path
appsignal-2.9.12.beta.0-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.12.beta.0 lib/appsignal/hooks/sequel.rb
appsignal-2.9.11 lib/appsignal/hooks/sequel.rb
appsignal-2.9.11-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.10 lib/appsignal/hooks/sequel.rb
appsignal-2.9.10-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.9-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.9 lib/appsignal/hooks/sequel.rb
appsignal-2.9.8-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.8 lib/appsignal/hooks/sequel.rb
appsignal-2.9.7-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.7 lib/appsignal/hooks/sequel.rb
appsignal-2.9.6-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.6 lib/appsignal/hooks/sequel.rb
appsignal-2.9.5 lib/appsignal/hooks/sequel.rb
appsignal-2.9.5-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.4 lib/appsignal/hooks/sequel.rb
appsignal-2.9.4-java lib/appsignal/hooks/sequel.rb
appsignal-2.9.3 lib/appsignal/hooks/sequel.rb
appsignal-2.9.3-java lib/appsignal/hooks/sequel.rb