raygun-apm 1.1.15.pre3

data/lib/raygun/apm/hooks/httpclient.rb

@@ -0,0 +1,43 @@
+require 'httpclient'
+
+module Raygun
+  module Apm
+    module Hooks
+      module HTTPClient
+        private
+
+        def do_request(method, uri, query, body, header, &filtered_block)
+          if tracer = Raygun::Apm::Tracer.instance
+            started = tracer.now
+            response = super
+            ended = tracer.now
+            event = raygun_apm_http_out_event
+            event[:pid] = Process.pid
+            event[:url] = raygun_apm_url(uri, query)
+            event[:verb] = method.to_s.upcase
+            event[:status] = response.code.to_i
+            event[:duration] = ended - started
+            event[:timestamp] = started
+            event[:tid] = tracer.get_thread_id(Thread.current)
+            tracer.emit(event)
+            response
+          else
+            super
+          end
+        end
+
+        def raygun_apm_url(uri, query)
+          uri = uri.to_param if uri.respond_to?(:to_param)
+          query = query.to_param if query.respond_to?(:to_param)
+          query && uri ? URI.join(uri, query).to_s : uri.to_s
+        end
+
+        def raygun_apm_http_out_event
+          @_raygun_apm_http_out_event ||= Raygun::Apm::Event::HttpOut.new
+        end
+      end
+    end
+  end
+end
+
+Raygun::Apm::Tracer.patch(HTTPClient, Raygun::Apm::Hooks::HTTPClient)

data/lib/raygun/apm/hooks/internals.rb

@@ -0,0 +1,95 @@
+require 'thread'
+
+module Raygun
+  module Apm
+    module Hooks
+      module Object
+        def system(*args)
+          super
+        end
+
+        def sleep(*args)
+          super
+        end
+
+        def exec(*args)
+          super
+        end
+
+        def spawn(*args)
+          super
+        end
+
+        def fork(*args)
+          super
+        end
+      end
+
+      module IO
+        def sycall(*args)
+          super
+        end
+
+        def open(*args)
+          super
+        end
+
+        def puts(*args)
+          super
+        end
+
+        def gets(*args)
+          super
+        end
+
+        def readline(*args)
+          super
+        end
+
+        def readlines(*args)
+          super
+        end
+      end
+
+      module Random
+        def srand(*args)
+          super
+        end
+
+        def rand(*args)
+          super
+        end
+      end
+
+      module Signal
+        def trap(*args)
+          super
+        end
+      end
+
+      module Mutex
+        def synchronize(*args)
+          super
+        end
+
+        def lock(*args)
+          super
+        end
+
+        def unlock(*args)
+          super
+        end
+
+        def sleep(*args)
+          super
+        end
+      end
+    end
+  end
+end
+
+Raygun::Apm::Tracer.patch(Object, Raygun::Apm::Hooks::Object)
+Raygun::Apm::Tracer.patch(IO, Raygun::Apm::Hooks::IO)
+Raygun::Apm::Tracer.patch(Random, Raygun::Apm::Hooks::Random)
+Raygun::Apm::Tracer.patch(Signal, Raygun::Apm::Hooks::Signal)
+Raygun::Apm::Tracer.patch(Thread::Mutex, Raygun::Apm::Hooks::Mutex)

data/lib/raygun/apm/hooks/mongodb.rb

@@ -0,0 +1,45 @@
+require 'mongo'
+
+module Raygun
+  module Apm
+    module Hooks
+      module MongoDB
+        def do_execute(connection, client, options = {})
+          result = nil
+          if tracer = Raygun::Apm::Tracer.instance
+            started = tracer.now
+            result = super
+            ended = tracer.now
+            event = raygun_apm_sql_event
+            event[:pid] = Process.pid
+            event[:query] = raygun_format_query(connection)
+            event[:provider] = "mongodb"
+            event[:host] = connection.address.to_s
+            event[:database] = client.database.name
+            event[:duration] = ended - started
+            event[:timestamp] = started
+            event[:tid] = tracer.get_thread_id(Thread.current)
+            tracer.emit(event)
+            result
+          else
+            super
+          end
+        end
+
+        private
+        def raygun_format_query(connection)
+          payload = message(connection).payload
+          return "#{payload["database_name"]}.#{payload["command_name"]} #{payload["command"]}}"
+        end
+
+        def raygun_apm_sql_event
+          @_raygun_apm_sql_event ||= Raygun::Apm::Event::Sql.new
+        end
+      end
+    end
+  end
+end
+
+Mongo::Operation.constants.each do |operation|
+  Raygun::Apm::Tracer.patch(Mongo::Operation.const_get(operation), Raygun::Apm::Hooks::MongoDB) rescue nil
+end

data/lib/raygun/apm/hooks/net_http.rb

