copperegg-apm 1.0.0.pre1

data/lib/copperegg/apm/engine.rb

@@ -0,0 +1,15 @@
+module CopperEgg
+  module APM
+    class Engine < ::Rails::Engine
+      initializer "copperegg_apm.helpers" do
+        ActiveSupport.on_load_all do
+          helper CopperEgg::APM::Rum
+        end
+      end
+      
+      rake_tasks do
+        load 'copperegg/apm/tasks.rb'
+      end
+    end
+  end
+end

data/lib/copperegg/apm/errors.rb

@@ -0,0 +1,6 @@
+module CopperEgg
+  module APM
+    class ConfigurationError < Exception; end
+    class PayloadError < Exception; end
+  end
+end

data/lib/copperegg/apm/ethon/easy/operations.rb

@@ -0,0 +1,37 @@
+module CopperEgg
+  module APM
+    module Ethon
+      module Easy
+        module Operations
+          def perform_with_ce_instrumentation
+            if CopperEgg::APM::Configuration.benchmark_http?
+              x = url.gsub(/\/\/[^:]+:[^@]@/,"//").gsub(/\?.*/,"")
+              starttime = (Time.now.to_f * 1000.0).to_i
+              result = perform_without_ce_instrumentation
+              time = (Time.now.to_f * 1000.0).to_i - starttime
+
+              CopperEgg::APM.send_payload(:url => x, :time => time)
+                
+              result
+            else
+              perform_without_ce_instrumentation
+            end
+          end
+        end
+      end
+    end
+  end
+end
+
+if defined?(::Ethon::Easy::Operations)
+
+  module Ethon
+    class Easy
+      module Operations
+        include CopperEgg::APM::Ethon::Easy::Operations
+        alias_method :perform_without_ce_instrumentation, :perform
+        alias_method :perform, :perform_with_ce_instrumentation
+      end
+    end
+  end
+end

data/lib/copperegg/apm/kernel.rb

@@ -0,0 +1,19 @@
+module CopperEgg
+  module APM
+    module Kernel
+      alias_method :raise_without_ce_instrumentation, :raise
+
+      def raise(*args)
+        super(ArgumentError, "wrong number of arguments", caller) if args.size > 3
+        CopperEgg::APM.capture_exception(*args) if CopperEgg::APM::Configuration.benchmark_exceptions?
+        raise_without_ce_instrumentation(*args)
+      end
+      
+      alias_method :fail, :raise
+    end
+  end
+end
+
+Object.class_eval do
+  include CopperEgg::APM::Kernel
+end

data/lib/copperegg/apm/middleware.rb

@@ -0,0 +1,20 @@
+module CopperEgg
+  module APM
+    class Middleware
+      def initialize(app)
+        @app = app
+      end
+      
+      def call(env)
+        CopperEgg::APM.benchmark(:url => env['REQUEST_URI'].gsub(/\/\/[^:]+:[^@]@/,"//").gsub(/\?.*/,"")) do
+          @status, @headers, @response = @app.call(env)
+        end
+        [@status, @headers, self]
+      end
+      
+      def each(&block)
+        @response.each(&block)
+      end
+    end
+  end
+end

data/lib/copperegg/apm/mysql.rb

@@ -0,0 +1,31 @@
+module CopperEgg
+  module APM
+    module Mysql
+      def query_with_ce_instrumentation(*args)
+        if CopperEgg::APM::Configuration.benchmark_sql?
+          starttime = (Time.now.to_f * 1000.0).to_i
+          result = query_without_ce_instrumentation(*args)
+          time = (Time.now.to_f * 1000.0).to_i - starttime
+
+          return result if args.first =~ /\A\s*(begin|commit|rollback|set)/i
+
+          CopperEgg::APM.send_payload(:sql => CopperEgg::APM.obfuscate_sql(args.first), :time => time)
+            
+          result
+        else
+          query_without_ce_instrumentation(*args)
+        end
+      end
+    end
+  end
+end
+
+if defined?(::Mysql)
+
+  class Mysql
+    include CopperEgg::APM::Mysql
+    alias_method :query_without_ce_instrumentation, :query
+    alias_method :query, :query_with_ce_instrumentation
+  end
+
+end

data/lib/copperegg/apm/mysql2/client.rb

