Contents

require 'sqreen/ecosystem/module_api'
require 'sqreen/ecosystem/module_api/instrumentation'
require 'sqreen/ecosystem/module_api/message_producer'
require 'sqreen/ecosystem/databases/database_connection_data'

module Sqreen
  module Ecosystem
    module Databases
      class Mongo
        include ModuleApi::Instrumentation
        include ModuleApi::MessageProducer

        def setup
          advice = wrap_for_interest(DatabaseConnectionData, &method(:after_advice))
          instrument 'Mongo::Client#initialize', after: advice
        end

        private

        # @param [Sqreen::Graft::CallbackCall] call
        def after_advice(call, _ball)
          return if call.raised

          client = call.instance
          server_addrs = client.cluster.servers.map(&:address)

          server_addrs.map do |addr|
            DatabaseConnectionData.new(
              transport: :mongo,
              host: addr.host,
              port: addr.port,
              db: client.database.name,
            )
          end
        end
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
sqreen-1.22.0-java	lib/sqreen/ecosystem/databases/mongo.rb
sqreen-1.22.0	lib/sqreen/ecosystem/databases/mongo.rb
sqreen-1.21.1-java	lib/sqreen/ecosystem/databases/mongo.rb
sqreen-1.21.1	lib/sqreen/ecosystem/databases/mongo.rb
sqreen-1.21.0-java	lib/sqreen/ecosystem/databases/mongo.rb
sqreen-1.21.0	lib/sqreen/ecosystem/databases/mongo.rb