instrument_all_the_things 1.0.4

data/lib/instrument_all_the_things/method_proxy.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+require_relative './method_instrumentor'
+
+module InstrumentAllTheThings
+  module MethodProxy
+    def self.for_class(klass)
+      find_for_class(klass) || install_on_class(klass)
+    end
+
+    def self.find_for_class(klass)
+      klass.ancestors.detect do |a|
+        a.is_a?(Instrumentor) &&
+          a._iatt_built_for == klass
+      end
+    end
+
+    def self.install_on_class(klass)
+      construct_for_class(klass).tap do |m|
+        klass.prepend(m)
+      end
+    end
+
+    def self.construct_for_class(klass)
+      Module.new do
+        extend Instrumentor
+      end.tap { |m| m._iatt_built_for = klass }
+    end
+
+    module Instrumentor
+      def inspect
+        "InstrumentAllTheThings::#{@_iatt_built_for}Proxy"
+      end
+
+      def _iatt_built_for
+        @_iatt_built_for
+      end
+
+      def _iatt_built_for=(val)
+        @_iatt_built_for = val
+      end
+
+      def set_context_tags(klass, settings, args, kwargs)
+        return unless settings.is_a?(Hash) && settings[:trace].is_a?(Hash) && settings[:trace][:tags]
+
+        settings[:context][:tags] = settings[:trace][:tags].map do |tag|
+          if tag.is_a?(Proc)
+            case tag.arity
+            when 2
+              tag.call(args, kwargs)
+            when 1
+              tag.parameters[0][1].to_s == 'args' ? tag.call(args) : tag.call(kwargs)
+            else
+              klass.instance_exec(&tag)
+            end
+          else
+            tag
+          end
+          rescue StandardError
+            nil
+        end.compact
+      end
+
+      def wrap_implementation(method_name, settings)
+        wrap = MethodInstrumentor.new(**settings)
+        set_tags = method(:set_context_tags)
+
+        define_method(method_name) do |*args, **kwargs, &blk|
+          set_tags.call(self, settings, args, kwargs)
+          wrap.invoke(klass: is_a?(Class) ? self : self.class) do
+            super(*args, **kwargs, &blk)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/instrument_all_the_things/testing/rspec_matchers.rb

@@ -0,0 +1,97 @@
+# frozen_string_literal: true
+
+module InstrumentAllTheThings
+  module Testing
+    module RSpecMatchers
+      def histogram_value(counter_name)
+        stats = InstrumentAllTheThings.stat_reporter.emitted_values[:histogram][counter_name]
+        stats.inject(0){|l, n| l + n[:args][0] }
+      end
+
+      def distribution_values(distribution_name, with_tags: nil)
+        stats = InstrumentAllTheThings.stat_reporter.emitted_values[:distribution][distribution_name]
+
+        if with_tags && !stats.empty?
+          stats = stats.select do |s|
+            with_tags.all?{|t| s[:tags].include?(t) }
+          end
+        end
+
+        stats&.map{|i| i[:args] }&.map(&:first) || []
+      end
+
+      def histogram_values(histogram_name, with_tags: nil)
+        stats = InstrumentAllTheThings.stat_reporter.emitted_values[:histogram][histogram_name]
+
+        if with_tags && !stats.empty?
+          stats = stats.select do |s|
+            with_tags.all?{|t| s[:tags].include?(t) }
+          end
+        end
+
+        stats&.map{|i| i[:args] }&.map(&:first) || []
+      end
+
+      def timing_values(timing_name, with_tags: nil)
+        stats = InstrumentAllTheThings.stat_reporter.emitted_values[:timing][timing_name]
+
+        if with_tags && !stats.empty?
+          stats = stats.select do |s|
+            with_tags.all?{|t| s[:tags].include?(t) }
+          end
+        end
+
+        stats&.map{|i| i[:args] }&.map(&:first) || []
+      end
+
+      def set_value(counter_name, with_tags: nil)
+        stats = InstrumentAllTheThings.stat_reporter.emitted_values[:set][counter_name]
+
+        if with_tags && !stats.empty?
+          stats = stats.select do |s|
+            with_tags.all?{|t| s[:tags].include?(t) }
+          end
+        end
+
+        data = stats&.map{|i| i[:args] }&.map(&:first)
+        data ? data.uniq.length : 0
+      end
+
+      def gauge_value(counter_name, with_tags: nil)
+        stats = InstrumentAllTheThings.stat_reporter.emitted_values[:gauge][counter_name]
+
+        if with_tags && !stats.empty?
+          stats = stats.select do |s|
+            with_tags.all?{|t| s[:tags].include?(t) }
+          end
+        end
+        stats.last&.fetch(:args)&.first
+      end
+
+      def counter_value(counter_name, with_tags: nil)
+        stats = InstrumentAllTheThings.stat_reporter.emitted_values[:count][counter_name]
+        if with_tags && !stats.empty?
+          stats = stats.select do |s|
+            with_tags.all?{|t| s[:tags].include?(t) }
+          end
+        end
+        stats.inject(0){|l, n| l + n[:args][0] }
+      end
+
+      def flush_traces
+        Datadog.tracer&.writer&.worker&.flush_data
+      end
+
+      def emitted_spans(filtered_by: nil)
+        sleep 0.01
+        traces = InstrumentAllTheThings::Testing::TraceTracker.tracker.traces.map(&:dup)
+        if filtered_by
+          filtered_by.transform_keys!(&:to_s)
+          traces.select! { |t| filtered_by < t }
+        end
+
+        traces
+      end
+    end
+  end
+end