@@ -0,0 +1,35 @@
+module CopperEgg
+  module APM
+    module Mysql2
+      module Client
+        def query_with_ce_instrumentation(*args)
+          if CopperEgg::APM::Configuration.benchmark_sql?
+            starttime = (Time.now.to_f * 1000.0).to_i
+            result = query_without_ce_instrumentation(*args)
+            time = (Time.now.to_f * 1000.0).to_i - starttime
+
+            return result if args.first =~ /\A\s*(begin|commit|rollback|set)/i
+
+            CopperEgg::APM.send_payload(:sql => CopperEgg::APM.obfuscate_sql(args.first), :time => time)
+              
+            result
+          else
+            query_without_ce_instrumentation(*args)
+          end
+        end
+      end
+    end
+  end
+end
+
+if defined?(::Mysql2::Client)
+
+  module Mysql2
+    class Client
+      include CopperEgg::APM::Mysql2::Client
+      alias_method :query_without_ce_instrumentation, :query
+      alias_method :query, :query_with_ce_instrumentation
+    end
+  end
+
+end

data/lib/copperegg/apm/net/http.rb

@@ -0,0 +1,39 @@
+module CopperEgg
+  module APM
+    module Net
+      module HTTP
+        def request_with_ce_instrumentation(req, body=nil, &block)
+          if CopperEgg::APM::Configuration.benchmark_http?
+            starttime = (Time.now.to_f * 1000.0).to_i
+            result = request_without_ce_instrumentation(req, body, &block)
+            time = (Time.now.to_f * 1000.0).to_i - starttime
+            url = "http#{"s" if @use_ssl}://#{address}#{":#{port}" if port != ::Net::HTTP.default_port}#{req.path.sub(/\?.*/,"")}"
+
+            CopperEgg::APM.send_payload(:url => url, :time => time)
+              
+            result
+          else
+            request_without_ce_instrumentation(req, body, &block)
+          end
+        end
+      end
+    end
+  end
+end
+
+begin
+  require 'net/http'
+rescue LoadError
+end
+
+if defined?(::Net::HTTP)
+
+  module Net
+    class HTTP
+      include CopperEgg::APM::Net::HTTP
+      alias_method :request_without_ce_instrumentation, :request
+      alias_method :request, :request_with_ce_instrumentation
+    end
+  end
+
+end

data/lib/copperegg/apm/pg/connection.rb

@@ -0,0 +1,35 @@
+module CopperEgg
+  module APM
+    module PG
+      module Connection
+        def exec_with_ce_instrumentation(*args)
+          if CopperEgg::APM::Configuration.benchmark_sql?
+            starttime = (Time.now.to_f * 1000.0).to_i
+            result = block_given? ? exec_without_ce_instrumentation(*args, &Proc.new) : exec_without_ce_instrumentation(*args)
+            time = (Time.now.to_f * 1000.0).to_i - starttime
+
+            return result if args.first =~ /\A\s*(begin|commit|rollback|set)/i
+
+            CopperEgg::APM.send_payload(:sql => CopperEgg::APM.obfuscate_sql(args.first), :time => time)
+              
+            result
+          else
+            block_given? ? exec_without_ce_instrumentation(*args, &Proc.new) : exec_without_ce_instrumentation(*args)
+          end
+        end
+      end
+    end
+  end
+end
+
+if defined?(::PG::Connection)
+
+  module PG
+    class Connection
+      include CopperEgg::APM::PG::Connection
+      alias_method :exec_without_ce_instrumentation, :exec
+      alias_method :exec, :exec_with_ce_instrumentation
+    end
+  end
+
+end

data/lib/copperegg/apm/restclient/request.rb

@@ -0,0 +1,33 @@
+module CopperEgg
+  module APM
+    module RestClient
+      module Request
+        def execute_with_ce_instrumentation(&block)
+          if CopperEgg::APM::Configuration.benchmark_http?
+            starttime = (Time.now.to_f * 1000.0).to_i
+            result = execute_without_ce_instrumentation(&block)
+            time = (Time.now.to_f * 1000.0).to_i - starttime
+
+            CopperEgg::APM.send_payload(:url => url.gsub(/\/\/[^:]+:[^@]@/,"//").gsub(/\?.*/,""), :time => time)
+              
+            result
+          else
+            execute_without_ce_instrumentation(&block)
+          end
+        end        
+      end
+    end
+  end
+end
+
+if defined?(::RestClient::Request)
+
+  module RestClient
+    class Request
+      include CopperEgg::APM::RestClient::Request
+      alias_method :execute_without_ce_instrumentation, :execute
+      alias_method :execute, :execute_with_ce_instrumentation
+    end
+  end
+
+end

