oboe 2.7.0.3-java

data/lib/oboe/frameworks/rails/inst/connection_adapters/mysql.rb

@@ -0,0 +1,43 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Inst
+    module ConnectionAdapters
+      module FlavorInitializers
+        def self.mysql
+          Oboe.logger.info "[oboe/loading] Instrumenting activerecord mysqladapter" if Oboe::Config[:verbose]
+
+          # ActiveRecord 3.2 and higher
+          if (::ActiveRecord::VERSION::MAJOR == 3 and ::ActiveRecord::VERSION::MINOR >= 2) or
+              ::ActiveRecord::VERSION::MAJOR == 4
+
+            # AbstractMysqlAdapter
+            Oboe::Util.send_include(::ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter,
+                                      Oboe::Inst::ConnectionAdapters::Utils)
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter, :execute)
+
+            # MysqlAdapter
+            Oboe::Util.send_include(::ActiveRecord::ConnectionAdapters::MysqlAdapter,
+                                      Oboe::Inst::ConnectionAdapters::Utils)
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::MysqlAdapter, :exec_query)
+
+          else
+            # ActiveRecord 3.1 and below
+
+            # MysqlAdapter
+            Oboe::Util.send_include(::ActiveRecord::ConnectionAdapters::MysqlAdapter,
+                                      Oboe::Inst::ConnectionAdapters::Utils)
+
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::MysqlAdapter, :execute)
+
+            if ::ActiveRecord::VERSION::MAJOR == 3 and ::ActiveRecord::VERSION::MINOR == 1
+              Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::MysqlAdapter, :begin_db_transaction)
+              Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::MysqlAdapter, :exec_delete)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/oboe/frameworks/rails/inst/connection_adapters/mysql2.rb

@@ -0,0 +1,28 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Inst
+    module ConnectionAdapters
+      module FlavorInitializers
+        def self.mysql2
+          Oboe.logger.info "[oboe/loading] Instrumenting activerecord mysql2adapter" if Oboe::Config[:verbose]
+
+          Oboe::Util.send_include(::ActiveRecord::ConnectionAdapters::Mysql2Adapter,
+                                    Oboe::Inst::ConnectionAdapters::Utils)
+
+          if (::ActiveRecord::VERSION::MAJOR == 3 and ::ActiveRecord::VERSION::MINOR == 0) or
+              ::ActiveRecord::VERSION::MAJOR == 2
+            # ActiveRecord 3.0 and prior
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::Mysql2Adapter, :execute)
+          else
+            # ActiveRecord 3.1 and above
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::Mysql2Adapter, :exec_insert)
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::Mysql2Adapter, :exec_query)
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::Mysql2Adapter, :exec_delete)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/oboe/frameworks/rails/inst/connection_adapters/oracle.rb

@@ -0,0 +1,18 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Inst
+    module ConnectionAdapters
+      module FlavorInitializers
+        def self.oracle
+          Oboe.logger.info "[oboe/loading] Instrumenting activerecord oracleenhancedadapter" if Oboe::Config[:verbose]
+          ::ActiveRecord::ConnectionAdapters::OracleEnhancedAdapter.module_eval do
+            include Oboe::Inst::ConnectionAdapters
+          end if defined?(::ActiveRecord::ConnectionAdapters::OracleEnhancedAdapter)
+        end
+      end
+    end
+  end
+end
+

data/lib/oboe/frameworks/rails/inst/connection_adapters/postgresql.rb

@@ -0,0 +1,30 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Inst
+    module ConnectionAdapters
+      module FlavorInitializers
+        def self.postgresql
+
+          Oboe.logger.info "[oboe/loading] Instrumenting activerecord postgresqladapter" if Oboe::Config[:verbose]
+
+          Oboe::Util.send_include(::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter,
+                                    Oboe::Inst::ConnectionAdapters::Utils)
+
+          if (::ActiveRecord::VERSION::MAJOR == 3 and ::ActiveRecord::VERSION::MINOR > 0) or
+                ::ActiveRecord::VERSION::MAJOR == 4
+
+            # ActiveRecord 3.1 and up
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter, :exec_query)
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter, :exec_delete)
+
+          else
+            # ActiveRecord 3.0 and prior
+            Oboe::Util.method_alias(::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter, :execute)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/oboe/frameworks/rails/inst/connection_adapters/utils.rb