data/lib/instrument_all_the_things/testing/stat_tracker.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require 'datadog/statsd'
+
+module InstrumentAllTheThings
+  module Testing
+    class StatTracker < Clients::StatReporter::DataDog
+      attr_reader :emitted_values
+      %i[
+      count
+      distribution
+      gauge
+      histogram
+      set
+      time
+      timing
+      ].each do |meth|
+        define_method(meth) do |*args, **kwargs, &blk|
+          @emitted_values[meth][args[0]] << {
+            args: args[1..-1],
+            tags: kwargs.fetch(:tags, []),
+            kwargs: kwargs,
+          }
+
+          super(*args, **kwargs, &blk)
+        end
+      end
+
+      def initialize(*args, **kwargs, &blk)
+        super
+        reset!
+      end
+
+      def reset!
+        @emitted_values = Hash.new do |h, k|
+          h[k] = Hash.new do |h2, k2|
+            h2[k2] = []
+          end
+        end
+      end
+    end
+  end
+end

data/lib/instrument_all_the_things/testing/trace_tracker.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module InstrumentAllTheThings
+  module Testing
+    class TraceTracker
+      attr_reader :traces
+
+      def self.tracker
+        @tracker ||= new
+      end
+
+      def initialize
+        reset!
+      end
+
+      def reset!
+        @traces = []
+      end
+
+      def <<(val)
+        @traces = @traces.concat(MessagePack.load(val[:body]).flatten)
+      end
+    end
+  end
+end

data/lib/instrument_all_the_things/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module InstrumentAllTheThings
+  VERSION = '1.0.4'
+end

metadata

@@ -0,0 +1,227 @@
+--- !ruby/object:Gem::Specification
+name: instrument_all_the_things
+version: !ruby/object:Gem::Version
+  version: 1.0.4
+platform: ruby
+authors:
+- Brian Malinconico
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2020-04-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ddtrace
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dogstatsd-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: benchmark-ips
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Wrappers to make instrumentation of methods easy and pleasant to read
+email:
+- bmalinconico@terminus.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".drone.yml"
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- ".ruby-version"
+- ".travis.yml"
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- instrument_all_the_things.gemspec
+- lib/instrument_all_the_things.rb
+- lib/instrument_all_the_things/clients/stat_reporter/datadog.rb
+- lib/instrument_all_the_things/clients/tracer/blackhole.rb
+- lib/instrument_all_the_things/context.rb
+- lib/instrument_all_the_things/helpers.rb
+- lib/instrument_all_the_things/instrumentors/all.rb
+- lib/instrument_all_the_things/instrumentors/error_logging.rb
+- lib/instrument_all_the_things/instrumentors/execution_count_and_timing.rb
+- lib/instrument_all_the_things/instrumentors/gc_stats.rb
+- lib/instrument_all_the_things/instrumentors/tracing.rb
+- lib/instrument_all_the_things/method_instrumentor.rb
+- lib/instrument_all_the_things/method_proxy.rb
+- lib/instrument_all_the_things/testing/rspec_matchers.rb
+- lib/instrument_all_the_things/testing/stat_tracker.rb
+- lib/instrument_all_the_things/testing/trace_tracker.rb
+- lib/instrument_all_the_things/version.rb
+- logo.jpg
+- vendor/cache/ast-2.4.0.gem
+- vendor/cache/benchmark-ips-2.7.2.gem
+- vendor/cache/coderay-1.1.2.gem
+- vendor/cache/ddtrace-0.34.0.gem
+- vendor/cache/diff-lcs-1.3.gem
+- vendor/cache/docile-1.3.2.gem
+- vendor/cache/dogstatsd-ruby-4.7.0.gem
+- vendor/cache/jaro_winkler-1.5.4.gem
+- vendor/cache/method_source-0.9.2.gem
+- vendor/cache/msgpack-1.3.3.gem
+- vendor/cache/parallel-1.19.1.gem
+- vendor/cache/parser-2.7.0.2.gem
+- vendor/cache/pry-0.12.2.gem
+- vendor/cache/rainbow-3.0.0.gem
+- vendor/cache/rake-10.5.0.gem
+- vendor/cache/rexml-3.2.4.gem
+- vendor/cache/rspec-3.9.0.gem
+- vendor/cache/rspec-core-3.9.1.gem
+- vendor/cache/rspec-expectations-3.9.0.gem
+- vendor/cache/rspec-mocks-3.9.1.gem
+- vendor/cache/rspec-support-3.9.2.gem
+- vendor/cache/rubocop-0.80.0.gem
+- vendor/cache/ruby-progressbar-1.10.1.gem
+- vendor/cache/simplecov-0.18.1.gem
+- vendor/cache/simplecov-html-0.11.0.gem
+- vendor/cache/unicode-display_width-1.6.1.gem
+homepage: https://github.com/GetTerminus/instrument-all-the-things
+licenses: []
+metadata:
+  allowed_push_host: https://www.rubygems.org
+  homepage_uri: https://github.com/GetTerminus/instrument-all-the-things
+  source_code_uri: https://github.com/GetTerminus/instrument-all-the-things
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.1
+signing_key: 
+specification_version: 4
+summary: Make instrumentation with DataDog easy peasy
+test_files: []