Module: NewRelic::Security::Instrumentation::Mysql2::Statement::Chain

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

Class Method Summary collapse

Class Method Details

.instrument! ⇒ Object 



35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/newrelic_security/instrumentation-security/mysql2/chain.rb', line 35

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

    alias_method :execute_without_security, :execute

    def execute(*args, **kwargs)
      retval = nil
      event = execute_on_enter(*args, **kwargs) { retval = execute_without_security(*args, **kwargs) }
      execute_on_exit(event) { return retval }
    end
    
  end
end