Contents

# encoding: utf-8

require 'one_apm/agent/datastore/mongo/obfuscator'

module OneApm
  module Agent
    module Datastore
      module Mongo
        module StatementFormatter

          OA_PLAINTEXT_KEYS = [
            :database,
            :collection,
            :operation,
            :fields,
            :skip,
            :limit,
            :order
          ]

          OA_OBFUSCATE_KEYS = [
            :selector
          ]

          def self.format(statement, operation)
            return nil unless OneApm::Manager.config[:'mongo.capture_queries']

            result = { :operation => operation }

            OA_PLAINTEXT_KEYS.each do |key|
              result[key] = statement[key] if statement.key?(key)
            end

            OA_OBFUSCATE_KEYS.each do |key|
              if statement.key?(key) && statement[key]
                obfuscated = obfuscate(statement[key])
                result[key] = obfuscated if obfuscated
              end
            end
            result
          end

          def self.obfuscate(statement)
            statement = Obfuscator.obfuscate_statement(statement) if OneApm::Manager.config[:'mongo.obfuscate_queries']
            statement
          end
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
oneapm_rpm-1.4.2	lib/one_apm/agent/datastore/mongo/statement_formatter.rb
oneapm_rpm-1.4.1	lib/one_apm/agent/datastore/mongo/statement_formatter.rb
oneapm_rpm-1.4.0	lib/one_apm/agent/datastore/mongo/statement_formatter.rb
oneapm_rpm-1.3.7	lib/one_apm/agent/datastore/mongo/statement_formatter.rb
oneapm_rpm-1.3.7.rc1	lib/one_apm/agent/datastore/mongo/statement_formatter.rb