traceview 3.0.0-java

data/lib/traceview/inst/memcache.rb

@@ -0,0 +1,102 @@
+# Copyright (c) 2013 AppNeta, Inc.
+# All rights reserved.
+
+module TraceView
+  module Inst
+    module MemCache
+      include TraceView::API::Memcache
+
+      def self.included(cls)
+        TraceView.logger.info '[traceview/loading] Instrumenting memcache' 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|
+              report_kvs = { :KVOp => m }
+              report_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:memcache][:collect_backtraces]
+
+              if TraceView.tracing?
+                TraceView::API.trace('memcache', report_kvs) do
+                  send("#{m}_without_traceview", *args)
+                end
+              else
+                send("#{m}_without_traceview", *args)
+              end
+            end
+
+            class_eval "alias #{m}_without_traceview #{m}"
+            class_eval "alias #{m} #{m}_with_traceview"
+          end
+        end
+
+        [:request_setup, :cache_get, :get_multi].each do |m|
+          if ::MemCache.method_defined? :request_setup
+            cls.class_eval "alias #{m}_without_traceview #{m}"
+            cls.class_eval "alias #{m} #{m}_with_traceview"
+          elsif TraceView::Config[:verbose]
+            TraceView.logger.warn "[traceview/loading] Couldn't properly instrument Memcache: #{m}"
+          end
+        end
+      end
+
+      def get_multi_with_traceview(*args)
+        return get_multi_without_traceview(args) unless TraceView.tracing?
+
+        info_kvs = {}
+
+        begin
+          info_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:memcache][:collect_backtraces]
+
+          if args.last.is_a?(Hash) || args.last.nil?
+            info_kvs[:KVKeyCount] = args.flatten.length - 1
+          else
+            info_kvs[:KVKeyCount] = args.flatten.length
+          end
+        rescue StandardError => e
+          TraceView.logger.debug "[traceview/debug] Error collecting info keys: #{e.message}"
+          TraceView.logger.debug e.backtrace
+        end
+
+        TraceView::API.trace('memcache', { :KVOp => :get_multi }, :get_multi) do
+          values = get_multi_without_traceview(args)
+
+          info_kvs[:KVHitCount] = values.length
+          TraceView::API.log('memcache', 'info', info_kvs)
+
+          values
+        end
+      end
+
+      def request_setup_with_traceview(*args)
+        if TraceView.tracing? && !TraceView.tracing_layer_op?(:get_multi)
+          server, cache_key = request_setup_without_traceview(*args)
+
+          info_kvs = { :KVKey => cache_key, :RemoteHost => server.host }
+          info_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:memcache][:collect_backtraces]
+          TraceView::API.log('memcache', 'info', info_kvs)
+
+          [server, cache_key]
+        else
+          request_setup_without_traceview(*args)
+        end
+      end
+
+      def cache_get_with_traceview(server, cache_key)
+        result = cache_get_without_traceview(server, cache_key)
+
+        info_kvs = { :KVHit => memcache_hit?(result) }
+        info_kvs[:Backtrace] = TraceView::API.backtrace if TraceView::Config[:memcache][:collect_backtraces]
+        TraceView::API.log('memcache', 'info', info_kvs)
+
+        result
+      end
+    end # module MemCache
+  end # module Inst
+end # module TraceView
+
+if defined?(::MemCache) && TraceView::Config[:memcache][:enabled]
+  ::MemCache.class_eval do
+    include TraceView::Inst::MemCache
+  end
+end

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