data/lib/copperegg/apm/rum.rb

@@ -0,0 +1,24 @@
+module CopperEgg
+  module APM
+    module Rum
+      def real_user_monitoring_javascript_tag
+        script = %Q(
+        <script type="text/javascript">
+        var BACON = BACON || {};
+
+        BACON.id = '#{CopperEgg::APM::Configuration.instrument_key}';
+        BACON.short_url = #{CopperEgg::APM::Configuration.rum_short_url};
+        BACON.beaconUrl = '#{CopperEgg::APM::Configuration.rum_beacon_url}';
+
+        BACON.starttime = new Date().getTime();
+        (function(d, s) { var js = d.createElement(s);
+            js.async = true; js.src = "http://cdn.copperegg.com/rum/bacon.min.js";
+            var s = d.getElementsByTagName(s)[0]; s.parentNode.insertBefore(js, s);
+        })(document, "script");
+        </script>
+        )
+        script.respond_to?(:html_safe) ? script.html_safe : script
+      end
+    end
+  end
+end

data/lib/copperegg/apm/sqlite3/database.rb

@@ -0,0 +1,35 @@
+module CopperEgg
+  module APM
+    module SQLite3
+      module Database
+        def execute_with_ce_instrumentation(sql, bind_vars = [], *args, &block)
+          if CopperEgg::APM::Configuration.benchmark_sql?
+            starttime = (Time.now.to_f * 1000.0).to_i
+            result = execute_without_ce_instrumentation(sql, bind_vars, *args, &block)
+            time = (Time.now.to_f * 1000.0).to_i - starttime
+
+            return result if sql =~ /\A\s*(begin|commit|rollback|set)/i
+
+            CopperEgg::APM.send_payload(:sql => CopperEgg::APM.obfuscate_sql(sql), :time => time)
+              
+            result
+          else
+            execute_without_ce_instrumentation(sql, bind_vars, *args, &block)
+          end
+        end
+      end
+    end
+  end
+end
+
+if defined?(::SQLite3::Database)
+
+  module SQLite3
+    class Database
+      include CopperEgg::APM::SQLite3::Database
+      alias_method :execute_without_ce_instrumentation, :execute
+      alias_method :execute, :execute_with_ce_instrumentation
+    end
+  end
+
+end

data/lib/copperegg/apm/tasks.rb

@@ -0,0 +1,11 @@
+require 'rake'
+require File.join(File.dirname(__FILE__), 'benchmark_methods_table')
+
+namespace :copperegg do
+  namespace :apm  do
+    desc "Prints a summary of all methods defined in your project and whether or not they are benchmarked."
+    task :methods => :environment do
+      CopperEgg::APM::BenchmarkMethodsTable.new.print_table
+    end
+  end
+end

data/lib/copperegg/apm/typhoeus/hydra.rb

@@ -0,0 +1,33 @@
+module CopperEgg
+  module APM
+    module Typhoeus
+      module Hydra
+        def handle_request_with_ce_instrumentation(request, response, live_request=true)
+          if CopperEgg::APM::Configuration.benchmark_http?
+            starttime = (Time.now.to_f * 1000.0).to_i
+            result = handle_request_without_ce_instrumentation(request, response, live_request)
+            time = (Time.now.to_f * 1000.0).to_i - starttime
+
+            CopperEgg::APM.send_payload(:url => request.url.gsub(/\/\/[^:]+:[^@]@/,"//").gsub(/\?.*/,""), :time => time)
+              
+            result
+          else
+            handle_request_without_ce_instrumentation(request, response, live_request)
+          end
+        end
+      end
+    end
+  end
+end
+
+if defined?(::Typhoeus::Hydra) && defined?(::Typhoeus::VERSION) && ::Typhoeus::VERSION =~ /\A0\.3\.?/
+  
+  module Typhoeus
+    class Hydra
+      include CopperEgg::APM::Typhoeus::Hydra
+      alias_method :handle_request_without_ce_instrumentation, :handle_request
+      alias_method :handle_request, :handle_request_with_ce_instrumentation
+    end
+  end
+
+end

