traceview 3.0.0

data/lib/traceview/inst/memcached.rb

@@ -0,0 +1,94 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module TraceView
+  module Inst
+    module Memcached
+      include TraceView::API::Memcache
+
+      def self.included(cls)
+        TraceView.logger.info '[traceview/loading] Instrumenting memcached' if TraceView::Config[:verbose]
+
+        cls.class_eval do
+          MEMCACHE_OPS.reject { |m| !method_defined?(m) }.each do |m|
+            define_method("#{m}_with_traceview") do |*args|
+              opts = { :KVOp => m }
+
+              if args.length && !args[0].is_a?(Array)
+                opts[:KVKey] = args[0].to_s
+                rhost = remote_host(args[0].to_s)
+                opts[:RemoteHost] = rhost if rhost
+              end
+
+              TraceView::API.trace('memcache', opts) do
+                result = send("#{m}_without_traceview", *args)
+
+                info_kvs = {}
+                info_kvs[:KVHit] = memcache_hit?(result) if m == :get && args.length && args[0].class == String
+                info_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:memcached][:collect_backtraces]
+
+                TraceView::API.log('memcache', 'info', info_kvs) unless info_kvs.empty?
+                result
+              end
+            end
+
+            class_eval "alias #{m}_without_traceview #{m}"
+            class_eval "alias #{m} #{m}_with_traceview"
+          end
+        end
+      end
+
+    end # module Memcached
+
+    module MemcachedRails
+      def self.included(cls)
+        cls.class_eval do
+          if ::Memcached::Rails.method_defined? :get_multi
+            alias get_multi_without_traceview get_multi
+            alias get_multi get_multi_with_traceview
+          elsif TraceView::Config[:verbose]
+            TraceView.logger.warn '[traceview/loading] Couldn\'t properly instrument Memcached.  Partial traces may occur.'
+          end
+        end
+      end
+
+      def get_multi_with_traceview(keys, raw = false)
+        if TraceView.tracing?
+          layer_kvs = {}
+          layer_kvs[:KVOp] = :get_multi
+
+          TraceView::API.trace('memcache', layer_kvs || {}, :get_multi) do
+            begin
+              info_kvs = {}
+              info_kvs[:KVKeyCount] = keys.flatten.length
+
+              values = get_multi_without_traceview(keys, raw)
+
+              info_kvs[:KVHitCount] = values.length
+              info_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:memcached][:collect_backtraces]
+
+              TraceView::API.log('memcache', 'info', info_kvs)
+            rescue
+              values = get_multi_without_traceview(keys, raw)
+            end
+            values
+          end
+        else
+          get_multi_without_traceview(keys, raw)
+        end
+      end
+    end # module MemcachedRails
+  end # module Inst
+end # module TraceView
+
+if defined?(::Memcached) && TraceView::Config[:memcached][:enabled]
+  ::Memcached.class_eval do
+    include TraceView::Inst::Memcached
+  end
+
+  if defined?(::Memcached::Rails)
+    ::Memcached::Rails.class_eval do
+      include TraceView::Inst::MemcachedRails
+    end
+  end
+end

data/lib/traceview/inst/mongo.rb