@@ -0,0 +1,44 @@
+require 'net/http'
+
+module Raygun
+  module Apm
+    module Hooks
+      module Net
+        module HTTP
+          private
+
+          def transport_request(request)
+            if tracer = Raygun::Apm::Tracer.instance
+              started = tracer.now
+              response = super
+              ended = tracer.now
+              event = raygun_apm_http_out_event
+              event[:pid] = Process.pid
+              event[:url] = raygun_apm_url(request)
+              event[:verb] = request.method
+              event[:status] = response.code.to_i
+              event[:duration] = ended - started
+              event[:timestamp] = started
+              event[:tid] = tracer.get_thread_id(Thread.current)
+              tracer.emit(event)
+              response
+            else
+              super
+            end
+          end
+
+          def raygun_apm_url(request)
+            return request.uri.to_s if request.uri
+            "#{use_ssl? ? "https" : "http"}://#{request["host"] || address}#{request.path}"
+          end
+
+          def raygun_apm_http_out_event
+            @_raygun_apm_http_out_event ||= Raygun::Apm::Event::HttpOut.new
+          end
+        end
+      end
+    end
+  end
+end
+
+Raygun::Apm::Tracer.patch(::Net::HTTP, Raygun::Apm::Hooks::Net::HTTP)

data/lib/raygun/apm/hooks/redis.rb

@@ -0,0 +1,46 @@
+module Raygun
+  module Apm
+    module Hooks
+      module Redis
+        def process(commands)
+          result = nil
+          if tracer = Raygun::Apm::Tracer.instance
+            started = tracer.now
+            result = super
+            ended = tracer.now
+            event = raygun_apm_sql_event
+            event[:pid] = Process.pid
+            event[:query] = raygun_format_query(commands)
+            event[:provider] = "redis"
+            event[:host] = "#{host}:#{port}"
+            event[:database] = db.to_s
+            event[:duration] = ended - started
+            event[:timestamp] = started
+            event[:tid] = tracer.get_thread_id(Thread.current)
+            tracer.emit(event)
+            result
+          else
+            super
+          end
+        end
+
+        private
+        def raygun_format_query(commands)
+          commands.map do |command|
+            command.map(&:to_s).join(" ")
+          end.join(", ")
+        end
+
+        def raygun_apm_sql_event
+          @_raygun_apm_sql_event ||= Raygun::Apm::Event::Sql.new
+        end
+
+        def raygun_redis_call
+          yield
+        end
+      end
+    end
+  end
+end
+
+Raygun::Apm::Tracer.patch(::Redis::Client, Raygun::Apm::Hooks::Redis)

data/lib/raygun/apm/tracer.rb

@@ -0,0 +1,131 @@
+require 'raygun/apm/blacklist'
+require 'rbconfig'
+
+module Raygun
+  module Apm
+    class Tracer
+      @__mutex = Mutex.new
+
+      @__pids ||= {}
+      class << self
+        def synchronize(&block)
+          @__mutex.synchronize { block.call }
+        end
+
+        def instance
+          @__pids[Process.pid]
+        end
+
+        def instance=(tracer)
+          @__pids[Process.pid] = tracer
+        end
+
+        def patch(concern, hook)
+          concern.prepend(hook) unless concern.ancestors.include?(hook)
+        end
+      end
+
+      attr_accessor :config
+
+      def initialize(env=ENV)
+        configure(env)
+        initialize_blacklist
+        register_known_library_paths
+        run_agent_connectivity_diagnostics
+        require_hooks
+        ObjectSpace.define_finalizer(self, proc{ disable_tracepoints })
+      # Any fails here is kamikaze for the tracer
+      rescue => e
+        # XXX works for the middleware wrapped case, not for standalone - revisit
+        raise Raygun::Apm::FatalError, "Raygun APM tracer could not be initialized: #{e.message} #{e.backtrace.join("\n")}"
+      end
+
+      def udp_sink!
+        sock = UDPSocket.new
+        # For UDP sockets, SO_SNDBUF is the max packet size and NOT send buffer as with a connection oriented transport
+        sock.setsockopt(Socket::SOL_SOCKET, Socket::SO_SNDBUF, Tracer::BATCH_PACKET_SIZE)
+        self.udp_sink(
+          socket: sock,
+          host: config.proton_udp_host,
+          port: config.proton_udp_port,
+          receive_buffer_size: sock.getsockopt(Socket::SOL_SOCKET, Socket::SO_RCVBUF).int
+        )
+      rescue => e
+        # XXX works for the middleware wrapped case, not for standalone - revisit
+        raise Raygun::Apm::FatalError, "Raygun APM UDP sink could not be initialized: #{e.message} #{e.backtrace.join("\n")}"
+      end
+
+      def tcp_sink!
+        self.tcp_sink(
+          host: config.proton_tcp_host,
+          port: config.proton_tcp_port
+        )
+      rescue => e
+        # XXX works for the middleware wrapped case, not for standalone - revisit
+        raise Raygun::Apm::FatalError, "Raygun APM TCP sink could not be initialized: #{e.message} #{e.backtrace.join("\n")}"
+      end
+
+      def enable_sink!
+        if config.proton_network_mode == "Udp"
+          udp_sink!
+        elsif config.proton_network_mode == "Tcp"
+          tcp_sink!
+        end
+      end
+
+      private
+      def configure(env)
+        @config = Config.new(env)
+        # Special assignments from config to the Tracer
+        self.log_level = config.loglevel
+        self.environment = config.environment
+        self.api_key = config.proton_api_key
+      end
+
+      def initialize_blacklist
+        @blacklist_parser = Raygun::Apm::Blacklist::Parser.new(self)
+        file = @config.blacklist_file
+        @blacklist = if file && File.exist?(file)
+          File.readlines(file)
+        else
+          []
+        end
+        # Defaults
+        @blacklist_parser.add_filters Raygun::Apm::Blacklist.resolve_entries
+        # From file
+        @blacklist_parser.add_filters @blacklist
+      end
+
+      def register_known_library_paths
+        if defined?(Bundler)
+          libs = Bundler.load.specs.map(&:full_gem_path).sort << RbConfig::CONFIG['rubylibdir']
+          libs.delete(Dir.getwd)
+          self.register_libraries libs
+        else
+          self.register_libraries [RbConfig::CONFIG['rubylibdir']]
+        end
+      end
+
+      def run_agent_connectivity_diagnostics
+        check = Raygun::Apm::Diagnostics.new
+        check.verify_agent(self)
+      end
+
+      def require_hooks
+        require "raygun/apm/hooks/internals" if @config.proton_hook_internals
+        require "raygun/apm/hooks/net_http"
+        # conditionally required - may not be bundled
+        conditional_hooks = %w(httpclient excon mongodb)
+        conditional_hooks.each do |hook|
+          begin
+            require "raygun/apm/hooks/#{hook}"
+          rescue LoadError
+          end
+        end
+        if @config.proton_hook_redis
+          require "raygun/apm/hooks/redis" if defined?(Redis::Client)
+        end
+      end
+    end
+  end
+end