data/lib/copperegg/apm/unbound_method.rb

@@ -0,0 +1,116 @@
+module CopperEgg
+  module APM
+    module UnboundMethod
+      def class_method?
+        self.class_method == true
+      end
+
+      def benchmarkable?
+        return false if !user_defined? || attribute?
+        name =~ /\A[\w\d\!\?]+\z/ ? true : false
+      end
+
+      def user_defined?
+        return false if source_filename.nil? || !File.exists?(source_filename)
+        paths = Gem.path + [Gem.dir, CopperEgg::APM::Configuration.gem_root, "/lib/ruby/", "/site_ruby/", "/vendor_ruby/", "/gems/"]
+        source_filename ? paths.select! {|path| source_filename.include?(path) }.empty? : false
+      end
+
+      def attribute?
+        parent_class.instance_methods.include?("#{name}=".to_sym)
+      end
+
+      def source_filename
+        @source_filename ||=  if respond_to?(:source_location)
+                                source_location.to_a.first
+                              elsif respond_to?(:__file__)
+                                __file__
+                              end
+      end
+
+      def source_line
+        @source_line ||=  if respond_to?(:source_location)
+                            source_location.to_a.last
+                          elsif respond_to?(:__line__)
+                            __line__
+                          end
+      end
+
+      def display_filename
+        return @display_filename if @display_filename
+        @display_filename = "#{source_filename}:#{source_line}"
+        @display_filename.sub! CopperEgg::APM::Configuration.app_root, "..."
+      end
+
+      def display_name
+        "#{parent_class}#{class_method? ? "." : "#"}#{name}"
+      end
+
+      def class_eval_string
+        <<-DEF
+          #{"class << self" if class_method?}
+          alias_method "_cu_#{name}", "#{name}"
+          def #{name}(*args)
+            starttime = (Time.now.to_f * 1000.0).to_i
+            result = block_given? ? _cu_#{name}(*args,&Proc.new) : _cu_#{name}(*args)
+            time = (Time.now.to_f * 1000.0).to_i - starttime
+
+            CopperEgg::APM.send_payload({:method => "#{display_name}", :time => time})
+              
+            result
+          end
+          #{"end" if class_method?}
+        DEF
+      end
+
+      def add_benchmarking
+        parent_class.module_eval(class_eval_string)
+      end
+
+      def benchmark_levels
+        levels = [:full, :custom]
+        return levels if !defined?(::Rails)
+
+        if parent_class < ::ActiveRecord::Base
+          levels << :moderate
+        end
+        if parent_class < ::ActionController::Base
+          levels << :basic
+        end
+        levels
+      end
+
+      def excluded?
+        return true if CopperEgg::APM::Configuration.benchmark_methods_level == :disabled
+
+        return false if CopperEgg::APM::Configuration.only_methods.include?(parent_class.to_s)
+        return false if CopperEgg::APM::Configuration.only_methods.include?(display_name)
+        return false if CopperEgg::APM::Configuration.only_methods.include?(name)
+
+        if CopperEgg::APM::Configuration.only_methods.length > 0
+          return true if !CopperEgg::APM::Configuration.only_methods.include?(parent_class.to_s) && !CopperEgg::APM::Configuration.only_methods.include?(display_name) && !CopperEgg::APM::Configuration.only_methods.include?(name)
+        end
+
+        return true if CopperEgg::APM::Configuration.exclude_methods.include?(parent_class.to_s)
+        return true if CopperEgg::APM::Configuration.exclude_methods.include?(display_name)
+        return true if CopperEgg::APM::Configuration.exclude_methods.include?(display_name)
+
+        return false if CopperEgg::APM::Configuration.include_methods.include?(parent_class.to_s)
+        return false if CopperEgg::APM::Configuration.include_methods.include?(display_name)
+        return false if CopperEgg::APM::Configuration.include_methods.include?(name)
+
+        return true if name =~ /\A_/
+        return true if name =~ /\A[\w\d]+\?\z/
+
+        return false if benchmark_levels.include?(CopperEgg::APM::Configuration.benchmark_methods_level)
+        
+        true
+      end
+    end
+  end
+end
+
+class UnboundMethod
+  include CopperEgg::APM::UnboundMethod
+  attr_accessor :parent_class, :class_method
+end