@@ -0,0 +1,238 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+require 'json'
+
+module TraceView
+  module Inst
+    module Mongo
+      FLAVOR = 'mongodb'
+
+      # Operations for Mongo::DB
+      DB_OPS         = [:create_collection, :drop_collection]
+
+      # Operations for Mongo::Cursor
+      CURSOR_OPS     = [:count]
+
+      # Operations for Mongo::Collection
+      COLL_WRITE_OPS = [:find_and_modify, :insert, :map_reduce, :remove, :rename, :update]
+      COLL_QUERY_OPS = [:distinct, :find, :group]
+      COLL_INDEX_OPS = [:create_index, :drop_index, :drop_indexes, :ensure_index, :index_information]
+    end
+  end
+end
+
+if defined?(::Mongo) && TraceView::Config[:mongo][:enabled]
+  TraceView.logger.info '[traceview/loading] Instrumenting mongo' if TraceView::Config[:verbose]
+
+  if defined?(::Mongo::DB)
+    module ::Mongo
+      class DB
+        include TraceView::Inst::Mongo
+
+        # Instrument DB operations
+        TraceView::Inst::Mongo::DB_OPS.reject { |m| !method_defined?(m) }.each do |m|
+          define_method("#{m}_with_traceview") do |*args|
+            report_kvs = {}
+
+            begin
+              report_kvs[:Flavor] = TraceView::Inst::Mongo::FLAVOR
+
+              report_kvs[:Database] = @name
+
+              # Mongo >= 1.10 uses @client instead of @connection
+              # Avoid the nil
+              if @connection
+                report_kvs[:RemoteHost] = @connection.host
+                report_kvs[:RemotePort] = @connection.port
+              elsif @client
+                report_kvs[:RemoteHost] = @client.host
+                report_kvs[:RemotePort] = @client.port
+              end
+
+              report_kvs[:QueryOp] = m
+
+              report_kvs[:New_Collection_Name] = args[0] if m == :create_collection
+              report_kvs[:Collection] = args[0]          if m == :drop_collection
+
+              report_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:mongo][:collect_backtraces]
+            rescue
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              send("#{m}_without_traceview", *args)
+            end
+          end
+
+          class_eval "alias #{m}_without_traceview #{m}"
+          class_eval "alias #{m} #{m}_with_traceview"
+        end
+      end
+    end
+  end
+
+  if defined?(::Mongo::Cursor)
+    module ::Mongo
+      class Cursor
+        include TraceView::Inst::Mongo
+
+        # Instrument DB cursor operations
+        TraceView::Inst::Mongo::CURSOR_OPS.reject { |m| !method_defined?(m) }.each do |m|
+          define_method("#{m}_with_traceview") do |*args|
+            report_kvs = {}
+
+            begin
+              report_kvs[:Flavor] = TraceView::Inst::Mongo::FLAVOR
+
+              report_kvs[:Database] = @db.name
+              report_kvs[:RemoteHost] = @connection.host
+              report_kvs[:RemotePort] = @connection.port
+
+              report_kvs[:QueryOp] = m
+              if m == :count
+                unless @selector.empty?
+                  report_kvs[:Query] = @selector.to_json
+                else
+                  report_kvs[:Query] = 'all'
+                end
+                report_kvs[:Limit] = @limit if @limit != 0
+              end
+            rescue
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              send("#{m}_without_traceview", *args)
+            end
+          end
+
+          class_eval "alias #{m}_without_traceview #{m}"
+          class_eval "alias #{m} #{m}_with_traceview"
+        end
+      end
+    end
+  end
+
+  if defined?(::Mongo::Collection)
+    module ::Mongo
+      class Collection
+        include TraceView::Inst::Mongo
+
+        def traceview_collect(m, args)
+          begin
+            report_kvs = {}
+            report_kvs[:Flavor] = TraceView::Inst::Mongo::FLAVOR
+
+            report_kvs[:Database] = @db.name
+            report_kvs[:RemoteHost] = @db.connection.host
+            report_kvs[:RemotePort] = @db.connection.port
+            report_kvs[:Collection] = @name
+
+            report_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:mongo][:collect_backtraces]
+
+            report_kvs[:QueryOp] = m
+            report_kvs[:Query] = args[0].to_json if args && !args.empty? && args[0].class == Hash
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Exception in traceview_collect KV collection: #{e.inspect}"
+          end
+          report_kvs
+        end
+
+        # Instrument Collection write operations
+        TraceView::Inst::Mongo::COLL_WRITE_OPS.reject { |m| !method_defined?(m) }.each do |m|
+          define_method("#{m}_with_traceview") do |*args|
+            report_kvs = traceview_collect(m, args)
+            args_length = args.length
+
+            begin
+              if m == :find_and_modify && args[0] && args[0].key?(:update)
+                report_kvs[:Update_Document] = args[0][:update].inspect
+              end
+
+              if m == :map_reduce
+                report_kvs[:Map_Function]    = args[0]
+                report_kvs[:Reduce_Function] = args[1]
+                report_kvs[:Limit] = args[2][:limit] if args[2] && args[2].key?(:limit)
+              end
+
+              report_kvs[:New_Collection_Name] = args[0] if m == :rename
+
+              if m == :update
+                if args_length >= 3
+                  report_kvs[:Update_Document] = args[1].to_json
+                  report_kvs[:Multi] = args[2][:multi] if args[2] && args[2].key?(:multi)
+                end
+              end
+            rescue
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              send("#{m}_without_traceview", *args)
+            end
+          end
+
+          class_eval "alias #{m}_without_traceview #{m}"
+          class_eval "alias #{m} #{m}_with_traceview"
+        end
+
+        # Instrument Collection query operations
+        TraceView::Inst::Mongo::COLL_QUERY_OPS.reject { |m| !method_defined?(m) }.each do |m|
+          define_method("#{m}_with_traceview") do |*args, &blk|
+            begin
+              report_kvs = traceview_collect(m, args)
+              args_length = args.length
+
+              if m == :distinct && args_length >= 2
+                report_kvs[:Key]   = args[0]
+                report_kvs[:Query] = args[1].to_json if args[1] && args[1].class == Hash
+              end
+
+              if m == :find && args_length > 0
+                report_kvs[:Limit] = args[0][:limit] if !args[0].nil? && args[0].key?(:limit)
+              end
+
+              if m == :group
+                unless args.empty?
+                  if args[0].is_a?(Hash)
+                    report_kvs[:Group_Key]       = args[0][:key].to_json     if args[0].key?(:key)
+                    report_kvs[:Group_Condition] = args[0][:cond].to_json    if args[0].key?(:cond)
+                    report_kvs[:Group_Initial]   = args[0][:initial].to_json if args[0].key?(:initial)
+                    report_kvs[:Group_Reduce]    = args[0][:reduce]          if args[0].key?(:reduce)
+                  end
+                end
+              end
+            rescue
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              send("#{m}_without_traceview", *args, &blk)
+            end
+          end
+
+          class_eval "alias #{m}_without_traceview #{m}"
+          class_eval "alias #{m} #{m}_with_traceview"
+        end
+
+        # Instrument Collection index operations
+        TraceView::Inst::Mongo::COLL_INDEX_OPS.reject { |m| !method_defined?(m) }.each do |m|
+          define_method("#{m}_with_traceview") do |*args|
+            report_kvs = traceview_collect(m, args)
+
+            begin
+              if [:create_index, :ensure_index, :drop_index].include?(m) && !args.empty?
+                report_kvs[:Index] = args[0].to_json
+              end
+            rescue
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              send("#{m}_without_traceview", *args)
+            end
+          end
+
+          class_eval "alias #{m}_without_traceview #{m}"
+          class_eval "alias #{m} #{m}_with_traceview"
+        end
+      end
+    end
+  end
+end

data/lib/traceview/inst/moped.rb

@@ -0,0 +1,474 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+require 'json'
+
+module TraceView
+  module Inst
+    module Moped
+      FLAVOR = 'mongodb'
+
+      # Moped::Database
+      DB_OPS         = [:command, :drop]
+
+      # Moped::Indexes
+      INDEX_OPS      = [:create, :drop]
+
+      # Moped::Query
+      QUERY_OPS      = [:count, :sort, :limit, :distinct, :update, :update_all, :upsert,
+                        :explain, :modify, :remove, :remove_all]
+
+      # Moped::Collection
+      COLLECTION_OPS = [:drop, :find, :indexes, :insert, :aggregate]
+    end
+  end
+end
+
+if defined?(::Moped) && TraceView::Config[:moped][:enabled]
+  TraceView.logger.info '[traceview/loading] Instrumenting moped' if TraceView::Config[:verbose]
+
+  if defined?(::Moped::Database)
+    module ::Moped
+      class Database
+        include TraceView::Inst::Moped
+
+        def extract_trace_details(op)
+          report_kvs = {}
+          begin
+            report_kvs[:Flavor] = TraceView::Inst::Moped::FLAVOR
+            # FIXME: We're only grabbing the first of potentially multiple servers here
+            if ::Moped::VERSION < '2.0.0'
+              report_kvs[:RemoteHost], report_kvs[:RemotePort] = session.cluster.seeds.first.split(':')
+            else
+              report_kvs[:RemoteHost] = session.cluster.seeds.first.address.host
+              report_kvs[:RemotePort] = session.cluster.seeds.first.address.port
+            end
+            report_kvs[:Database] = name
+            report_kvs[:QueryOp] = op.to_s
+            report_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:moped][:collect_backtraces]
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+          report_kvs
+        end
+
+        def command_with_traceview(command)
+          if TraceView.tracing? && !TraceView.layer_op && command.key?(:mapreduce)
+            begin
+              report_kvs = extract_trace_details(:map_reduce)
+              report_kvs[:Map_Function] = command[:map]
+              report_kvs[:Reduce_Function] = command[:reduce]
+            rescue StandardError => e
+              TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              command_without_traceview(command)
+            end
+          else
+            command_without_traceview(command)
+          end
+        end
+
+        def drop_with_traceview
+          return drop_without_traceview unless TraceView.tracing?
+
+          report_kvs = extract_trace_details(:drop_database)
+
+          TraceView::API.trace('mongo', report_kvs) do
+            drop_without_traceview
+          end
+        end
+
+        TraceView::Inst::Moped::DB_OPS.each do |m|
+          if method_defined?(m)
+            class_eval "alias #{m}_without_traceview #{m}"
+            class_eval "alias #{m} #{m}_with_traceview"
+          else TraceView.logger.warn "[traceview/loading] Couldn't properly instrument moped (#{m}).  Partial traces may occur."
+          end
+        end
+      end
+    end
+  end
+
+  if defined?(::Moped::Indexes)
+    module ::Moped
+      class Indexes
+        include TraceView::Inst::Moped
+
+        def extract_trace_details(op)
+          report_kvs = {}
+          begin
+            report_kvs[:Flavor] = TraceView::Inst::Moped::FLAVOR
+            # FIXME: We're only grabbing the first of potentially multiple servers here
+            if ::Moped::VERSION < '2.0.0'
+              report_kvs[:RemoteHost], report_kvs[:RemotePort] = database.session.cluster.seeds.first.split(':')
+            else
+              report_kvs[:RemoteHost] = database.session.cluster.seeds.first.address.host
+              report_kvs[:RemotePort] = database.session.cluster.seeds.first.address.port
+            end
+            report_kvs[:Database] = database.name
+            report_kvs[:QueryOp] = op.to_s
+            report_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:moped][:collect_backtraces]
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+          report_kvs
+        end
+
+        def create_with_traceview(key, options = {})
+          return create_without_traceview(key, options = {}) unless TraceView.tracing?
+
+          begin
+            # We report :create_index here to be consistent
+            # with other mongo implementations
+            report_kvs = extract_trace_details(:create_index)
+            report_kvs[:Key] = key.to_json
+            report_kvs[:Options] = options.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs, :create_index) do
+            create_without_traceview(key, options = {})
+          end
+        end
+
+        def drop_with_traceview(key = nil)
+          return drop_without_traceview(key = nil) unless TraceView.tracing?
+
+          begin
+            # We report :drop_indexes here to be consistent
+            # with other mongo implementations
+            report_kvs = extract_trace_details(:drop_indexes)
+            report_kvs[:Key] = key.nil? ? 'all' : key.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            drop_without_traceview(key = nil)
+          end
+        end
+
+        TraceView::Inst::Moped::INDEX_OPS.each do |m|
+          if method_defined?(m)
+            class_eval "alias #{m}_without_traceview #{m}"
+            class_eval "alias #{m} #{m}_with_traceview"
+          else TraceView.logger.warn "[traceview/loading] Couldn't properly instrument moped (#{m}).  Partial traces may occur."
+          end
+        end
+      end
+    end
+  end
+
+  if defined?(::Moped::Query)
+    module ::Moped
+      class Query
+        include TraceView::Inst::Moped
+
+        def extract_trace_details(op)
+          report_kvs = {}
+          begin
+            report_kvs[:Flavor] = TraceView::Inst::Moped::FLAVOR
+            # FIXME: We're only grabbing the first of potentially multiple servers here
+            if ::Moped::VERSION < '2.0.0'
+              report_kvs[:RemoteHost], report_kvs[:RemotePort] = collection.database.session.cluster.seeds.first.split(':')
+            else
+              report_kvs[:RemoteHost] = collection.database.session.cluster.seeds.first.address.host
+              report_kvs[:RemotePort] = collection.database.session.cluster.seeds.first.address.port
+            end
+            report_kvs[:Database] = collection.database.name
+            report_kvs[:Collection] = collection.name
+            report_kvs[:QueryOp] = op.to_s
+            report_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:moped][:collect_backtraces]
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+          report_kvs
+        end
+
+        def count_with_traceview
+          return count_without_traceview unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:count)
+            report_kvs[:Query] = selector.empty? ? 'all' : selector.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            count_without_traceview
+          end
+        end
+
+        def sort_with_traceview(sort)
+          return sort_without_traceview(sort) unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:sort)
+            report_kvs[:Query] = selector.empty? ? 'all' : selector.to_json
+            report_kvs[:Order] = sort.to_s
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            sort_without_traceview(sort)
+          end
+        end
+
+        def limit_with_traceview(limit)
+          if TraceView.tracing? && !TraceView.tracing_layer_op?(:explain)
+            begin
+              report_kvs = extract_trace_details(:limit)
+              report_kvs[:Query] = selector.empty? ? 'all' : selector.to_json
+              report_kvs[:Limit] = limit.to_s
+            rescue StandardError => e
+              TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              limit_without_traceview(limit)
+            end
+          else
+            limit_without_traceview(limit)
+          end
+        end
+
+        def distinct_with_traceview(key)
+          return distinct_without_traceview(key) unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:distinct)
+            report_kvs[:Query] = selector.empty? ? 'all' : selector.to_json
+            report_kvs[:Key] = key.to_s
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            distinct_without_traceview(key)
+          end
+        end
+
+        def update_with_traceview(change, flags = nil)
+          if TraceView.tracing? && !TraceView.tracing_layer_op?([:update_all, :upsert])
+            begin
+              report_kvs = extract_trace_details(:update)
+              report_kvs[:Flags] = flags.to_s if flags
+              report_kvs[:Update_Document] = change.to_json
+            rescue StandardError => e
+              TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+            end
+
+            TraceView::API.trace('mongo', report_kvs) do
+              update_without_traceview(change, flags = nil)
+            end
+          else
+            update_without_traceview(change, flags = nil)
+          end
+        end
+
+        def update_all_with_traceview(change)
+          return update_all_without_traceview(change) unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:update_all)
+            report_kvs[:Update_Document] = change.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs, :update_all) do
+            update_all_without_traceview(change)
+          end
+        end
+
+        def upsert_with_traceview(change)
+          return upsert_without_traceview(change) unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:upsert)
+            report_kvs[:Query] = selector.to_json
+            report_kvs[:Update_Document] = change.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs, :upsert) do
+            upsert_without_traceview(change)
+          end
+        end
+
+        def explain_with_traceview
+          return explain_without_traceview unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:explain)
+            report_kvs[:Query] = selector.empty? ? 'all' : selector.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs, :explain) do
+            explain_without_traceview
+          end
+        end
+
+        def modify_with_traceview(change, options = {})
+          return modify_without_traceview(change, options) unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:modify)
+            report_kvs[:Update_Document] = selector.empty? ? 'all' : selector.to_json
+            report_kvs[:Change] = change.to_json
+            report_kvs[:Options] = options.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            modify_without_traceview(change, options)
+          end
+        end
+
+        def remove_with_traceview
+          return remove_without_traceview unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:remove)
+            report_kvs[:Query] = selector.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            remove_without_traceview
+          end
+        end
+
+        def remove_all_with_traceview
+          return remove_all_without_traceview unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:remove_all)
+            report_kvs[:Query] = selector.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            remove_all_without_traceview
+          end
+        end
+
+        TraceView::Inst::Moped::QUERY_OPS.each do |m|
+          if method_defined?(m)
+            class_eval "alias #{m}_without_traceview #{m}"
+            class_eval "alias #{m} #{m}_with_traceview"
+          else TraceView.logger.warn "[traceview/loading] Couldn't properly instrument moped (#{m}).  Partial traces may occur."
+          end
+        end
+      end
+    end
+  end # ::Moped::Query
+
+  if defined?(::Moped::Collection)
+    module ::Moped
+      class Collection
+        include TraceView::Inst::Moped
+
+        def extract_trace_details(op)
+          report_kvs = {}
+          begin
+            report_kvs[:Flavor] = TraceView::Inst::Moped::FLAVOR
+            # FIXME: We're only grabbing the first of potentially multiple servers here
+            if ::Moped::VERSION < '2.0.0'
+              report_kvs[:RemoteHost], report_kvs[:RemotePort] = database.session.cluster.seeds.first.split(':')
+            else
+              report_kvs[:RemoteHost] = database.session.cluster.seeds.first.address.host
+              report_kvs[:RemotePort] = database.session.cluster.seeds.first.address.port
+            end
+            report_kvs[:Database] = database.name
+            report_kvs[:Collection] = name
+            report_kvs[:QueryOp] = op.to_s
+            report_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:moped][:collect_backtraces]
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+          report_kvs
+        end
+
+        def drop_with_traceview
+          return drop_without_traceview unless TraceView.tracing?
+
+          # We report :drop_collection here to be consistent
+          # with other mongo implementations
+          report_kvs = extract_trace_details(:drop_collection)
+
+          TraceView::API.trace('mongo', report_kvs) do
+            drop_without_traceview
+          end
+        end
+
+        def find_with_traceview(selector = {})
+          return find_without_traceview(selector) unless TraceView.tracing?
+
+          begin
+            report_kvs = extract_trace_details(:find)
+            report_kvs[:Query] = selector.empty? ? 'all' : selector.to_json
+          rescue StandardError => e
+            TraceView.logger.debug "[traceview/debug] Moped KV collection error: #{e.inspect}"
+          end
+
+          TraceView::API.trace('mongo', report_kvs) do
+            find_without_traceview(selector)
+          end
+        end
+
+        def indexes_with_traceview
+          return indexes_without_traceview unless TraceView.tracing?
+
+          report_kvs = extract_trace_details(:indexes)
+
+          TraceView::API.trace('mongo', report_kvs) do
+            indexes_without_traceview
+          end
+        end
+
+        def insert_with_traceview(documents, flags = nil)
+          if TraceView.tracing? && !TraceView.tracing_layer_op?(:create_index)
+            report_kvs = extract_trace_details(:insert)
+
+            TraceView::API.trace('mongo', report_kvs) do
+              insert_without_traceview(documents, flags)
+            end
+          else
+            insert_without_traceview(documents, flags)
+          end
+        end
+
+        def aggregate_with_traceview(*pipeline)
+          return aggregate_without_traceview(pipeline) unless TraceView.tracing?
+
+          report_kvs = extract_trace_details(:aggregate)
+          report_kvs[:Query] = pipeline
+
+          TraceView::API.trace('mongo', report_kvs) do
+            aggregate_without_traceview(pipeline)
+          end
+        end
+
+        TraceView::Inst::Moped::COLLECTION_OPS.each do |m|
+          if method_defined?(m)
+            class_eval "alias #{m}_without_traceview #{m}"
+            class_eval "alias #{m} #{m}_with_traceview"
+          else TraceView.logger.warn "[traceview/loading] Couldn't properly instrument moped (#{m}).  Partial traces may occur."
+          end
+        end
+      end
+    end
+  end # ::Moped::Collection
+end