newrelic-vertica 0.0.1 → 0.0.2

data/lib/newrelic-vertica/version.rb

@@ -1,5 +1,5 @@
 module Newrelic
   module Vertica
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/lib/newrelic-vertica/vertica.rb

@@ -1,14 +1,44 @@
 require 'new_relic/agent/method_tracer'
+require 'benchmark'
+
+module NewRelic
+  module Agent
+    module Instrumentation
+      module VerticaInstrumentation
+
+        def self.included(klass)
+          klass.class_eval do
+            alias_method :query_without_instrumentation, :query
+            alias_method :query, :query_with_instrumentation
+          end
+        end
+
+        def query_with_instrumentation(sql, options = {}, &block)
+          result = nil
+          duration = Benchmark.realtime do
+            result = query_without_instrumentation(sql, options, &block)
+          end
+
+          if NewRelic::Agent.is_execution_traced?
+            NewRelic::Agent.instance.transaction_sampler.notice_sql(sql, nil, duration)
+          end
+
+          return result
+        end
+      end
+    end
+  end
+end
 
 DependencyDetection.defer do
   depends_on do
-    defined?(::Vertica)
+    defined?(::Vertica) && defined?(::Vertica::Connection)
   end
 
   executes do
-    ::Vertica::Connection.class_eval do
-      add_method_tracer :query, 'Database/Vertica/query'
-      add_method_tracer :copy, 'Database/Vertica/copy'
+    ::Vertica::Connection.instance_eval do
+      include ::NewRelic::Agent::Instrumentation::VerticaInstrumentation
     end
   end
 end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: newrelic-vertica
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors: