seanwalbran-rpm_contrib 2.1.6.1

data/lib/rpm_contrib/instrumentation/workling.rb

@@ -0,0 +1,27 @@
+# Workling instrumentation contributed by Chad Ingram of Aurora Feint
+#
+DependencyDetection.defer do
+  @name = :workling
+  
+  depends_on do
+    defined?(::Workling) and not ::NewRelic::Control.instance['disable_workling']
+  end
+
+  executes do
+    NewRelic::Agent.logger.debug 'Installing Workling instrumentation'
+  end
+
+  executes do
+    ::Workling::Base.class_eval do
+      include ::NewRelic::Agent::Instrumentation::ControllerInstrumentation
+    end
+
+    ::Workling::Discovery.discovered.each do |clazz|
+      (clazz.public_instance_methods - ::Workling::Base.public_instance_methods).each do |method|
+        puts "added method tracer Workling/#{clazz.name}/#{method}"
+        clazz.send(:add_method_tracer, method, "Workling/#{clazz.name}/#{method}", :category => :task)
+      end
+    end
+  end
+
+end

data/lib/rpm_contrib/instrumentation/yajl.rb

@@ -0,0 +1,41 @@
+DependencyDetection.defer do
+  @name = :yajl_parser
+  
+  depends_on do
+    defined?(::Yajl::Parser) && !NewRelic::Control.instance['disable_yajl_instrumentation']
+  end
+
+  executes do
+    NewRelic::Agent.logger.debug 'Installing Yajl::Parser instrumentation'
+  end  
+
+  executes do
+    ::Yajl::Parser.class_eval do
+      class << self
+        include ::NewRelic::Agent::MethodTracer
+        add_method_tracer :parse, 'Parser/Yajl/parse'
+      end
+    end
+  end
+end
+
+DependencyDetection.defer do
+  @name = :yajl_encoder
+  
+  depends_on do
+    defined?(::Yajl::Encoder) && !NewRelic::Control.instance['disable_yajl_instrumentation']
+  end
+
+  executes do
+    NewRelic::Agent.logger.debug 'Installing Yajl::Encoder instrumentation'
+  end
+
+  executes do
+    ::Yajl::Encoder.class_eval do
+      class << self
+        include ::NewRelic::Agent::MethodTracer
+        add_method_tracer :encode, 'Encoder/Yajl/encode'
+      end
+    end
+  end
+end

data/lib/rpm_contrib/instrumentation.rb

@@ -0,0 +1,16 @@
+require 'newrelic_rpm'
+module RpmContrib
+  # Contributed instrumentation files for use with newrelic_rpm gem
+  module Instrumentation
+  end
+end
+
+pattern = File.expand_path "../instrumentation/**/*.rb",  __FILE__
+Dir.glob pattern do |file|
+  begin
+    require file.to_s
+  rescue Exception => e
+    NewRelic::Agent.logger.error "Skipping instrumentation file '#{file}': #{e}"
+    NewRelic::Agent.logger.debug e.backtrace.join("\n")
+  end
+end

data/lib/rpm_contrib/language_support.rb

@@ -0,0 +1,31 @@
+module RPMContrib::LanguageSupport
+  extend self
+  
+  @@forkable = nil
+  
+  def can_fork?
+    # this is expensive to check, so we should only check once
+    return @@forkable if @@forkable != nil
+
+    if Process.respond_to?(:fork)
+      # if this is not 1.9.2 or higher, we have to make sure
+      @@forkable = ::RUBY_VERSION < '1.9.2' ? test_forkability : true
+    else
+      @@forkable = false
+    end
+
+    @@forkable
+  end
+
+  private
+
+  def test_forkability
+    child = Process.fork { exit! }
+    # calling wait here doesn't seem like it should necessary, but it seems to
+    # resolve some weird edge cases with resque forking.
+    Process.wait child
+    true
+  rescue NotImplementedError
+    false
+  end
+end

data/lib/rpm_contrib/samplers.rb

@@ -0,0 +1,17 @@
+module RpmContrib
+  # Samplers are subclasses of NewRelic::Agent::Sampler which periodically collect metric data in a 
+  # background thread.  Sampler classes belong in the sampler subdirectory and must be loaded before
+  # the agent starts.
+  module Samplers
+  end
+end
+
+pattern = File.expand_path "../samplers/**/*.rb",  __FILE__
+Dir.glob pattern do |file|
+  begin
+    require file.to_s
+  rescue Exception => e
+    NewRelic::Agent.logger.error "Skipping instrumentation file '#{file}': #{e}"
+    NewRelic::Agent.logger.debug e.backtrace.join("\n")
+  end
+end

data/lib/rpm_contrib.rb

@@ -0,0 +1,36 @@
+RPM_CONTRIB_LIB = File.dirname(__FILE__)
+
+module RPMContrib; end
+
+require 'rpm_contrib/detection'
+
+require 'newrelic_rpm'
+require 'rpm_contrib/agent_compatibility'
+require 'rpm_contrib/instrumentation'
+
+# Load all the Sampler class definitions.  These will register
+# automatically with the agent.
+require 'rpm_contrib/samplers'
+
+if defined? Rails
+  # Rails 3.x+
+  if Rails.respond_to?(:version) && Rails.version =~ /^3/
+    module NewRelic
+      class Railtie < Rails::Railtie
+        initializer("rpm_contrib.start_plugin"){ NewRelic::Control.instance.init_plugin }
+      end
+    end
+  # Rails 2.x
+  elsif defined?(Rails) && Rails.respond_to?(:configuration)
+    Rails.configuration.after_initialize { NewRelic::Control.instance.init_plugin }
+  else
+    raise "The rpm_contrib gem supports Rails 2.2+ only."
+  end
+else
+  # If not running Rails, it is important that you load the contrib gem as late 
+  # as possible so the agent initializes after everything else.  Either that 
+  # or make the following call yourself at the end of your startup sequence
+  # (it is idempotent).
+  NewRelic::Control.instance.init_plugin 
+end
+

data/test/helper.rb

@@ -0,0 +1,9 @@
+require 'rubygems'
+require 'test/unit'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require File.expand_path("../../lib/rpm_contrib.rb", __FILE__)
+
+require 'schema.rb'

data/test/schema.rb

@@ -0,0 +1,19 @@
+# Use this file to add tables you need to do testing
+# These are created in a sqllite memory DB
+
+begin
+  require 'sqlite3'
+  require 'active_record'
+
+  ActiveRecord::Base.establish_connection(:adapter => 'sqlite3', :database => ':memory:')
+  ActiveRecord::Migration.verbose = false
+
+  ActiveRecord::Schema.define do
+
+    create_table :stories, :force => true do |table|
+      table.string :text
+    end
+  end
+rescue LoadError
+  # Skip AR tests
+end

data/test/test_curb.rb

@@ -0,0 +1,69 @@
+require "#{File.dirname(__FILE__)}/helper"
+begin
+require 'curb' 
+rescue LoadError
+end
+
+class NewRelic::Agent::NetInstrumentationTest < Test::Unit::TestCase
+  include NewRelic::Agent::Instrumentation::ControllerInstrumentation
+  def setup
+    NewRelic::Agent.manual_start
+    @engine = NewRelic::Agent.instance.stats_engine
+    @engine.clear_stats
+  end
+  def test_get
+    curl = Curl::Easy.new('http://www.google.com/index.html')
+    curl.perform
+    assert_match /<head>/, curl.body_str
+    assert_equal %w[External/www.google.com/Curl::Easy External/Curl::Multi
+                    External/allOther External/www.google.com/all].sort,
+       @engine.metrics.sort
+  end
+
+  def test_multi
+    multi = Curl::Multi.new
+    curl1 = Curl::Easy.new('http://www.google.com/index.html')
+    multi.add curl1
+    curl2 = Curl::Easy.new('http://www.yahoo.com/')
+    multi.add curl2
+    multi.perform
+    assert_match /<head>/, curl1.body_str
+    assert_match /<head>/, curl2.body_str
+    assert_equal %w[External/Curl::Multi External/allOther].sort,
+       @engine.metrics.sort
+  end
+
+  def test_background
+    perform_action_with_newrelic_trace("task", :category => :task) do
+      curl = Curl::Easy.new('http://www.google.com/index.html')
+      curl.perform
+      assert_match /<head>/, curl.body_str
+    end
+    assert_equal %w[External/Curl::Multi
+          External/Curl::Multi:OtherTransaction/Background/NewRelic::Agent::NetInstrumentationTest/task
+          External/www.google.com/Curl::Easy External/allOther External/www.google.com/all
+          External/www.google.com/Curl::Easy:OtherTransaction/Background/NewRelic::Agent::NetInstrumentationTest/task].sort,
+       @engine.metrics.select{|m| m =~ /^External/}.sort
+  end
+
+  def test_transactional
+    perform_action_with_newrelic_trace("task") do
+      curl = Curl::Easy.new('http://www.google.com/index.html')
+      curl.perform
+      assert_match /<head>/, curl.body_str
+    end
+    assert_equal %w[External/Curl::Multi
+          External/Curl::Multi:Controller/NewRelic::Agent::NetInstrumentationTest/task
+          External/www.google.com/Curl::Easy External/allWeb External/www.google.com/all
+          External/www.google.com/Curl::Easy:Controller/NewRelic::Agent::NetInstrumentationTest/task].sort,
+       @engine.metrics.select{|m| m =~ /^External/}.sort
+  end
+  def test_ignore
+    NewRelic::Agent.disable_all_tracing do
+      curl = Curl::Easy.new('http://www.google.com/index.html')
+      curl.http_post('data')
+    end
+    assert_equal 0, @engine.metrics.size
+  end
+
+end if defined? ::Curl::Easy

data/test/test_picky.rb

@@ -0,0 +1,55 @@
+require 'picky'
+
+require "#{File.dirname(__FILE__)}/helper"
+
+class NewRelic::Agent::PickyIntrumentationTest < Test::Unit::TestCase
+
+  def tokens_for *tokens
+    tokens.map{|t|
+      token = 'whatever'
+
+      token.extend Module.new{
+        define_method(:'partial?'){ t[:partial] }
+        define_method(:'similar?'){ t[:similar] }
+        define_method(:'qualifiers'){ t[:qualifiers] }
+      }
+
+      token
+    }
+  end
+  
+  def test_obfuscate_tokens
+    tokens = tokens_for({})
+    assert_equal 'xxx', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for({:similar => true})
+    assert_equal 'xxx~', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for({:partial => true})
+    assert_equal 'xxx*', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for({:qualifiers => [:haha]})
+    assert_equal 'haha:xxx', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for( {:partial => true}, {:similar => true} )
+    assert_equal 'xxx* xxx~', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for( {:similar => true}, {:partial => true} )
+    assert_equal 'xxx* xxx~', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for( {:partial => true}, {:partial => true} )
+    assert_equal 'xxx* xxx*', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for(
+      {:similar => true, :qualifiers => [:bla]},
+      {:partial => true, :qualifiers => [:bla, :blub]}
+    )
+    assert_equal 'bla,blub:xxx* bla:xxx~', Picky::NewRelic.obfuscate_tokens(tokens)
+
+    tokens = tokens_for(
+      {:similar => true, :qualifiers => [:bla]},
+      {:partial => true, :qualifiers => [:blub, :bla]}
+    )
+    assert_equal 'bla,blub:xxx* bla:xxx~', Picky::NewRelic.obfuscate_tokens(tokens)
+  end
+end

data/test/test_redis.rb

@@ -0,0 +1,34 @@
+require "#{File.dirname(__FILE__)}/helper"
+begin
+  require 'redis'
+  require 'ruby-debug'
+rescue LoadError
+end
+
+require "#{File.dirname(__FILE__)}/../lib/rpm_contrib/instrumentation/redis"
+
+if defined?(::Redis)
+
+
+  class RedisTest < Test::Unit::TestCase
+
+    # Called before every test method runs. Can be used
+    # to set up fixture information.
+    def setup
+      # Do nothing
+    end
+
+    # Called after every test method runs. Can be used to tear
+    # down fixture information.
+
+    def teardown
+      # Do nothing
+    end
+
+    # Fake test
+    def test_fail
+
+
+    end
+  end
+end

data/test/test_resque.rb

@@ -0,0 +1,74 @@
+require 'resque'
+require 'mocha'
+require File.expand_path(File.dirname(__FILE__) + "/helper")
+require File.expand_path(File.dirname(__FILE__) + "/../lib/rpm_contrib")
+
+class ResqueTest < Test::Unit::TestCase
+  class ExtendoJorb < ::Resque::Job
+    def self.perform
+      true
+    end
+  end
+  
+  class GoodJorb
+    def self.perform
+      true
+    end
+  end
+
+  class BadJorb
+    def self.perform
+      raise "I'm doing a bad jorb"
+    end
+  end
+
+  def setup
+    @worker = Resque::Worker.new(:test)
+    NewRelic::Agent.manual_start
+    @engine = NewRelic::Agent.instance.stats_engine
+  end
+
+  def teardown
+    @engine.clear_stats
+  end
+  
+  def test_should_instrument_job_extending_from_resque_job
+    @worker.perform(Resque::Job.new(:test,
+                                    'class' => 'ResqueTest::ExtendoJorb'))
+
+    metrics = [ 'OtherTransaction/all', 'OtherTransaction/ResqueJob/all',
+                'OtherTransaction/ResqueJob/ResqueTest::ExtendoJorb/perform' ]
+    metrics.each do |metric|
+      assert(@engine.metrics.include?(metric),
+             "#{@engine.metrics.inspect} missing #{metric}")
+    end
+    assert !@engine.metrics.include?('Errors/all')
+  end
+
+  def test_should_instrument_poro_job
+    @worker.perform(Resque::Job.new(:test, 'class' => 'ResqueTest::GoodJorb'))
+
+    metrics = [ 'OtherTransaction/all', 'OtherTransaction/ResqueJob/all',
+                'OtherTransaction/ResqueJob/ResqueTest::GoodJorb/perform' ]
+    metrics.each do |metric|
+      assert(@engine.metrics.include?(metric),
+             "#{@engine.metrics.inspect} missing #{metric}")
+    end
+    assert !@engine.metrics.include?('Errors/all')
+  end
+
+  def test_should_collect_job_errors
+    begin
+      @worker.perform(Resque::Job.new(:test, 'class' => 'ResqueTest::BadJorb'))
+    rescue
+    end
+    
+    metrics = [ 'OtherTransaction/all', 'OtherTransaction/ResqueJob/all',
+                'OtherTransaction/ResqueJob/ResqueTest::BadJorb/perform',
+                'Errors/all' ]
+    metrics.each do |metric|
+      assert(@engine.metrics.include?(metric),
+             "#{@engine.metrics.inspect} missing #{metric}")
+    end      
+  end
+end

data/test/test_workling.rb

@@ -0,0 +1,34 @@
+require "#{File.dirname(__FILE__)}/helper"
+begin
+  require 'redis'
+  require 'ruby-debug'
+rescue LoadError
+end
+
+require "#{File.dirname(__FILE__)}/../lib/rpm_contrib/instrumentation/workling"
+
+if defined?(::Workling)
+
+
+  class WorklingTest < Test::Unit::TestCase
+
+    # Called before every test method runs. Can be used
+    # to set up fixture information.
+    def setup
+      # Do nothing
+    end
+
+    # Called after every test method runs. Can be used to tear
+    # down fixture information.
+
+    def teardown
+      # Do nothing
+    end
+
+    # Fake test
+    def test_fail
+
+
+    end
+  end
+end

metadata

@@ -0,0 +1,148 @@
+--- !ruby/object:Gem::Specification 
+name: seanwalbran-rpm_contrib
+version: !ruby/object:Gem::Version 
+  hash: 125
+  prerelease: 
+  segments: 
+  - 2
+  - 1
+  - 6
+  - 1
+  version: 2.1.6.1
+platform: ruby
+authors: 
+- Bill Kayser
+- Jon Guymon
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-02-18 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: newrelic_rpm
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 1
+        segments: 
+        - 3
+        - 1
+        - 1
+        version: 3.1.1
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: newrelic_rpm
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 1
+        segments: 
+        - 3
+        - 1
+        - 1
+        version: 3.1.1
+  type: :runtime
+  version_requirements: *id002
+description: |
+  Community contributed instrumentation for various frameworks based on
+  the New Relic Ruby monitoring gem newrelic_rpm.
+
+email: support@newrelic.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- CHANGELOG
+- LICENSE
+- README.md
+files: 
+- CHANGELOG
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/new_relic/control/camping.rb
+- lib/rpm_contrib.rb
+- lib/rpm_contrib/agent_compatibility.rb
+- lib/rpm_contrib/detection.rb
+- lib/rpm_contrib/detection/camping.rb
+- lib/rpm_contrib/instrumentation.rb
+- lib/rpm_contrib/instrumentation/active_messaging.rb
+- lib/rpm_contrib/instrumentation/aws.rb
+- lib/rpm_contrib/instrumentation/camping.rb
+- lib/rpm_contrib/instrumentation/cassandra.rb
+- lib/rpm_contrib/instrumentation/crack.rb
+- lib/rpm_contrib/instrumentation/curb.rb
+- lib/rpm_contrib/instrumentation/elastic_search.rb
+- lib/rpm_contrib/instrumentation/kyototycoon.rb
+- lib/rpm_contrib/instrumentation/mongo.rb
+- lib/rpm_contrib/instrumentation/paperclip.rb
+- lib/rpm_contrib/instrumentation/picky.rb
+- lib/rpm_contrib/instrumentation/redis.rb
+- lib/rpm_contrib/instrumentation/resque.rb
+- lib/rpm_contrib/instrumentation/riak_client.rb
+- lib/rpm_contrib/instrumentation/ripple.rb
+- lib/rpm_contrib/instrumentation/sinatra.rb
+- lib/rpm_contrib/instrumentation/thinking_sphinx.rb
+- lib/rpm_contrib/instrumentation/typhoeus.rb
+- lib/rpm_contrib/instrumentation/ultrasphinx.rb
+- lib/rpm_contrib/instrumentation/workling.rb
+- lib/rpm_contrib/instrumentation/yajl.rb
+- lib/rpm_contrib/language_support.rb
+- lib/rpm_contrib/samplers.rb
+- test/helper.rb
+- test/schema.rb
+- test/test_curb.rb
+- test/test_picky.rb
+- test/test_redis.rb
+- test/test_resque.rb
+- test/test_workling.rb
+homepage: http://github.com/newrelic/rpm_contrib
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Contributed Instrumentation for New Relic RPM
+- -m
+- README.md
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Contributed Instrumentation for New Relic RPM
+test_files: []
+