Contents

require 'skylight/formatters/http'

module Skylight
  module Probes
    module NetHTTP
      class Probe
        DISABLED_KEY = :__skylight_net_http_disabled

        def self.disable
          Thread.current[DISABLED_KEY] = true
          yield
        ensure
          Thread.current[DISABLED_KEY] = false
        end

        def self.disabled?
          !!Thread.current[DISABLED_KEY]
        end

        def install
          Net::HTTP.class_eval do
            alias request_without_sk request

            def request(req, body = nil, &block)
              if !started? || Skylight::Probes::NetHTTP::Probe.disabled?
                return request_without_sk(req, body, &block)
              end

              method = req.method

              # req['host'] also includes special handling for default ports
              host, port = req['host'] ? req['host'].split(':') : nil

              # If we're connected with a persistent socket
              host ||= self.address
              port ||= self.port

              path   = req.path
              scheme = use_ssl? ? "https" : "http"

              # Contained in the path
              query  = nil

              opts = Formatters::HTTP.build_opts(method, scheme, host, port, path, query)

              Skylight.instrument(opts) do
                request_without_sk(req, body, &block)
              end
            end
          end
        end
      end
    end

    register("Net::HTTP", "net/http", NetHTTP::Probe.new)
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
skylight-1.4.0.beta	lib/skylight/probes/net_http.rb
skylight-1.3.1	lib/skylight/probes/net_http.rb
skylight-1.3.0	lib/skylight/probes/net_http.rb
skylight-1.2.2	lib/skylight/probes/net_http.rb
skylight-1.2.1	lib/skylight/probes/net_http.rb
skylight-1.2.0	lib/skylight/probes/net_http.rb
skylight-1.1.0	lib/skylight/probes/net_http.rb