@@ -0,0 +1,118 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Inst
+    module ConnectionAdapters
+      module Utils
+
+        def extract_trace_details(sql, name = nil, binds = [])
+          opts = {}
+
+          begin
+            if Oboe::Config[:sanitize_sql]
+              # Sanitize SQL and don't report binds
+              opts[:Query] = sql.gsub(/\'[\s\S][^\']*\'/, '?')
+            else
+              # Report raw SQL and any binds if they exist
+              opts[:Query] = sql.to_s
+              opts[:QueryArgs] = binds.map { |col, val| type_cast(val, col) } unless binds.empty?
+            end
+
+            opts[:Name] = name.to_s if name
+            opts[:Backtrace] = Oboe::API.backtrace if Oboe::Config[:active_record][:collect_backtraces]
+
+            if ::Rails::VERSION::MAJOR == 2
+              config = ::Rails.configuration.database_configuration[::Rails.env]
+            else
+              config = ::Rails.application.config.database_configuration[::Rails.env]
+            end
+
+            opts[:Database]   = config["database"] if config.has_key?("database")
+            opts[:RemoteHost] = config["host"]     if config.has_key?("host")
+            opts[:Flavor]     = config["adapter"]  if config.has_key?("adapter")
+          rescue StandardError => e
+            Oboe.logger.debug "Exception raised capturing ActiveRecord KVs: #{e.inspect}"
+            Oboe.logger.debug e.backtrace.join("\n")
+          end
+
+          return opts || {}
+        end
+
+        # We don't want to trace framework caches.  Only instrument SQL that
+        # directly hits the database.
+        def ignore_payload?(name)
+          %w(SCHEMA EXPLAIN CACHE).include? name.to_s or
+            (name and name.to_sym == :skip_logging) or
+              name == "ActiveRecord::SchemaMigration Load"
+        end
+
+        #def cfg
+        #  @config
+        #end
+
+        def execute_with_oboe(sql, name = nil)
+          if Oboe.tracing? and !ignore_payload?(name)
+
+            opts = extract_trace_details(sql, name)
+            Oboe::API.trace('activerecord', opts || {}) do
+              execute_without_oboe(sql, name)
+            end
+          else
+            execute_without_oboe(sql, name)
+          end
+        end
+
+        def exec_query_with_oboe(sql, name = nil, binds = [])
+          if Oboe.tracing? and !ignore_payload?(name)
+
+            opts = extract_trace_details(sql, name, binds)
+            Oboe::API.trace('activerecord', opts || {}) do
+              exec_query_without_oboe(sql, name, binds)
+            end
+          else
+            exec_query_without_oboe(sql, name, binds)
+          end
+        end
+
+        def exec_delete_with_oboe(sql, name = nil, binds = [])
+          if Oboe.tracing? and !ignore_payload?(name)
+
+            opts = extract_trace_details(sql, name, binds)
+            Oboe::API.trace('activerecord', opts || {}) do
+              exec_delete_without_oboe(sql, name, binds)
+            end
+          else
+            exec_delete_without_oboe(sql, name, binds)
+          end
+        end
+
+        def exec_insert_with_oboe(sql, name = nil, binds = [], *args)
+          if Oboe.tracing? and !ignore_payload?(name)
+
+            opts = extract_trace_details(sql, name, binds)
+            Oboe::API.trace('activerecord', opts || {}) do
+              exec_insert_without_oboe(sql, name, binds, *args)
+            end
+          else
+            exec_insert_without_oboe(sql, name, binds, *args)
+          end
+        end
+
+        def begin_db_transaction_with_oboe()
+          if Oboe.tracing?
+            opts = {}
+
+            opts[:Query] = "BEGIN"
+            Oboe::API.trace('activerecord', opts || {}) do
+              begin_db_transaction_without_oboe()
+            end
+          else
+            begin_db_transaction_without_oboe()
+          end
+        end
+      end # Utils
+    end
+  end
+end
+

data/lib/oboe/frameworks/sinatra.rb

@@ -0,0 +1,96 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Sinatra
+    module Base
+      def self.included(klass)
+        ::Oboe::Util.method_alias(klass, :dispatch!,         ::Sinatra::Base)
+        ::Oboe::Util.method_alias(klass, :handle_exception!, ::Sinatra::Base)
+      end
+
+      def dispatch_with_oboe
+        if Oboe.tracing?
+          report_kvs = {}
+
+          report_kvs[:Controller] = self.class
+          report_kvs[:Action] = env['PATH_INFO']
+
+          # Fall back to the raw tracing API so we can pass KVs
+          # back on exit (a limitation of the Oboe::API.trace
+          # block method) This removes the need for an info
+          # event to send additonal KVs
+          ::Oboe::API.log_entry('sinatra', {})
+
+          begin
+            dispatch_without_oboe
+          ensure
+            ::Oboe::API.log_exit('sinatra', report_kvs)
+          end
+        else
+          dispatch_without_oboe
+        end
+      end
+
+      def handle_exception_with_oboe(boom)
+        Oboe::API.log_exception(nil, boom) if Oboe.tracing?
+        handle_exception_without_oboe(boom)
+      end
+
+      @@rum_xhr_tmpl = File.read(File.dirname(__FILE__) + '/rails/helpers/rum/rum_ajax_header.js.erb')
+      @@rum_hdr_tmpl = File.read(File.dirname(__FILE__) + '/rails/helpers/rum/rum_header.js.erb')
+      @@rum_ftr_tmpl = File.read(File.dirname(__FILE__) + '/rails/helpers/rum/rum_footer.js.erb')
+
+      def oboe_rum_header
+        return unless Oboe::Config.rum_id
+        if Oboe.tracing?
+          if request.xhr?
+            return ERB.new(@@rum_xhr_tmpl).result
+          else
+            return ERB.new(@@rum_hdr_tmpl).result
+          end
+        end
+      rescue StandardError => e
+        Oboe.logger.warn "oboe_rum_header: #{e.message}."
+        return ""
+      end
+
+      def oboe_rum_footer
+        return unless Oboe::Config.rum_id
+        if Oboe.tracing?
+          # Even though the footer template is named xxxx.erb, there are no ERB tags in it so we'll
+          # skip that step for now
+          return @@rum_ftr_tmpl
+        end
+      rescue StandardError => e
+        Oboe.logger.warn "oboe_rum_footer: #{e.message}."
+        return ""
+      end
+    end
+  end
+end
+
+if defined?(::Sinatra)
+  require 'oboe/inst/rack'
+  require 'oboe/frameworks/sinatra/templates'
+
+  Oboe.logger.info "[oboe/loading] Instrumenting Sinatra" if Oboe::Config[:verbose]
+
+  Oboe::Loading.load_access_key
+  Oboe::Inst.load_instrumentation
+
+  ::Sinatra::Base.use Oboe::Rack
+
+  # When in the gem TEST environment, we load this instrumentation regardless.
+  # Otherwise, only when Padrino isn't around.
+  unless defined?(::Padrino) and not (ENV.has_key?("OBOE_GEM_TEST"))
+    # Padrino has 'enhanced' routes and rendering so the Sinatra
+    # instrumentation won't work anyways.  Only load for pure Sinatra apps.
+    ::Oboe::Util.send_include(::Sinatra::Base,      ::Oboe::Sinatra::Base)
+    ::Oboe::Util.send_include(::Sinatra::Templates, ::Oboe::Sinatra::Templates)
+
+    # Report __Init after fork when in Heroku
+    Oboe::API.report_init unless Oboe.heroku?
+  end
+end
+

data/lib/oboe/frameworks/sinatra/templates.rb

@@ -0,0 +1,56 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Sinatra
+    module Templates
+      def self.included(klass)
+        ::Oboe::Util.method_alias(klass, :render, ::Sinatra::Templates)
+      end
+
+      def render_with_oboe(engine, data, options = {}, locals = {}, &block)
+        if Oboe.tracing?
+          report_kvs = {}
+
+          report_kvs[:engine] = engine
+          report_kvs[:template] = data
+
+          if Oboe.tracing_layer_op?('render')
+            # For recursive calls to :render (for sub-partials and layouts),
+            # use method profiling.
+            begin
+              name = data
+              report_kvs[:FunctionName] = :render
+              report_kvs[:Class]        = :Templates
+              report_kvs[:Module]       = 'Sinatra::Templates'
+              report_kvs[:File]         = __FILE__
+              report_kvs[:LineNumber]   = __LINE__
+            rescue StandardError => e
+              ::Oboe.logger.debug e.message
+              ::Oboe.logger.debug e.backtrace.join(", ")
+            end
+
+            Oboe::API.profile(name, report_kvs, false) do
+              render_without_oboe(engine, data, options, locals, &block)
+            end
+
+          else
+            # Fall back to the raw tracing API so we can pass KVs
+            # back on exit (a limitation of the Oboe::API.trace
+            # block method) This removes the need for an info
+            # event to send additonal KVs
+            ::Oboe::API.log_entry('render', {}, 'render')
+
+            begin
+              render_without_oboe(engine, data, options, locals, &block)
+            ensure
+              ::Oboe::API.log_exit('render', report_kvs)
+            end
+          end
+        else
+          render_without_oboe(engine, data, options, locals, &block)
+        end
+      end
+    end
+  end
+end

data/lib/oboe/inst/cassandra.rb

@@ -0,0 +1,281 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module Oboe
+  module Inst
+    module Cassandra
+      def extract_trace_details(op, column_family, keys, args, options = {})
+        report_kvs = {}
+
+        begin
+          report_kvs[:Op] = op.to_s
+          report_kvs[:Cf] = column_family.to_s if column_family
+          report_kvs[:Key] = keys.inspect if keys
+
+          # Open issue - how to handle multiple Cassandra servers
+          report_kvs[:RemoteHost], report_kvs[:RemotePort] = @servers.first.split(":")
+
+          report_kvs[:Backtrace] = Oboe::API.backtrace if Oboe::Config[:cassandra][:collect_backtraces]
+
+          if options.empty? and args.is_a?(Array)
+            options = args.last if args.last.is_a?(Hash)
+          end
+
+          unless options.empty?
+            [:start_key, :finish_key, :key_count, :batch_size, :columns, :count, :start,
+             :stop, :finish, :finished, :reversed, :consistency, :ttl].each do |k|
+              report_kvs[k.to_s.capitalize] = options[k] if options.has_key?(k)
+            end
+
+            if op == :get_indexed_slices
+              index_clause = columns_and_options[:index_clause] || {}
+              unless index_clause.empty?
+                [:column_name, :value, :comparison].each do |k|
+                  report_kvs[k.to_s.capitalize] = index_clause[k] if index_clause.has_key?(k)
+                end
+              end
+            end
+          end
+        rescue
+        end
+
+        report_kvs
+      end
+
+      def insert_with_oboe(column_family, key, hash, options = {})
+        return insert_without_oboe(column_family, key, hash, options = {}) unless Oboe.tracing?
+
+        report_kvs = extract_trace_details(:insert, column_family, key, hash, options)
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          insert_without_oboe(column_family, key, hash, options = {})
+        end
+      end
+
+      def remove_with_oboe(column_family, key, *columns_and_options)
+        return send :remove_without_oboe, *args unless Oboe.tracing?
+
+        args = [column_family, key] + columns_and_options
+        report_kvs = extract_trace_details(:remove, column_family, key, columns_and_options)
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          send :remove_without_oboe, *args
+        end
+      end
+
+      def count_columns_with_oboe(column_family, key, *columns_and_options)
+        return send :count_columns_without_oboe, *args unless Oboe.tracing?
+
+        args = [column_family, key] + columns_and_options
+        report_kvs = extract_trace_details(:count_columns, column_family, key, columns_and_options)
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          send :count_columns_without_oboe, *args
+        end
+      end
+
+      def get_columns_with_oboe(column_family, key, *columns_and_options)
+        args = [column_family, key] + columns_and_options
+
+        if Oboe.tracing? and not Oboe.tracing_layer_op?(:multi_get_columns)
+          report_kvs = extract_trace_details(:get_columns, column_family, key, columns_and_options)
+
+          Oboe::API.trace('cassandra', report_kvs) do
+            send :get_columns_without_oboe, *args
+          end
+        else
+          send :get_columns_without_oboe, *args
+        end
+      end
+
+      def multi_get_columns_with_oboe(column_family, key, *columns_and_options)
+        return send :multi_get_columns_without_oboe, *args unless Oboe.tracing?
+
+        args = [column_family, key] + columns_and_options
+        report_kvs = extract_trace_details(:multi_get_columns, column_family, key, columns_and_options)
+
+        Oboe::API.trace('cassandra', report_kvs, :multi_get_columns) do
+          send :multi_get_columns_without_oboe, *args
+        end
+      end
+
+      def get_with_oboe(column_family, key, *columns_and_options)
+        return send :get_without_oboe, *args unless Oboe.tracing?
+
+        args = [column_family, key] + columns_and_options
+        report_kvs = extract_trace_details(:get, column_family, key, columns_and_options)
+
+        Oboe::API.trace('cassandra', report_kvs, :get) do
+          send :get_without_oboe, *args
+        end
+      end
+
+      def multi_get_with_oboe(column_family, key, *columns_and_options)
+        args = [column_family, key] + columns_and_options
+
+        if Oboe.tracing? and not Oboe.tracing_layer_op?(:get)
+          report_kvs = extract_trace_details(:multi_get, column_family, key, columns_and_options)
+
+          Oboe::API.trace('cassandra', report_kvs) do
+            send :multi_get_without_oboe, *args
+          end
+        else
+          send :multi_get_without_oboe, *args
+        end
+      end
+
+      def exists_with_oboe?(column_family, key, *columns_and_options)
+        return send :exists_without_oboe?, *args unless Oboe.tracing?
+
+        args = [column_family, key] + columns_and_options
+        report_kvs = extract_trace_details(:exists?, column_family, key, columns_and_options)
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          send :exists_without_oboe?, *args
+        end
+      end
+
+      def get_range_single_with_oboe(column_family, options = {})
+        if Oboe.tracing? and not Oboe.tracing_layer_op?(:get_range_batch)
+          report_kvs = extract_trace_details(:get_range_single, column_family, nil, nil)
+          args = [column_family, options]
+
+          Oboe::API.trace('cassandra', report_kvs) do
+            get_range_single_without_oboe(column_family, options)
+          end
+        else
+          get_range_single_without_oboe(column_family, options)
+        end
+      end
+
+      def get_range_batch_with_oboe(column_family, options = {})
+        return get_range_batch_without_oboe(column_family, options) unless Oboe.tracing?
+
+        report_kvs = extract_trace_details(:get_range_batch, column_family, nil, nil)
+        args = [column_family, options]
+
+        Oboe::API.trace('cassandra', report_kvs, :get_range_batch) do
+          get_range_batch_without_oboe(column_family, options)
+        end
+      end
+
+      def get_indexed_slices_with_oboe(column_family, index_clause, *columns_and_options)
+        return send :get_indexed_slices_without_oboe, *args unless Oboe.tracing?
+
+        args = [column_family, index_clause] + columns_and_options
+        report_kvs = extract_trace_details(:get_indexed_slices, column_family, nil, columns_and_options)
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          send :get_indexed_slices_without_oboe, *args
+        end
+      end
+
+      def create_index_with_oboe(keyspace, column_family, column_name, validation_class)
+        unless Oboe.tracing?
+          return create_index_without_oboe(keyspace, column_family, column_name, validation_class)
+        end
+
+        report_kvs = extract_trace_details(:create_index, column_family, nil, nil)
+        begin
+          report_kvs[:Keyspace] = keyspace.to_s
+          report_kvs[:Column_name] = column_name.to_s
+          report_kvs[:Validation_class] = validation_class.to_s
+        rescue
+        end
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          create_index_without_oboe(keyspace, column_family, column_name, validation_class)
+        end
+      end
+
+      def drop_index_with_oboe(keyspace, column_family, column_name)
+        return drop_index_without_oboe(keyspace, column_family, column_name) unless Oboe.tracing?
+
+        report_kvs = extract_trace_details(:drop_index, column_family, nil, nil)
+        begin
+          report_kvs[:Keyspace] = keyspace.to_s
+          report_kvs[:Column_name] = column_name.to_s
+        rescue
+        end
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          drop_index_without_oboe(keyspace, column_family, column_name)
+        end
+      end
+
+      def add_column_family_with_oboe(cf_def)
+        return add_column_family_without_oboe(cf_def) unless Oboe.tracing?
+
+        report_kvs = extract_trace_details(:add_column_family, nil, nil, nil)
+        begin
+          report_kvs[:Cf] = cf_def[:name] if cf_def.is_a?(Hash) and cf_def.has_key?(:name)
+        rescue
+        end
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          add_column_family_without_oboe(cf_def)
+        end
+      end
+
+      def drop_column_family_with_oboe(column_family)
+        return drop_column_family_without_oboe(column_family) unless Oboe.tracing?
+
+        report_kvs = extract_trace_details(:drop_column_family, column_family, nil, nil)
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          drop_column_family_without_oboe(column_family)
+        end
+      end
+
+      def add_keyspace_with_oboe(ks_def)
+        return add_keyspace_without_oboe(ks_def) unless Oboe.tracing?
+
+        report_kvs = extract_trace_details(:add_keyspace, nil, nil, nil)
+        report_kvs[:Name] = ks_def.name rescue ""
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          add_keyspace_without_oboe(ks_def)
+        end
+      end
+
+      def drop_keyspace_with_oboe(keyspace)
+        return drop_keyspace_without_oboe(keyspace) unless Oboe.tracing?
+
+        report_kvs = extract_trace_details(:drop_keyspace, nil, nil, nil)
+        report_kvs[:Name] = keyspace.to_s rescue ""
+
+        Oboe::API.trace('cassandra', report_kvs) do
+          drop_keyspace_without_oboe(keyspace)
+        end
+      end
+    end
+  end
+end
+
+if defined?(::Cassandra) and Oboe::Config[:cassandra][:enabled]
+  Oboe.logger.info "[oboe/loading] Instrumenting cassandra" if Oboe::Config[:verbose]
+
+  class ::Cassandra
+    include Oboe::Inst::Cassandra
+
+    [ :insert, :remove, :count_columns, :get_columns, :multi_get_columns, :get,
+      :multi_get, :get_range_single, :get_range_batch, :get_indexed_slices,
+      :create_index, :drop_index, :add_column_family, :drop_column_family,
+      :add_keyspace, :drop_keyspace].each do |m|
+      if method_defined?(m)
+        class_eval "alias #{m}_without_oboe #{m}"
+        class_eval "alias #{m} #{m}_with_oboe"
+      else Oboe.logger.warn "[oboe/loading] Couldn't properly instrument Cassandra (#{m}).  Partial traces may occur."
+      end
+    end
+
+    # Special case handler for question mark methods
+    if method_defined?(:exists?)
+      alias exists_without_oboe? exists?
+      alias exists? exists_with_oboe?
+    else Oboe.logger.warn "[oboe/loading] Couldn't properly instrument Cassandra (exists?).  Partial traces may occur."
+    end
+  end # class Cassandra
+end
+
+