Module: NewRelic::Security::Instrumentation::Mysql2::Client::Chain

Defined in:
lib/newrelic_security/instrumentation-security/mysql2/chain.rb

Class Method Summary collapse

Class Method Details

.instrument!Object



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# File 'lib/newrelic_security/instrumentation-security/mysql2/chain.rb', line 7

def self.instrument!
  ::Mysql2::Client.class_eval do
    include NewRelic::Security::Instrumentation::Mysql2::Client

    alias_method :query_without_security, :query

    def query(sql, options = {})
      retval = nil
      event = query_on_enter(sql, options) { retval = query_without_security(sql, options) }
      query_on_exit(event) { return retval }
    end

    alias_method :prepare_without_security, :prepare
    
    def prepare(sql)
      retval = nil
      event = prepare_on_enter(sql) { retval = prepare_without_security(sql) }
      prepare_on_exit(event, retval, sql) { return retval }
    end
    
  end
end