class RorVsWild::Plugin::NetHttp

Constants

HTTP

Public Class Methods

ignore(&block) click to toggle source
# File lib/rorvswild/plugin/net_http.rb, line 36
def self.ignore(&block)
  old_value = Thread.current[:rorvswild_ignore_net_http]
  Thread.current[:rorvswild_ignore_net_http] = true
  block.call
ensure
  Thread.current[:rorvswild_ignore_net_http] = old_value
end
setup() click to toggle source
# File lib/rorvswild/plugin/net_http.rb, line 6
def self.setup
  return if !defined?(Net::HTTP)
  return if Net::HTTP.method_defined?(:request_without_rorvswild)

  Net::HTTP.class_eval do
    alias_method :request_without_rorvswild, :request

    def request(req, body = nil, &block)
      if Thread.current[:rorvswild_ignore_net_http]
        request_without_rorvswild(req, body, &block)
      else
        request_with_rorvswild(req, body, &block)
      end
    end

    def request_with_rorvswild(req, body = nil, &block)
      return request_without_rorvswild(req, body, &block) if !RorVsWild.agent || request_called_twice?
      RorVsWild.agent.measure_section("#{req.method} #{address}", kind: HTTP) do
        request_without_rorvswild(req, body, &block)
      end
    end

    def request_called_twice?
      # Net::HTTP#request calls itself when connection is not started.
      # This condition prevents from counting twice the request.
      (current_section = RorVsWild::Section.current) && current_section.kind == HTTP
    end
  end
end

Public Instance Methods

request(req, body = nil, &block) click to toggle source
# File lib/rorvswild/plugin/net_http.rb, line 13
def request(req, body = nil, &block)
  if Thread.current[:rorvswild_ignore_net_http]
    request_without_rorvswild(req, body, &block)
  else
    request_with_rorvswild(req, body, &block)
  end
end
request_called_twice?() click to toggle source
# File lib/rorvswild/plugin/net_http.rb, line 28
def request_called_twice?
  # Net::HTTP#request calls itself when connection is not started.
  # This condition prevents from counting twice the request.
  (current_section = RorVsWild::Section.current) && current_section.kind == HTTP
end
request_with_rorvswild(req, body = nil, &block) click to toggle source
# File lib/rorvswild/plugin/net_http.rb, line 21
def request_with_rorvswild(req, body = nil, &block)
  return request_without_rorvswild(req, body, &block) if !RorVsWild.agent || request_called_twice?
  RorVsWild.agent.measure_section("#{req.method} #{address}", kind: HTTP) do
    request_without_rorvswild(req, body, &block)
  end
end