data/lib/raygun/apm/version.rb

@@ -0,0 +1,6 @@
+module Raygun
+  module Apm
+    VERSION = "1.1.15.pre3"
+    MINIMUM_AGENT_VERSION = "1.0.1190.0"
+  end
+end

data/lib/raygun/apm.rb

@@ -0,0 +1,18 @@
+require "raygun/apm/version"
+require "socket"
+
+begin
+  # Attempt to load a precompiled shared object (released gem)
+  RUBY_VERSION =~ /(\d+\.\d+)/
+  require "raygun/#{$1}/raygun_ext"
+rescue LoadError
+  # Attempt to load the development specific extension (non-released gem, local dev)
+  require "raygun/raygun_ext"
+end
+
+require "raygun/apm/config"
+require "raygun/apm/diagnostics"
+require "raygun/apm/blacklist/parser"
+require "raygun/apm/blacklist/translator"
+require "raygun/apm/tracer"
+require "raygun/apm/event"

data/raygun-apm.gemspec

@@ -0,0 +1,43 @@
+
+require File.expand_path('../lib/raygun/apm/version', __FILE__)
+
+Gem::Specification.new do |spec|
+  spec.name          = "raygun-apm"
+  spec.version       = Raygun::Apm::VERSION
+
+  spec.authors       = ["Raygun Limited"]
+  spec.email         = ["ruby-apm@raygun.io", "support@raygun.com"]
+
+  spec.summary       = %q{Raygun application performance monitoring core Profiler}
+  spec.homepage      = "https://raygun.com/platform/apm"
+  spec.license       = "MIT"
+
+  spec.files         = Dir['README.rdoc', 'raygun-apm.gemspec', 'LICENSE', 'LICENSE.bipbuffer', 'COPYING.rax', 'lib/**/*', 'bin/**/*'].reject { |f| f.match(/raygun_ext\./) }
+  spec.files         += Dir['lib/raygun/**/{2}*/raygun_ext.*']
+  spec.bindir        = "bin"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib", "ext"]
+
+  spec.platform = Gem::Platform::RUBY
+
+  spec.extensions = ["ext/raygun/extconf.rb"]
+  spec.required_ruby_version = '>= 3.0.0'
+
+  # Required at runtime for native extension compilation against Ruby VM internals
+  spec.add_dependency "debase-ruby_core_source", ">= 3.3.6"
+
+  spec.add_development_dependency "debase-ruby_core_source", ">= 3.3.6"
+  spec.add_development_dependency "bundler", ">= 2.2.15"
+  spec.add_development_dependency "rake", "~> 13.0.3"
+  spec.add_development_dependency "minitest", "~> 5.16"
+  spec.add_development_dependency "rake-compiler", "~> 1.1.1"
+  spec.add_development_dependency "rake-compiler-dock", "~> 1.11.0"
+  spec.add_development_dependency "benchmark_driver", "~> 0.15.9"
+  spec.add_development_dependency "faraday", "~> 1.0.1"
+  spec.add_development_dependency "multipart-post", "~> 2.1.1"
+  spec.add_development_dependency "rest-client", "~> 2.1.0"
+  spec.add_development_dependency "excon", "~> 0.73.0"
+  spec.add_development_dependency "httparty", "~> 0.18.0"
+  spec.add_development_dependency "httpclient", "~> 2.8.3"
+  spec.add_development_dependency "mongoid", "~> 7.1.2"
+end