Sha256: 90e92abfe34f061440d9006cb11c0183db526cc8d1d543530c3e678860c315a5

Contents?: true

Size: 1.44 KB

Versions: 9

Compression:

Stored size: 1.44 KB

Contents

module Appsignal
  class Hooks
    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

9 entries across 9 versions & 1 rubygems

Version Path
appsignal-2.0.6 lib/appsignal/hooks/sequel.rb
appsignal-2.0.5 lib/appsignal/hooks/sequel.rb
appsignal-2.0.5.beta.1 lib/appsignal/hooks/sequel.rb
appsignal-2.1.0.alpha.3 lib/appsignal/hooks/sequel.rb
appsignal-2.1.0.alpha.2 lib/appsignal/hooks/sequel.rb
appsignal-2.1.0.alpha.1 lib/appsignal/hooks/sequel.rb
appsignal-2.0.4 lib/appsignal/hooks/sequel.rb
appsignal-2.0.3 lib/appsignal/hooks/sequel.rb
appsignal-2.0.2 lib/appsignal/hooks/sequel.rb